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.context.annotation.Bean;
+import org.springframework.kafka.core.ConsumerFactory;
import javax.annotation.PreDestroy;
-import java.util.List;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.TimeUnit;
@SpringBootApplication
public class Application implements ApplicationRunner
{
@Autowired
- EndlessConsumer endlessConsumer;
+ Consumer<String, Message> consumer;
@Autowired
- ExecutorService executor;
+ SimpleConsumer simpleConsumer;
@Override
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting EndlessConsumer");
- endlessConsumer.start();
+ simpleConsumer.start();
}
@PreDestroy
- public void stopExecutor()
+ public void shutdown()
{
- try
- {
- log.info("Shutting down the ExecutorService.");
- executor.shutdown();
- log.info("Waiting 5 seconds for the ExecutorService to terminate...");
- executor.awaitTermination(5, TimeUnit.SECONDS);
- }
- catch (InterruptedException e)
- {
- log.error("Exception while waiting for the termination of the ExecutorService: {}", e.toString());
- }
- finally
- {
- if (!executor.isShutdown())
- {
- log.warn("Forcing shutdown of ExecutorService!");
- executor
- .shutdownNow()
- .forEach(runnable -> log.warn("Unfinished task: {}", runnable.getClass().getSimpleName()));
- }
- log.info("Shutdow of ExecutorService finished");
- }
+ log.info("Signaling the consumer to quit its work");
+ consumer.wakeup();
+ }
+
+ @Bean(destroyMethod = "close")
+ public Consumer<String, Message> kafkaConsumer(ConsumerFactory<String, Message> factory)
+ {
+ return factory.createConsumer();
}