import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.scheduling.annotation.EnableAsync;
import javax.annotation.PreDestroy;
import java.util.concurrent.ExecutionException;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Future;
@SpringBootApplication
+@EnableAsync
@Slf4j
public class Application implements ApplicationRunner
{
- @Autowired
- ExecutorService executorService;
@Autowired
Consumer<?, ?> consumer;
@Autowired
SimpleConsumer simpleConsumer;
- Future<?> consumerJob;
-
@Override
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting SimpleConsumer");
- consumerJob = executorService.submit(simpleConsumer);
+ simpleConsumer.start();
}
@PreDestroy
- public void shutdown() throws ExecutionException, InterruptedException
+ public void stop() throws ExecutionException, InterruptedException
{
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");
}