Update MySQL and Sqlite libraries to use new StatementCache mechanism

yesodweb · Mar 31, 2021 · 0926409 · 0926409
1 parent d340716
commit 0926409
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/persistent-mysql/Database/Persist/MySQL.hs b/persistent-mysql/Database/Persist/MySQL.hs
@@ -126,7 +126,7 @@ open' :: MySQL.ConnectInfo -> LogFunc -> IO SqlBackend
 open' ci logFunc = do
     conn <- MySQL.connect ci
     MySQLBase.autocommit conn False -- disable autocommit!
-    smap <- newIORef $ Map.empty
+    smap <- makeSimpleStatementCache
     return $ SqlBackend
         { connPrepare    = prepare' conn
         , connStmtMap    = smap

diff --git a/persistent-sqlite/Database/Persist/Sqlite.hs b/persistent-sqlite/Database/Persist/Sqlite.hs
@@ -266,7 +266,7 @@ wrapConnectionInfo connInfo conn logFunc = do
         Sqlite.reset conn stmt
         Sqlite.finalize stmt
 
-    smap <- newIORef $ Map.empty
+    smap <- makeSimpleStatementCache
     return $ SqlBackend
         { connPrepare = prepare' conn
         , connStmtMap = smap
@@ -455,7 +455,7 @@ migrate' allDefs getter val = do
 -- with the difference that an actual database isn't needed for it.
 mockMigration :: Migration -> IO ()
 mockMigration mig = do
-  smap <- newIORef $ Map.empty
+  smap <- makeSimpleStatementCache
   let sqlbackend = SqlBackend
                    { connPrepare = \_ -> do
                                      return Statement