Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* 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.common.utils.Bytes;
import org.apache.kafka.streams.kstream.internals.ConsumedInternal;
import org.apache.kafka.streams.kstream.internals.KTableSource;
import org.apache.kafka.streams.kstream.internals.MaterializedInternal;
import org.apache.kafka.streams.kstream.internals.TimestampedKeyValueStoreMaterializer;
import org.apache.kafka.streams.processor.api.ProcessorSupplier;
import org.apache.kafka.streams.processor.internals.InternalTopologyBuilder;
import org.apache.kafka.streams.state.KeyValueStore;
import org.apache.kafka.streams.state.StoreBuilder;
import org.apache.kafka.streams.state.TimestampedKeyValueStore;
import java.util.Collections;
/**
* Used to represent either a KTable source or a GlobalKTable source. A boolean flag is used to indicate if this represents a GlobalKTable a {@link
* org.apache.kafka.streams.kstream.GlobalKTable}
*/
public class TableSourceNode extends StreamSourceNode {
private final MaterializedInternal materializedInternal;
private final ProcessorParameters processorParameters;
private final String sourceName;
private final boolean isGlobalKTable;
private boolean shouldReuseSourceTopicForChangelog = false;
private TableSourceNode(final String nodeName,
final String sourceName,
final String topic,
final ConsumedInternal consumedInternal,
final MaterializedInternal materializedInternal,
final ProcessorParameters processorParameters,
final boolean isGlobalKTable) {
super(nodeName,
Collections.singletonList(topic),
consumedInternal);
this.sourceName = sourceName;
this.isGlobalKTable = isGlobalKTable;
this.processorParameters = processorParameters;
this.materializedInternal = materializedInternal;
}
public void reuseSourceTopicForChangeLog(final boolean shouldReuseSourceTopicForChangelog) {
this.shouldReuseSourceTopicForChangelog = shouldReuseSourceTopicForChangelog;
}
@Override
public String toString() {
return "TableSourceNode{" +
"materializedInternal=" + materializedInternal +
", processorParameters=" + processorParameters +
", sourceName='" + sourceName + '\'' +
", isGlobalKTable=" + isGlobalKTable +
"} " + super.toString();
}
public static TableSourceNodeBuilder tableSourceNodeBuilder() {
return new TableSourceNodeBuilder<>();
}
@Override
@SuppressWarnings("unchecked")
public void writeToTopology(final InternalTopologyBuilder topologyBuilder) {
final String topicName = topicNames().iterator().next();
// TODO: we assume source KTables can only be timestamped-key-value stores for now.
// should be expanded for other types of stores as well.
final StoreBuilder> storeBuilder =
new TimestampedKeyValueStoreMaterializer<>((MaterializedInternal>) materializedInternal).materialize();
if (isGlobalKTable) {
topologyBuilder.addGlobalStore(
storeBuilder,
sourceName,
consumedInternal().timestampExtractor(),
consumedInternal().keyDeserializer(),
consumedInternal().valueDeserializer(),
topicName,
processorParameters.processorName(),
(ProcessorSupplier) processorParameters.processorSupplier()
);
} else {
topologyBuilder.addSource(consumedInternal().offsetResetPolicy(),
sourceName,
consumedInternal().timestampExtractor(),
consumedInternal().keyDeserializer(),
consumedInternal().valueDeserializer(),
topicName);
topologyBuilder.addProcessor(processorParameters.processorName(), processorParameters.processorSupplier(), sourceName);
// only add state store if the source KTable should be materialized
final KTableSource ktableSource = processorParameters.kTableSourceSupplier();
if (ktableSource.materialized()) {
topologyBuilder.addStateStore(storeBuilder, nodeName());
if (shouldReuseSourceTopicForChangelog) {
storeBuilder.withLoggingDisabled();
topologyBuilder.connectSourceStoreAndTopic(storeBuilder.name(), topicName);
}
}
}
}
public static final class TableSourceNodeBuilder {
private String nodeName;
private String sourceName;
private String topic;
private ConsumedInternal consumedInternal;
private MaterializedInternal materializedInternal;
private ProcessorParameters processorParameters;
private boolean isGlobalKTable = false;
private TableSourceNodeBuilder() {
}
public TableSourceNodeBuilder withSourceName(final String sourceName) {
this.sourceName = sourceName;
return this;
}
public TableSourceNodeBuilder withTopic(final String topic) {
this.topic = topic;
return this;
}
public TableSourceNodeBuilder withMaterializedInternal(final MaterializedInternal materializedInternal) {
this.materializedInternal = materializedInternal;
return this;
}
public TableSourceNodeBuilder withConsumedInternal(final ConsumedInternal consumedInternal) {
this.consumedInternal = consumedInternal;
return this;
}
public TableSourceNodeBuilder withProcessorParameters(final ProcessorParameters processorParameters) {
this.processorParameters = processorParameters;
return this;
}
public TableSourceNodeBuilder withNodeName(final String nodeName) {
this.nodeName = nodeName;
return this;
}
public TableSourceNodeBuilder isGlobalKTable(final boolean isGlobaKTable) {
this.isGlobalKTable = isGlobaKTable;
return this;
}
public TableSourceNode build() {
return new TableSourceNode<>(nodeName,
sourceName,
topic,
consumedInternal,
materializedInternal,
processorParameters,
isGlobalKTable);
}
}
}
