wrapConnection

meteficha · Mar 3, 2013 · 364af6c · 364af6c
1 parent a9334c4
commit 364af6c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/persistent-sqlite/Database/Persist/Sqlite.hs b/persistent-sqlite/Database/Persist/Sqlite.hs
@@ -12,6 +12,7 @@ module Database.Persist.Sqlite
     , module Database.Persist.GenericSql
     , SqliteConf (..)
     , runSqlite
+    , wrapConnection
     ) where
 
 import Database.Persist hiding (Entity (..))
@@ -37,6 +38,7 @@ import Data.Conduit
 import qualified Data.Conduit.List as CL
 import Control.Applicative
 import Data.Int (Int64)
+import Control.Monad ((>=>))
 
 createSqlitePool :: MonadIO m => Text -> Int -> m ConnectionPool
 createSqlitePool s = createSqlPool $ open' s
@@ -52,8 +54,13 @@ withSqliteConn :: (MonadBaseControl IO m, MonadIO m)
 withSqliteConn = withSqlConn . open'
 
 open' :: Text -> IO Connection
-open' s = do
-    conn <- Sqlite.open s
+open' = Sqlite.open >=> wrapConnection
+
+-- | Wrap up a raw 'Sqlite.Connection' as a Persistent SQL 'Connection'.
+--
+-- Since 1.1.5
+wrapConnection :: Sqlite.Connection -> IO Connection
+wrapConnection conn = do
     smap <- newIORef $ Map.empty
     return Connection
         { prepare = prepare' conn

diff --git a/persistent-sqlite/persistent-sqlite.cabal b/persistent-sqlite/persistent-sqlite.cabal
@@ -1,5 +1,5 @@
 name:            persistent-sqlite
-version:         1.1.4.1
+version:         1.1.5
 license:         MIT
 license-file:    LICENSE
 author:          Michael Snoyman <michael@snoyman.com>