package de.juplo.kafka;
+import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.ApplicationArguments;
+import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
-import org.springframework.boot.context.properties.EnableConfigurationProperties;
-import org.springframework.context.annotation.Bean;
-import org.springframework.util.Assert;
-import java.util.concurrent.Executors;
+import javax.annotation.PreDestroy;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.TimeUnit;
@SpringBootApplication
-@EnableConfigurationProperties(ApplicationProperties.class)
-public class Application
+@Slf4j
+public class Application implements ApplicationRunner
{
@Autowired
- ApplicationProperties properties;
+ EndlessConsumer endlessConsumer;
+ @Autowired
+ ExecutorService executor;
+
+ @Override
+ public void run(ApplicationArguments args) throws Exception
+ {
+ log.info("Starting EndlessConsumer");
+ endlessConsumer.start();
+ }
- @Bean
- public EndlessConsumer consumer(PartitionStatisticsRepository repository)
+ @PreDestroy
+ public void shutdown()
{
- Assert.hasText(properties.getBootstrapServer(), "consumer.bootstrap-server must be set");
- Assert.hasText(properties.getGroupId(), "consumer.group-id must be set");
- Assert.hasText(properties.getClientId(), "consumer.client-id must be set");
- Assert.hasText(properties.getTopic(), "consumer.topic must be set");
-
- EndlessConsumer consumer =
- new EndlessConsumer(
- Executors.newFixedThreadPool(1),
- repository,
- properties.getBootstrapServer(),
- properties.getGroupId(),
- properties.getClientId(),
- properties.getTopic(),
- properties.getAutoOffsetReset());
-
- consumer.start();
-
- return consumer;
+ try
+ {
+ log.info("Stopping EndlessConsumer");
+ endlessConsumer.stop();
+ }
+ catch (IllegalStateException e)
+ {
+ log.info("Was already stopped: {}", e.toString());
+ }
+ catch (Exception e)
+ {
+ log.error("Unexpected exception while stopping EndlessConsumer: {}", e);
+ }
+
+ 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);
+ }
+ finally
+ {
+ if (!executor.isTerminated())
+ {
+ log.warn("Forcing shutdown of ExecutorService!");
+ executor
+ .shutdownNow()
+ .forEach(runnable -> log.warn("Unprocessed task: {}", runnable.getClass().getSimpleName()));
+ }
+ log.info("Shutdow of ExecutorService finished");
+ }
}
+
public static void main(String[] args)
{
SpringApplication.run(Application.class, args);