insert into cache on reads too

Change-Id: Id00000007f84d7de38391d240b94068587588764
kadena-io · Jan 23, 2025 · c158a15 · c158a15
1 parent 959d3d3
commit c158a15
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/Chainweb/Pact5/Backend/ChainwebPactDb.hs b/src/Chainweb/Pact5/Backend/ChainwebPactDb.hs
@@ -369,9 +369,14 @@ doReadRow mlim d k = do
                 Nothing -> return Nothing
                 Just (encodedValue, decodedValue) -> do
                     case d of
-                        Pact.DModules ->
+                        Pact.DModules -> do
                             BlockHandler $ lift $ lift
                                 $ Pact.chargeGasM (Pact.GModuleOp (Pact.MOpLoadModule (BS.length encodedValue)))
+                            bsPendingTx . _Just . _1 . pendingWrites %=
+                                InMemDb.insert d k (InMemDb.ReadEntry encodedValue decodedValue)
+                        Pact.DUserTables {} -> do
+                            bsPendingTx . _Just . _1 . pendingWrites %=
+                                InMemDb.insert d k (InMemDb.ReadEntry encodedValue decodedValue)
                         _ -> return ()
                     return (Just decodedValue)
   where

diff --git a/test/unit/Chainweb/Test/Pact5/TransactionExecTest.hs b/test/unit/Chainweb/Test/Pact5/TransactionExecTest.hs
@@ -476,6 +476,12 @@ applyCmdSpec rdb = readFromAfterGenesis v rdb $ do
                                     P.match InMemDb._WriteEntry P.succeed
                                 ]
                         ]
+
+                , P.fun InMemDb.modules
+                    ? P.alignExact ? Map.fromList
+                        [ ModuleName "coin" Nothing :=>
+                            P.match InMemDb._ReadEntry P.succeed
+                        ]
                 ]
 
 quirkSpec :: RocksDb -> IO ()