public class Application implements ApplicationRunner
{
@Autowired
- Consumer<?, ?> consumer;
+ Consumer<?, ?> kafkaConsumer;
@Autowired
SimpleConsumer simpleConsumer;
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting SimpleConsumer");
- simpleConsumer.start();
+ simpleConsumer.run();
}
@PreDestroy
public void stop() throws ExecutionException, InterruptedException
{
log.info("Signaling SimpleConsumer to quit its work");
- consumer.wakeup();
+ 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");
}