package de.juplo.kafka;
import lombok.extern.slf4j.Slf4j;
+import org.apache.kafka.clients.consumer.Consumer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+
+import javax.annotation.PreDestroy;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Future;
@SpringBootApplication
public class Application implements ApplicationRunner
{
@Autowired
- EndlessConsumer endlessConsumer;
+ ThreadPoolTaskExecutor taskExecutor;
+ @Autowired
+ Consumer<?, ?> kafkaConsumer;
+ @Autowired
+ SimpleConsumer simpleConsumer;
+ Future<?> consumerJob;
@Override
public void run(ApplicationArguments args) throws Exception
{
- log.info("Starting EndlessConsumer");
- endlessConsumer.start();
+ log.info("Starting SimpleConsumer");
+ consumerJob = taskExecutor.submit(simpleConsumer);
+ }
+
+ @PreDestroy
+ public void shutdown() throws ExecutionException, InterruptedException
+ {
+ log.info("Signaling SimpleConsumer to quit its work");
+ kafkaConsumer.wakeup();
+ log.info("Waiting for SimpleConsumer to finish its work");
+ consumerJob.get();
+ log.info("SimpleConsumer finished its work");
}