import lombok.extern.slf4j.Slf4j;
import org.apache.kafka.clients.producer.KafkaProducer;
import org.apache.kafka.clients.producer.ProducerRecord;
+import org.apache.kafka.common.serialization.LongSerializer;
import org.apache.kafka.common.serialization.StringSerializer;
import javax.annotation.PreDestroy;
import java.util.Properties;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Future;
@Slf4j
private final String id;
private final String topic;
private final int throttleMs;
- private final KafkaProducer<String, String> producer;
+ private final KafkaProducer<String, Long> producer;
private boolean running = false;
private long i = 0;
private long produced = 0;
- private Future<?> future = null;
public EndlessProducer(
ExecutorService executor,
props.put("client.id", clientId);
props.put("acks", acks);
props.put("key.serializer", StringSerializer.class.getName());
- props.put("value.serializer", StringSerializer.class.getName());
+ props.put("value.serializer", LongSerializer.class.getName());
this.producer = new KafkaProducer<>(props);
}
{
for (; running; i++)
{
- send(Long.toString(i%10), Long.toString(i));
+ send(Long.toString(i%10), i);
if (throttleMs > 0)
{
}
catch (Exception e)
{
-
+ log.error("{} - Unexpected Exception:", id, e);
+ }
+ finally
+ {
+ synchronized (this)
+ {
+ running = false;
+ log.info("{} - Stopped - produced {} messages so far", id, produced);
+ }
}
}
- void send(String key, String value)
+ void send(String key, Long value)
{
final long time = System.currentTimeMillis();
- final ProducerRecord<String, String> record = new ProducerRecord<>(
- "test", // Topic
+ final ProducerRecord<String, Long> record = new ProducerRecord<>(
+ topic, // Topic
key, // Key
value // Value
);
log.info("{} - Starting - produced {} messages before", id, produced);
running = true;
- future = executor.submit(this);
+ executor.submit(this);
}
public synchronized void stop() throws ExecutionException, InterruptedException
log.info("{} - Stopping...", id);
running = false;
- future.get();
- log.info("{} - Stopped - produced {} messages so far", id, produced);
}
@PreDestroy