Log more

zio · Apr 20, 2024 · 6a1a694 · 6a1a694
1 parent b1c526e
commit 6a1a694
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/zio-kafka/src/main/scala/zio/kafka/consumer/Consumer.scala b/zio-kafka/src/main/scala/zio/kafka/consumer/Consumer.scala
@@ -726,11 +726,15 @@ private[consumer] final class ConsumerLive private[consumer] (
         fib <- withStream(control.stream)
                  .onInterrupt(ZIO.logError("withStream in runWithGracefulShutdown interrupted, this should not happen"))
                  .forkDaemon
-        result <- fib.join.onInterrupt(
-                    control.stop *> fib.join.disconnect
-                      .timeout(shutdownTimeout)
-                      .ignore
-                  )
+        result <-
+          fib.join.onInterrupt(
+            control.stop *> fib.join
+              .timeout(shutdownTimeout)
+              .tapErrorCause(cause =>
+                ZIO.logErrorCause("Error joining withStream fiber in runWithGracefulShutdown", cause)
+              )
+              .ignore
+          )
       } yield result
     }