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 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<String, Message> consumer;
+ Consumer<?, ?> kafkaConsumer;
@Autowired
SimpleConsumer simpleConsumer;
- Future<?> consumerJob;
-
@Override
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting SimpleConsumer");
- consumerJob = executorService.submit(simpleConsumer);
+ simpleConsumer.run();
}
@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();
+ 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");
}