import org.springframework.kafka.core.ConsumerFactory;
import javax.annotation.PreDestroy;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Future;
@SpringBootApplication
@Slf4j
public class Application implements ApplicationRunner
{
+ @Autowired
+ ExecutorService executorService;
@Autowired
Consumer<String, Message> consumer;
@Autowired
SimpleConsumer simpleConsumer;
+ Future<?> consumerJob;
@Override
public void run(ApplicationArguments args) throws Exception
{
- log.info("Starting EndlessConsumer");
- simpleConsumer.start();
+ log.info("Starting SimpleConsumer");
+ consumerJob = executorService.submit(simpleConsumer);
}
@PreDestroy
- public void shutdown()
+ public void shutdown() throws ExecutionException, InterruptedException
{
- log.info("Signaling the consumer to quit its work");
+ log.info("Signaling SimpleConsumer to quit its work");
consumer.wakeup();
+ log.info("Waiting for SimpleConsumer to finish its work");
+ consumerJob.get();
+ log.info("SimpleConsumer finished its work");
}
@Bean(destroyMethod = "close")