Add the resetting of the registry in case of errors

library/Hasql/PreparedStatementRegistry.hs

@@ -2,14 +2,15 @@ module Hasql.PreparedStatementRegistry
   ( PreparedStatementRegistry,
     new,
     update,
+    reset,
     LocalKey (..),
   )
 where
 
 import ByteString.StrictBuilder qualified as B
 import Data.HashTable.IO qualified as A
 import Hasql.LibPq14 qualified as Pq
-import Hasql.Prelude hiding (lookup)
+import Hasql.Prelude hiding (lookup, reset)
 
 data PreparedStatementRegistry
   = PreparedStatementRegistry !(A.BasicHashTable LocalKey ByteString) !(IORef Word)
@@ -42,6 +43,15 @@ update localKey onNewRemoteKey onOldRemoteKey (PreparedStatementRegistry table c
     old =
       onOldRemoteKey
 
+reset :: PreparedStatementRegistry -> IO ()
+reset (PreparedStatementRegistry table counter) = do
+  -- TODO: This is a temporary measure.
+  -- We should just move to a pure implementation.
+  do
+    entries <- A.toList table
+    forM_ entries \(k, _) -> A.delete table k
+  writeIORef counter 0
+
 -- |
 -- Local statement key.
 data LocalKey

library/Hasql/Session/Core.hs

@@ -11,6 +11,7 @@ import Hasql.IO qualified as IO
 import Hasql.LibPq14 qualified as Pq
 import Hasql.Pipeline.Core qualified as Pipeline
 import Hasql.Prelude
+import Hasql.PreparedStatementRegistry qualified as PreparedStatementRegistry
 import Hasql.Statement qualified as Statement
 
 -- |
@@ -29,11 +30,13 @@ run (Session impl) connection =
       runExceptT $ runReaderT impl connection
     handler =
       case connection of
-        Connection.Connection pqConnVar _ _ ->
+        Connection.Connection pqConnVar _ registry ->
           withMVar pqConnVar \pqConn ->
             Pq.transactionStatus pqConn >>= \case
               Pq.TransIdle -> pure ()
-              _ -> Pq.reset pqConn
+              _ -> do
+                PreparedStatementRegistry.reset registry
+                Pq.reset pqConn
 
 -- |
 -- Possibly a multi-statement query,