import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.kafka.config.KafkaListenerEndpointRegistry;
import javax.annotation.PreDestroy;
-import java.util.List;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.TimeUnit;
@SpringBootApplication
public class Application implements ApplicationRunner
{
@Autowired
- EndlessConsumer endlessConsumer;
- @Autowired
- ExecutorService executor;
+ KafkaListenerEndpointRegistry registry;
+ @Value("${consumer.client-id}")
+ String clientId;
@Override
public void run(ApplicationArguments args) throws Exception
{
log.info("Starting EndlessConsumer");
- endlessConsumer.start();
+ this.registry.getListenerContainer(clientId).start();
}
@PreDestroy
public void stopExecutor()
{
- try
- {
- log.info("Shutting down the ExecutorService.");
- executor.shutdown();
- log.info("Waiting 5 seconds for the ExecutorService to terminate...");
- executor.awaitTermination(5, TimeUnit.SECONDS);
- }
- catch (InterruptedException e)
- {
- log.error("Exception while waiting for the termination of the ExecutorService: {}", e.toString());
- }
- finally
- {
- if (!executor.isShutdown())
- {
- log.warn("Forcing shutdown of ExecutorService!");
- executor
- .shutdownNow()
- .forEach(runnable -> log.warn("Unfinished task: {}", runnable.getClass().getSimpleName()));
- }
- log.info("Shutdow of ExecutorService finished");
- }
+ log.info("Stopping EndlessConsumer");
+ this.registry.getListenerContainer(clientId).stop();
}