content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements. See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to You under the Apache License, Version 2.0
   * (the "License"); you may not use this file except in compliance with
   * the License. You may obtain a copy of the License at
   *
   *    http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.apache.kafka.streams.kstream.internals;
  
  import org.apache.kafka.common.serialization.Serdes;
  import org.apache.kafka.common.utils.Bytes;
  import org.apache.kafka.streams.kstream.Aggregator;
  import org.apache.kafka.streams.kstream.CogroupedKStream;
  import org.apache.kafka.streams.kstream.Initializer;
  import org.apache.kafka.streams.kstream.KGroupedStream;
  import org.apache.kafka.streams.kstream.KTable;
  import org.apache.kafka.streams.kstream.Materialized;
  import org.apache.kafka.streams.kstream.Named;
  import org.apache.kafka.streams.kstream.Reducer;
  import org.apache.kafka.streams.kstream.SessionWindowedKStream;
  import org.apache.kafka.streams.kstream.SessionWindows;
  import org.apache.kafka.streams.kstream.SlidingWindows;
  import org.apache.kafka.streams.kstream.TimeWindowedKStream;
  import org.apache.kafka.streams.kstream.Window;
  import org.apache.kafka.streams.kstream.Windows;
  import org.apache.kafka.streams.kstream.internals.graph.GraphNode;
  import org.apache.kafka.streams.state.KeyValueStore;
  import org.apache.kafka.streams.state.VersionedBytesStoreSupplier;
  
  import java.util.Objects;
  import java.util.Set;
  
  class KGroupedStreamImpl<K, V> extends AbstractStream<K, V> implements KGroupedStream<K, V> {
  
      static final String REDUCE_NAME = "KSTREAM-REDUCE-";
      static final String AGGREGATE_NAME = "KSTREAM-AGGREGATE-";
  
      private final GroupedStreamAggregateBuilder<K, V> aggregateBuilder;
      final boolean repartitionRequired;
      final String userProvidedRepartitionTopicName;
  
      KGroupedStreamImpl(final String name,
                         final Set<String> subTopologySourceNodes,
                         final GroupedInternal<K, V> groupedInternal,
                         final boolean repartitionRequired,
                         final GraphNode graphNode,
                         final InternalStreamsBuilder builder) {
          super(name, groupedInternal.keySerde(), groupedInternal.valueSerde(), subTopologySourceNodes, graphNode, builder);
          this.repartitionRequired = repartitionRequired;
          this.userProvidedRepartitionTopicName = groupedInternal.name();
          this.aggregateBuilder = new GroupedStreamAggregateBuilder<>(
              builder,
              groupedInternal,
              repartitionRequired,
              subTopologySourceNodes,
              name,
              graphNode
          );
      }
  
      @Override
      public KTable<K, V> reduce(final Reducer<V> reducer) {
          return reduce(reducer, Materialized.with(keySerde, valueSerde));
      }
  
      @Override
      public KTable<K, V> reduce(final Reducer<V> reducer,
                                 final Materialized<K, V, KeyValueStore<Bytes, byte[]>> materialized) {
          return reduce(reducer, NamedInternal.empty(), materialized);
      }
  
      @Override
      public KTable<K, V> reduce(final Reducer<V> reducer,
                                 final Named named,
                                 final Materialized<K, V, KeyValueStore<Bytes, byte[]>> materialized) {
          Objects.requireNonNull(reducer, "reducer can't be null");
          Objects.requireNonNull(materialized, "materialized can't be null");
          Objects.requireNonNull(named, "name can't be null");
  
          final MaterializedInternal<K, V, KeyValueStore<Bytes, byte[]>> materializedInternal =
              new MaterializedInternal<>(materialized, builder, REDUCE_NAME);
  
          if (materializedInternal.keySerde() == null) {
              materializedInternal.withKeySerde(keySerde);
          }
          if (materializedInternal.valueSerde() == null) {
              materializedInternal.withValueSerde(valueSerde);
          }
  
          final String name = new NamedInternal(named).orElseGenerateWithPrefix(builder, REDUCE_NAME);
          final KeyValueStoreMaterializer<K, V> storeFactory = new KeyValueStoreMaterializer<>(materializedInternal);
  
          return doAggregate(
              new KStreamReduce<>(storeFactory, reducer),
              name,
              storeFactory
          );
      }
  
      @Override
      public <VOut> KTable<K, VOut> aggregate(final Initializer<VOut> initializer,
                                              final Aggregator<? super K, ? super V, VOut> aggregator,
                                              final Materialized<K, VOut, KeyValueStore<Bytes, byte[]>> materialized) {
          return aggregate(initializer, aggregator, NamedInternal.empty(), materialized);
      }
  
      @Override
      public <VOut> KTable<K, VOut> aggregate(final Initializer<VOut> initializer,
                                              final Aggregator<? super K, ? super V, VOut> aggregator,
                                              final Named named,
                                              final Materialized<K, VOut, KeyValueStore<Bytes, byte[]>> materialized) {
          Objects.requireNonNull(initializer, "initializer can't be null");
          Objects.requireNonNull(aggregator, "aggregator can't be null");
          Objects.requireNonNull(materialized, "materialized can't be null");
          Objects.requireNonNull(named, "named can't be null");
  
          final MaterializedInternal<K, VOut, KeyValueStore<Bytes, byte[]>> materializedInternal =
              new MaterializedInternal<>(materialized, builder, AGGREGATE_NAME);
  
          if (materializedInternal.keySerde() == null) {
              materializedInternal.withKeySerde(keySerde);
          }
  
          final String name = new NamedInternal(named).orElseGenerateWithPrefix(builder, AGGREGATE_NAME);
          final KeyValueStoreMaterializer<K, VOut> storeFactory = new KeyValueStoreMaterializer<>(materializedInternal);
  
          return doAggregate(
              new KStreamAggregate<>(storeFactory, initializer, aggregator),
              name,
              storeFactory
          );
      }
  
      @Override
      public <VOut> KTable<K, VOut> aggregate(final Initializer<VOut> initializer,
                                              final Aggregator<? super K, ? super V, VOut> aggregator) {
          return aggregate(initializer, aggregator, Materialized.with(keySerde, null));
      }
  
      @Override
      public KTable<K, Long> count() {
          return doCount(NamedInternal.empty(), Materialized.with(keySerde, Serdes.Long()));
      }
  
      @Override
      public KTable<K, Long> count(final Named named) {
          Objects.requireNonNull(named, "named can't be null");
          return doCount(named, Materialized.with(keySerde, Serdes.Long()));
      }
  
      @Override
      public KTable<K, Long> count(final Materialized<K, Long, KeyValueStore<Bytes, byte[]>> materialized) {
          return count(NamedInternal.empty(), materialized);
      }
  
      @Override
      public KTable<K, Long> count(final Named named, final Materialized<K, Long, KeyValueStore<Bytes, byte[]>> materialized) {
          Objects.requireNonNull(materialized, "materialized can't be null");
  
          // TODO: remove this when we do a topology-incompatible release
          // we used to burn a topology name here, so we have to keep doing it for compatibility
          if (new MaterializedInternal<>(materialized).storeName() == null) {
              builder.newStoreName(AGGREGATE_NAME);
          }
  
          return doCount(named, materialized);
      }
  
      private KTable<K, Long> doCount(final Named named, final Materialized<K, Long, KeyValueStore<Bytes, byte[]>> materialized) {
          final MaterializedInternal<K, Long, KeyValueStore<Bytes, byte[]>> materializedInternal =
              new MaterializedInternal<>(materialized, builder, AGGREGATE_NAME);
  
          if (materializedInternal.keySerde() == null) {
              materializedInternal.withKeySerde(keySerde);
          }
          if (materializedInternal.valueSerde() == null) {
              materializedInternal.withValueSerde(Serdes.Long());
          }
  
          final String name = new NamedInternal(named).orElseGenerateWithPrefix(builder, AGGREGATE_NAME);
          final KeyValueStoreMaterializer<K, Long> storeFactory = new KeyValueStoreMaterializer<>(materializedInternal);
  
          return doAggregate(
              new KStreamAggregate<>(storeFactory, aggregateBuilder.countInitializer, aggregateBuilder.countAggregator),
              name,
              storeFactory);
      }
  
      @Override
      public <W extends Window> TimeWindowedKStream<K, V> windowedBy(final Windows<W> windows) {
  
          return new TimeWindowedKStreamImpl<>(
              windows,
              builder,
              subTopologySourceNodes,
              name,
              keySerde,
              valueSerde,
              aggregateBuilder,
              graphNode
          );
      }
  
      @Override
      public TimeWindowedKStream<K, V> windowedBy(final SlidingWindows windows) {
  
          return new SlidingWindowedKStreamImpl<>(
                  windows,
                  builder,
                  subTopologySourceNodes,
                  name,
                  keySerde,
                  valueSerde,
                  aggregateBuilder,
                  graphNode
          );
      }
  
      @Override
      public SessionWindowedKStream<K, V> windowedBy(final SessionWindows windows) {
  
          return new SessionWindowedKStreamImpl<>(
              windows,
              builder,
              subTopologySourceNodes,
              name,
              keySerde,
              valueSerde,
              aggregateBuilder,
              graphNode
          );
      }
  
      private <T> KTable<K, T> doAggregate(final KStreamAggProcessorSupplier<K, V, K, T> aggregateSupplier,
                                           final String functionName,
                                           final KeyValueStoreMaterializer<K, T> storeFactory) {
  
          return aggregateBuilder.buildNonWindowed(
              new NamedInternal(functionName),
              storeFactory.storeName(),
              aggregateSupplier,
              storeFactory.queryableStoreName(),
              storeFactory.keySerde(),
              storeFactory.valueSerde(),
              storeFactory.storeSupplier() instanceof VersionedBytesStoreSupplier);
      }
  
      @Override
      public <VOut> CogroupedKStream<K, VOut> cogroup(final Aggregator<? super K, ? super V, VOut> aggregator) {
          Objects.requireNonNull(aggregator, "aggregator can't be null");
          return new CogroupedKStreamImpl<K, VOut>(name, subTopologySourceNodes, graphNode, builder)
              .cogroup(this, aggregator);
      }
  }
  