withDB' simple wrapper to disregard exceptions

mightybyte · Dec 22, 2010 · f9e8a95 · f9e8a95
1 parent 7f00b97
commit f9e8a95
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/Snap/Extension/MongoDB.hs b/src/Snap/Extension/MongoDB.hs
@@ -37,10 +37,19 @@ import           Snap.Types
 class MonadSnap m => MonadMongoDB m where
 
   ----------------------------------------------------------------------------
-  -- | 
+  -- | Run given MongoDB action against the database
   withDB :: ReaderT Database (Action m) a -> m (Either Failure a)
 
 
+  ----------------------------------------------------------------------------
+  -- | Same as 'withDB' but calls 'error' if there is an exception
+  withDB' :: ReaderT Database (Action m) a -> m a
+  withDB' run = do
+    r <- withDB run 
+    either (error . show) return r
+
+
+
 ------------------------------------------------------------------------------
 -- | Get strict ByteString to work directly with BSON auto-casting
 instance Val B8.ByteString where
@@ -55,3 +64,4 @@ instance Val [Word8] where
     val = val . fmap w2c
     cast' x = fmap (fmap c2w) . cast' $ x
     cast' _ = Nothing
+