@@ -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
@@ -255,6 +256,10 @@ concatJobDbColumns = concatJobDbColumns_ jobDbColumns ""
255256findJobByIdQuery :: PGS.Query
256257findJobByIdQuery = "SELECT " <> concatJobDbColumns <> " FROM ? WHERE id = ?"
257258
259+withResource :: MonadUnliftIO m => Pool a -> (a -> m b) -> m b
260+withResource pool fa =
261+ withRunInIO $ \runInIO -> Pool.withResource pool (runInIO . fa)
262+
258263withDbConnection :: (HasJobRunner m)
259264 => (Connection -> m a)
260265 -> m a