import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.ConsumerRecords;
import org.apache.kafka.common.errors.WakeupException;
+import org.springframework.scheduling.annotation.Async;
import java.time.Duration;
import java.util.Arrays;
-import java.util.concurrent.ExecutorService;
@Slf4j
@RequiredArgsConstructor
-public class SimpleConsumer implements Runnable
+public class SimpleConsumer
{
- private final ExecutorService executor;
private final String id;
private final String topic;
- private final Consumer<String, Message> consumer;
+ private final Consumer<String, String> consumer;
+ private volatile boolean running = false;
private long consumed = 0;
- @Override
+ @Async
public void run()
{
try
{
log.info("{} - Subscribing to topic test", id);
consumer.subscribe(Arrays.asList(topic));
+ running = true;
while (true)
{
- ConsumerRecords<String, Message> records =
+ ConsumerRecords<String, String> records =
consumer.poll(Duration.ofSeconds(1));
log.info("{} - Received {} messages", id, records.count());
- for (ConsumerRecord<String, Message> record : records)
+ for (ConsumerRecord<String, String> record : records)
{
consumed++;
log.info(
}
finally
{
+ running = false;
log.info("{} - Closing the KafkaConsumer", id);
consumer.close();
log.info("{}: Consumed {} messages in total, exiting!", id, consumed);
}
}
+
+ public boolean isRunning()
+ {
+ return running;
+ }
}