Add withIOCtx

well-typed · Dec 10, 2024 · 11f3cf8 · 11f3cf8
1 parent ab50126
commit 11f3cf8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/System/IO/BlockIO.hs b/src/System/IO/BlockIO.hs
@@ -7,9 +7,10 @@ module System.IO.BlockIO (
 
     -- * I\/O context and initialisation
     IOCtx,
-    initIOCtx,
     IOCtxParams(..),
     defaultIOCtxParams,
+    withIOCtx,
+    initIOCtx,
     closeIOCtx,
 
     -- * Performing I\/O
@@ -34,7 +35,7 @@ import Control.Concurrent.MVar
 import Control.Concurrent.QSemN
 import Control.Concurrent.Chan
 import Control.Exception (mask_, throw, ArrayException(UndefinedElement),
-                          finally, assert, throwIO)
+                          finally, assert, throwIO, bracket)
 import System.IO.Error
 import GHC.IO.Exception (IOErrorType(ResourceVanished, InvalidArgument))
 
@@ -90,6 +91,9 @@ defaultIOCtxParams =
     ioctxConcurrencyLimit = 64 * 3
   }
 
+withIOCtx :: IOCtxParams -> (IOCtx -> IO a) -> IO a
+withIOCtx params = bracket (initIOCtx params) closeIOCtx
+
 initIOCtx :: IOCtxParams -> IO IOCtx
 initIOCtx IOCtxParams {ioctxBatchSizeLimit, ioctxConcurrencyLimit} = do
 #if MIN_VERSION_base(4,16,0)