juplo.de Git - demos/kafka/training/blob - src/main/java/de/juplo/kafka/Application.java

   1 package de.juplo.kafka;
   2
   3 import lombok.extern.slf4j.Slf4j;
   4 import org.apache.kafka.clients.consumer.Consumer;
   5 import org.springframework.beans.factory.annotation.Autowired;
   6 import org.springframework.boot.ApplicationArguments;
   7 import org.springframework.boot.ApplicationRunner;
   8 import org.springframework.boot.SpringApplication;
   9 import org.springframework.boot.autoconfigure.SpringBootApplication;
  10 import org.springframework.context.ConfigurableApplicationContext;
  11 import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
  12 import org.springframework.util.concurrent.ListenableFuture;
  13
  14 import javax.annotation.PreDestroy;
  15 import java.util.concurrent.ExecutionException;
  16
  17
  18 @SpringBootApplication
  19 @Slf4j
  20 public class Application implements ApplicationRunner
  21 {
  22   @Autowired
  23   ThreadPoolTaskExecutor taskExecutor;
  24   @Autowired
  25   Consumer<?, ?> kafkaConsumer;
  26   @Autowired
  27   SimpleConsumer simpleConsumer;
  28   @Autowired
  29   ConfigurableApplicationContext context;
  30
  31   ListenableFuture<Integer> consumerJob;
  32
  33   @Override
  34   public void run(ApplicationArguments args) throws Exception
  35   {
  36     log.info("Starting SimpleConsumer");
  37     consumerJob = taskExecutor.submitListenable(simpleConsumer);
  38     consumerJob.addCallback(
  39       exitStatus ->
  40       {
  41         log.info("SimpleConsumer exited normally, exit-status: {}", exitStatus);
  42         SpringApplication.exit(context, () -> exitStatus);
  43         },
  44       t ->
  45       {
  46         log.error("SimpleConsumer exited abnormally!", t);
  47         SpringApplication.exit(context, () -> 2);
  48       });
  49   }
  50
  51   @PreDestroy
  52   public void shutdown() throws ExecutionException, InterruptedException
  53   {
  54     log.info("Signaling SimpleConsumer to quit its work");
  55     kafkaConsumer.wakeup();
  56     log.info("Waiting for SimpleConsumer to finish its work");
  57     consumerJob.get();
  58     log.info("SimpleConsumer finished its work");
  59   }
  60
  61
  62   public static void main(String[] args)
  63   {
  64     SpringApplication.run(Application.class, args);
  65   }
  66 }