flaky test

LangStream · May 20, 2024 · 58eedc0 · 58eedc0
1 parent 6ada992
commit 58eedc0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/langstream-kafka-runtime/src/main/java/ai/langstream/kafka/runner/KafkaReaderWrapper.java b/langstream-kafka-runtime/src/main/java/ai/langstream/kafka/runner/KafkaReaderWrapper.java
@@ -107,7 +107,11 @@ private OffsetPerPartition parseOffset() throws IOException {
     @Override
     public void close() {
         if (consumer != null) {
-            consumer.close();
+            try {
+                consumer.close();
+            } catch (org.apache.kafka.common.errors.InterruptException e) {
+                log.warn("Interrupted while closing Kafka consumer", e);
+            }
         }
     }