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.concurrent.ThreadPoolTaskExecutor;
import javax.annotation.PreDestroy;
import java.util.concurrent.ExecutionException;
-import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
public class Application implements ApplicationRunner
{
@Autowired
- ExecutorService executorService;
+ ThreadPoolTaskExecutor taskExecutor;
@Autowired
- Consumer<String, Message> consumer;
+ Consumer<?, ?> kafkaConsumer;
@Autowired
SimpleConsumer simpleConsumer;
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting SimpleConsumer");
- consumerJob = executorService.submit(simpleConsumer);
+ consumerJob = taskExecutor.submit(simpleConsumer);
}
@PreDestroy
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");
}
- @Bean(destroyMethod = "close")
- public Consumer<String, Message> kafkaConsumer(ConsumerFactory<String, Message> factory)
- {
- return factory.createConsumer();
- }
-
public static void main(String[] args)
{