org.apache.kafka.streams.kstream.internals.graph.StatefulProcessorNode Maven / Gradle / Ivy
/*
* 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.graph;
import org.apache.kafka.streams.kstream.internals.KTableValueGetterSupplier;
import org.apache.kafka.streams.processor.api.ProcessorSupplier;
import org.apache.kafka.streams.processor.internals.InternalTopologyBuilder;
import org.apache.kafka.streams.state.StoreBuilder;
import java.util.Arrays;
import java.util.Set;
import java.util.stream.Stream;
public class StatefulProcessorNode extends ProcessorGraphNode {
private final String[] storeNames;
private final StoreBuilder storeBuilder;
/**
* Create a node representing a stateful processor, where the named stores have already been registered.
*/
public StatefulProcessorNode(final ProcessorParameters processorParameters,
final Set> preRegisteredStores,
final Set> valueGetterSuppliers) {
super(processorParameters.processorName(), processorParameters);
final Stream registeredStoreNames = preRegisteredStores.stream().map(StoreBuilder::name);
final Stream valueGetterStoreNames = valueGetterSuppliers.stream().flatMap(s -> Arrays.stream(s.storeNames()));
storeNames = Stream.concat(registeredStoreNames, valueGetterStoreNames).toArray(String[]::new);
storeBuilder = null;
}
/**
* Create a node representing a stateful processor, where the named stores have already been registered.
*/
public StatefulProcessorNode(final String nodeName,
final ProcessorParameters processorParameters,
final String[] storeNames) {
super(nodeName, processorParameters);
this.storeNames = storeNames;
this.storeBuilder = null;
}
/**
* Create a node representing a stateful processor,
* where the store needs to be built and registered as part of building this node.
*/
public StatefulProcessorNode(final String nodeName,
final ProcessorParameters processorParameters,
final StoreBuilder materializedKTableStoreBuilder) {
super(nodeName, processorParameters);
this.storeNames = null;
this.storeBuilder = materializedKTableStoreBuilder;
}
@Override
public String toString() {
return "StatefulProcessorNode{" +
"storeNames=" + Arrays.toString(storeNames) +
", storeBuilder=" + storeBuilder +
"} " + super.toString();
}
@Override
public void writeToTopology(final InternalTopologyBuilder topologyBuilder) {
final String processorName = processorParameters().processorName();
final ProcessorSupplier processorSupplier = processorParameters().processorSupplier();
topologyBuilder.addProcessor(processorName, processorSupplier, parentNodeNames());
if (storeNames != null && storeNames.length > 0) {
topologyBuilder.connectProcessorAndStateStores(processorName, storeNames);
}
if (storeBuilder != null) {
topologyBuilder.addStateStore(storeBuilder, processorName);
}
if (processorSupplier.stores() != null) {
for (final StoreBuilder storeBuilder : processorSupplier.stores()) {
topologyBuilder.addStateStore(storeBuilder, processorName);
}
}
// temporary hack until KIP-478 is fully implemented
final org.apache.kafka.streams.processor.ProcessorSupplier oldProcessorSupplier =
processorParameters().oldProcessorSupplier();
if (oldProcessorSupplier != null && oldProcessorSupplier.stores() != null) {
for (final StoreBuilder storeBuilder : oldProcessorSupplier.stores()) {
topologyBuilder.addStateStore(storeBuilder, processorName);
}
}
}
}