@@ -75,7 +75,8 @@ module OddJobs.Job
7575where
7676
7777import OddJobs.Types
78−import Data.Pool
78+import qualified Data.Pool as Pool
79+import Data.Pool(Pool)
7980import Data.Text as T
8081import Database.PostgreSQL.Simple as PGS
8182import Database.PostgreSQL.Simple.Notification
@@ -256,6 +257,10 @@ concatJobDbColumns = concatJobDbColumns_ jobDbColumns ""
256257findJobByIdQuery :: PGS.Query
257258findJobByIdQuery = "SELECT " <> concatJobDbColumns <> " FROM ? WHERE id = ?"
258259
260+withResource :: MonadUnliftIO m => Pool a -> (a -> m b) -> m b
261+withResource pool fa =
262+ withRunInIO $ \runInIO -> Pool.withResource pool (runInIO . fa)
263+
259264withDbConnection :: (HasJobRunner m)
260265 => (Connection -> m a)
261266 -> m a