diff --git a/src/Quasar/ResourceManager.hs b/src/Quasar/ResourceManager.hs index 6f22bb72bc6ea96bc3cf3bf933097f420989da96..940cd52e5d8928a54f2844f23e15e31427281df2 100644 --- a/src/Quasar/ResourceManager.hs +++ b/src/Quasar/ResourceManager.hs @@ -14,6 +14,7 @@ module Quasar.ResourceManager ( onResourceManager, captureDisposable, captureDisposable_, + disposeOnError, liftResourceManagerIO, handleByResourceManager, @@ -193,6 +194,14 @@ captureDisposable action = do captureDisposable_ :: MonadResourceManager m => m () -> m Disposable captureDisposable_ = snd <<$>> captureDisposable +-- | Disposes all resources created by the computation if the computation throws an exception. +disposeOnError :: MonadResourceManager m => m a -> m a +disposeOnError action = do + bracketOnError + newResourceManager + dispose + \resourceManager -> localResourceManager resourceManager action + -- | Run a computation and throw any exception that occurs to the resource manager. -- -- This can be used to run e.g. callbacks that belong to a different resource context.