import org.apache.kafka.common.serialization.StringSerializer;
import java.util.Properties;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Future;
@Slf4j
private final KafkaProducer<String, String> producer;
private long produced = 0;
+ private volatile boolean running = true;
+ private volatile boolean done = false;
- public SimpleProducer(String clientId, String topic)
+ public SimpleProducer(String broker, String topic, String clientId)
{
- // tag::create[]
Properties props = new Properties();
- props.put("bootstrap.servers", "localhost:9092");
+ props.put("bootstrap.servers", broker);
+ props.put("client.id", clientId); // Nur zur Wiedererkennung
props.put("key.serializer", StringSerializer.class.getName());
props.put("value.serializer", StringSerializer.class.getName());
- KafkaProducer<String, String> producer = new KafkaProducer<>(props);
- // end::create[]
+ producer = new KafkaProducer<>(props);
- this.id = clientId;
this.topic = topic;
- this.producer = producer;
+ this.id = clientId;
}
public void run()
try
{
- for (; i < 100 ; i++)
+ for (; running ; i++)
{
send(Long.toString(i%10), Long.toString(i));
+ Thread.sleep(500);
}
-
- log.info("{} - Done", id);
}
+ catch (InterruptedException e) {}
finally
{
log.info("{}: Closing the KafkaProducer", id);
producer.close();
log.info("{}: Produced {} messages in total, exiting!", id, produced);
+ done = true;
}
}
public static void main(String[] args) throws Exception
{
- SimpleProducer producer = new SimpleProducer("P", "test");
- producer.run();
+ String broker = ":9092";
+ String topic = "test";
+ String clientId = "DEV";
+
+ switch (args.length)
+ {
+ case 3:
+ clientId = args[2];
+ case 2:
+ topic = args[1];
+ case 1:
+ broker = args[0];
+ }
+
+ SimpleProducer instance = new SimpleProducer(broker, topic, clientId);
+
+ Runtime.getRuntime().addShutdownHook(new Thread(() ->
+ {
+ instance.running = false;
+ while (!instance.done)
+ {
+ log.info("Waiting for main-thread...");
+ try
+ {
+ Thread.sleep(1000);
+ }
+ catch (InterruptedException e) {}
+ }
+ log.info("Shutdown completed.");
+ }));
+
+ log.info(
+ "Running SimpleProducer: broker={}, topic={}, client-id={}",
+ broker,
+ topic,
+ clientId);
+ instance.run();
}
}