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