X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmain%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fimplementation%2Fkafka%2FKafkaServicesApplicationRunner.java;h=44f411f80ab4d018c325eebf729bafdebdf7e565;hb=a4e68e9998d1c0da34f78a8d4bc1cb10dcf345f4;hp=da7ee75d65ffc881783934ffc8d543e2aa0e9edc;hpb=a3edea0816d84596daa9c2c4fa0f221ac60a2c3f;p=demos%2Fkafka%2Fchat

diff --git a/src/main/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaServicesApplicationRunner.java b/src/main/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaServicesApplicationRunner.java
index da7ee75d..44f411f8 100644
--- a/src/main/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaServicesApplicationRunner.java
+++ b/src/main/java/de/juplo/kafka/chat/backend/implementation/kafka/KafkaServicesApplicationRunner.java
@@ -1,5 +1,6 @@
 package de.juplo.kafka.chat.backend.implementation.kafka;
 
+import jakarta.annotation.PreDestroy;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.boot.ApplicationArguments;
@@ -17,12 +18,18 @@ import org.springframework.stereotype.Component;
 @Slf4j
 public class KafkaServicesApplicationRunner implements ApplicationRunner
 {
-  private final ConsumerTaskExecutor chatRoomChannelTaskExecutor;
+  private final ConsumerTaskRunner consumerTaskRunner;
 
 
   @Override
   public void run(ApplicationArguments args) throws Exception
   {
-    chatRoomChannelTaskExecutor.executeConsumerTask();
+    consumerTaskRunner.executeConsumerTasks();
+  }
+
+  @PreDestroy
+  public void joinConsumerTasks() throws InterruptedException
+  {
+    consumerTaskRunner.joinConsumerTasks();
   }
 }