Passing resource to the subsequent conduit correctly

snoyberg · Apr 25, 2024 · 8a4fe46 · 8a4fe46
1 parent 0deecaa
commit 8a4fe46
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/conduit-extra/Data/Conduit/Binary.hs b/conduit-extra/Data/Conduit/Binary.hs
@@ -321,7 +321,8 @@ sinkCacheLength = do
         (\(fp, h) -> IO.hClose h `finally` removeFile fp)
     len <- sinkHandleLen h
     liftIO $ IO.hClose h
-    return (len, CC.sourceFile fp >> release releaseKey)
+    cleanup <- fromMaybe (pure ()) <$> unprotect releaseKey
+    return (len, CC.sourceFile fp >> void (register cleanup))
   where
     sinkHandleLen :: MonadResource m => IO.Handle -> ConduitT S.ByteString o m Word64
     sinkHandleLen h =