import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
-import org.springframework.scheduling.annotation.EnableAsync;
import javax.annotation.PreDestroy;
import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Executor;
+import java.util.concurrent.Future;
@SpringBootApplication
-@EnableAsync
@Slf4j
public class Application implements ApplicationRunner
{
+ @Autowired
+ Executor executor;
@Autowired
Consumer<?, ?> kafkaConsumer;
@Autowired
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting SimpleConsumer");
- simpleConsumer.run();
+ executor.execute(simpleConsumer);
}
@PreDestroy
- public void stop() throws ExecutionException, InterruptedException
+ public void shutdown() throws ExecutionException, InterruptedException
{
log.info("Signaling SimpleConsumer to quit its work");
kafkaConsumer.wakeup();
-
- while (simpleConsumer.isRunning())
- {
- log.info("Waiting for SimpleConsumer to finish its work");
- try
- {
- Thread.sleep(1000);
- }
- catch (InterruptedException e) {}
- }
- log.info("SimpleConsumer finished its work");
}
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;
@Slf4j
@RequiredArgsConstructor
-public class SimpleConsumer
+public class SimpleConsumer implements Runnable
{
private final String id;
private final String topic;
private final Consumer<String, String> consumer;
- private volatile boolean running = false;
private long consumed = 0;
- @Async
+ @Override
public void run()
{
try
{
log.info("{} - Subscribing to topic test", id);
consumer.subscribe(Arrays.asList(topic));
- running = true;
while (true)
{
}
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;
- }
}