Examples with GlobalStateUpdateTask org.apache.kafka.streams.processor.internals.GlobalStateUpdateTask used on opensource projects

Example 1 with GlobalStateUpdateTask

use of org.apache.kafka.streams.processor.internals.GlobalStateUpdateTask in project kafka by apache.

the class TopologyTestDriver method setupGlobalTask.

private void setupGlobalTask(final Time mockWallClockTime, final StreamsConfig streamsConfig, final StreamsMetricsImpl streamsMetrics, final ThreadCache cache) {
    if (globalTopology != null) {
        final MockConsumer<byte[], byte[]> globalConsumer = new MockConsumer<>(OffsetResetStrategy.NONE);
        for (final String topicName : globalTopology.sourceTopics()) {
            final TopicPartition partition = new TopicPartition(topicName, 0);
            globalPartitionsByInputTopic.put(topicName, partition);
            offsetsByTopicOrPatternPartition.put(partition, new AtomicLong());
            globalConsumer.updatePartitions(topicName, Collections.singletonList(new PartitionInfo(topicName, 0, null, null, null)));
            globalConsumer.updateBeginningOffsets(Collections.singletonMap(partition, 0L));
            globalConsumer.updateEndOffsets(Collections.singletonMap(partition, 0L));
        }
        globalStateManager = new GlobalStateManagerImpl(logContext, mockWallClockTime, globalTopology, globalConsumer, stateDirectory, stateRestoreListener, streamsConfig);
        final GlobalProcessorContextImpl globalProcessorContext = new GlobalProcessorContextImpl(streamsConfig, globalStateManager, streamsMetrics, cache, mockWallClockTime);
        globalStateManager.setGlobalProcessorContext(globalProcessorContext);
        globalStateTask = new GlobalStateUpdateTask(logContext, globalTopology, globalProcessorContext, globalStateManager, new LogAndContinueExceptionHandler());
        globalStateTask.initialize();
        globalProcessorContext.setRecordContext(null);
    } else {
        globalStateManager = null;
        globalStateTask = null;
    }
}