Reset finalizers to avoid leaks

zio · Jul 10, 2024 · 06539bc · 06539bc
1 parent d4af6b6
commit 06539bc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/zio-query/shared/src/main/scala/zio/query/internal/QueryScope.scala b/zio-query/shared/src/main/scala/zio/query/internal/QueryScope.scala
@@ -33,7 +33,10 @@ private[query] object QueryScope {
     def closeAndExitWith[E, A](exit: Exit[E, A])(implicit trace: Trace): IO[E, A] = {
       val finalizers = ref.get
       if (finalizers.isEmpty) exit
-      else ZIO.foreachDiscard(finalizers)(_(exit)) *> exit
+      else {
+        ref.set(Nil)
+        ZIO.foreachDiscard(finalizers)(_(exit)) *> exit
+      }
     }
   }
 }