Browse Source
Reviewers: Bill Bejeck <bill@confluent.io>, Boyang Chen <boyang@confluent.io>, Bruno Cadonna <bruno@confluent.io>, Guozhang Wang <guozhang@confuent.io>pull/6879/head
Matthias J. Sax
6 years ago
committed by
GitHub
13 changed files with 119 additions and 24 deletions
@ -0,0 +1,90 @@
@@ -0,0 +1,90 @@
|
||||
/* |
||||
* 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.tests; |
||||
|
||||
import org.apache.kafka.common.utils.Utils; |
||||
import org.apache.kafka.streams.KafkaStreams; |
||||
import org.apache.kafka.streams.StreamsBuilder; |
||||
import org.apache.kafka.streams.StreamsConfig; |
||||
import org.apache.kafka.streams.kstream.KStream; |
||||
import org.apache.kafka.streams.processor.AbstractProcessor; |
||||
import org.apache.kafka.streams.processor.ProcessorContext; |
||||
import org.apache.kafka.streams.processor.ProcessorSupplier; |
||||
|
||||
import java.util.Properties; |
||||
|
||||
public class StreamsUpgradeTest { |
||||
|
||||
@SuppressWarnings("unchecked") |
||||
public static void main(final String[] args) throws Exception { |
||||
if (args.length < 2) { |
||||
System.err.println("StreamsUpgradeTest requires two argument (kafka-url, properties-file) but only " + args.length + " provided: " |
||||
+ (args.length > 0 ? args[0] : "")); |
||||
} |
||||
final String kafka = args[0]; |
||||
final String propFileName = args.length > 1 ? args[1] : null; |
||||
|
||||
final Properties streamsProperties = Utils.loadProps(propFileName); |
||||
|
||||
System.out.println("StreamsTest instance started (StreamsUpgradeTest v2.2)"); |
||||
System.out.println("kafka=" + kafka); |
||||
System.out.println("props=" + streamsProperties); |
||||
|
||||
final StreamsBuilder builder = new StreamsBuilder(); |
||||
final KStream dataStream = builder.stream("data"); |
||||
dataStream.process(printProcessorSupplier()); |
||||
dataStream.to("echo"); |
||||
|
||||
final Properties config = new Properties(); |
||||
config.setProperty(StreamsConfig.APPLICATION_ID_CONFIG, "StreamsUpgradeTest"); |
||||
config.setProperty(StreamsConfig.BOOTSTRAP_SERVERS_CONFIG, kafka); |
||||
config.put(StreamsConfig.COMMIT_INTERVAL_MS_CONFIG, 1000); |
||||
config.putAll(streamsProperties); |
||||
|
||||
final KafkaStreams streams = new KafkaStreams(builder.build(), config); |
||||
streams.start(); |
||||
|
||||
Runtime.getRuntime().addShutdownHook(new Thread(() -> { |
||||
streams.close(); |
||||
System.out.println("UPGRADE-TEST-CLIENT-CLOSED"); |
||||
System.out.flush(); |
||||
})); |
||||
} |
||||
|
||||
private static <K, V> ProcessorSupplier<K, V> printProcessorSupplier() { |
||||
return () -> new AbstractProcessor<K, V>() { |
||||
private int numRecordsProcessed = 0; |
||||
|
||||
@Override |
||||
public void init(final ProcessorContext context) { |
||||
System.out.println("[2.2] initializing processor: topic=data taskId=" + context.taskId()); |
||||
numRecordsProcessed = 0; |
||||
} |
||||
|
||||
@Override |
||||
public void process(final K key, final V value) { |
||||
numRecordsProcessed++; |
||||
if (numRecordsProcessed % 100 == 0) { |
||||
System.out.println("processed " + numRecordsProcessed + " records from topic=data"); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void close() {} |
||||
}; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue