From 8aceaef19f1b62ed77bf659adf3d76e0346e8266 Mon Sep 17 00:00:00 2001
From: Kostas Dermentzis <kostas.dermentzis@iohk.io>
Date: Wed, 20 Nov 2024 12:27:08 +0200
Subject: [PATCH] Don't try to set null consumed when it's disabled

Fixes https://github.com/IntersectMBO/cardano-db-sync/issues/1902
---
 .../src/Cardano/DbSync/Rollback.hs            |  4 +-
 .../src/Cardano/Db/Operations/Delete.hs       | 63 ++++++++++---------
 2 files changed, 35 insertions(+), 32 deletions(-)

diff --git a/cardano-db-sync/src/Cardano/DbSync/Rollback.hs b/cardano-db-sync/src/Cardano/DbSync/Rollback.hs
index 1ba1d13f8..9124bae6d 100644
--- a/cardano-db-sync/src/Cardano/DbSync/Rollback.hs
+++ b/cardano-db-sync/src/Cardano/DbSync/Rollback.hs
@@ -48,7 +48,7 @@ rollbackFromBlockNo syncEnv blkNo = do
         , textShow blkNo
         ]
     lift $ do
-      deletedBlockCount <- DB.deleteBlocksBlockId trce txOutTableType blockId epochNo (Just (DB.pcmConsumedTxOut $ getPruneConsume syncEnv))
+      deletedBlockCount <- DB.deleteBlocksBlockId trce txOutTableType blockId epochNo (DB.pcmConsumedTxOut $ getPruneConsume syncEnv)
       when (deletedBlockCount > 0) $ do
         -- We use custom constraints to improve input speeds when syncing.
         -- If they don't already exists we add them here as once a rollback has happened
@@ -111,4 +111,4 @@ prepareRollback syncEnv point serverTip =
 unsafeRollback :: Trace IO Text -> DB.TxOutTableType -> DB.PGConfig -> SlotNo -> IO (Either SyncNodeError ())
 unsafeRollback trce txOutTableType config slotNo = do
   logWarning trce $ "Starting a forced rollback to slot: " <> textShow (unSlotNo slotNo)
-  Right <$> DB.runDbNoLogging (DB.PGPassCached config) (void $ DB.deleteBlocksSlotNo trce txOutTableType slotNo Nothing)
+  Right <$> DB.runDbNoLogging (DB.PGPassCached config) (void $ DB.deleteBlocksSlotNo trce txOutTableType slotNo True)
diff --git a/cardano-db/src/Cardano/Db/Operations/Delete.hs b/cardano-db/src/Cardano/Db/Operations/Delete.hs
index ebcf84966..e84c71cec 100644
--- a/cardano-db/src/Cardano/Db/Operations/Delete.hs
+++ b/cardano-db/src/Cardano/Db/Operations/Delete.hs
@@ -9,13 +9,13 @@
 {-# LANGUAGE TypeOperators #-}
 
 module Cardano.Db.Operations.Delete (
-  deleteBlocksSlotNo,
-  deleteBlocksSlotNoNoTrace,
   deleteDelistedPool,
   deleteBlocksBlockId,
+  queryDelete,
+  deleteBlocksSlotNo,
+  deleteBlocksSlotNoNoTrace,
   deleteBlocksForTests,
   deleteBlock,
-  queryDelete,
 ) where
 
 import Cardano.BM.Trace (Trace, logInfo, logWarning, nullTracer)
@@ -54,9 +54,6 @@ import Database.Persist (
  )
 import Database.Persist.Sql (Filter, SqlBackend, delete, deleteWhere, deleteWhereCount, selectKeysList)
 
-deleteBlocksSlotNoNoTrace :: MonadIO m => TxOutTableType -> SlotNo -> ReaderT SqlBackend m Bool
-deleteBlocksSlotNoNoTrace txOutTableType slotNo = deleteBlocksSlotNo nullTracer txOutTableType slotNo Nothing
-
 -- | Delete a block if it exists. Returns 'True' if it did exist and has been
 -- deleted and 'False' if it did not exist.
 deleteBlocksSlotNo ::
@@ -64,22 +61,18 @@ deleteBlocksSlotNo ::
   Trace IO Text ->
   TxOutTableType ->
   SlotNo ->
-  Maybe Bool ->
+  Bool ->
   ReaderT SqlBackend m Bool
-deleteBlocksSlotNo trce txOutTableType (SlotNo slotNo) mIsConsumedTxOut = do
+deleteBlocksSlotNo trce txOutTableType (SlotNo slotNo) isConsumedTxOut = do
   mBlockId <- queryNearestBlockSlotNo slotNo
   case mBlockId of
     Nothing -> do
       liftIO $ logWarning trce $ "deleteBlocksSlotNo: No block contains the the slot: " <> pack (show slotNo)
       pure False
     Just (blockId, epochN) -> do
-      void $ deleteBlocksBlockId trce txOutTableType blockId epochN mIsConsumedTxOut
+      void $ deleteBlocksBlockId trce txOutTableType blockId epochN isConsumedTxOut
       pure True
 
-deleteBlocksForTests :: MonadIO m => TxOutTableType -> BlockId -> Word64 -> ReaderT SqlBackend m ()
-deleteBlocksForTests txOutTableType blockId epochN = do
-  void $ deleteBlocksBlockId nullTracer txOutTableType blockId epochN Nothing
-
 -- | Delete starting from a 'BlockId'.
 deleteBlocksBlockId ::
   MonadIO m =>
@@ -89,9 +82,9 @@ deleteBlocksBlockId ::
   -- | The 'EpochNo' of the block to delete.
   Word64 ->
   -- | Is ConsumeTxout
-  Maybe Bool ->
+  Bool ->
   ReaderT SqlBackend m Int64
-deleteBlocksBlockId trce txOutTableType blockId epochN mIsConsumedTxOut = do
+deleteBlocksBlockId trce txOutTableType blockId epochN isConsumedTxOut = do
   mMinIds <- fmap (textToMinIds txOutTableType =<<) <$> queryReverseIndexBlockId blockId
   (cminIds, completed) <- findMinIdsRec mMinIds mempty
   mTxId <- queryMinRefId TxBlockId blockId
@@ -99,10 +92,9 @@ deleteBlocksBlockId trce txOutTableType blockId epochN mIsConsumedTxOut = do
   deleteEpochLogs <- deleteUsingEpochNo epochN
   (deleteBlockCount, blockDeleteLogs) <- deleteTablesAfterBlockId txOutTableType blockId mTxId minIds
   setNullLogs <-
-    maybe
-      (pure ("ConsumedTxOut is not active so no Nulls set", 0))
-      (\_ -> querySetNullTxOut txOutTableType mTxId)
-      mIsConsumedTxOut
+    if isConsumedTxOut
+      then querySetNullTxOut txOutTableType mTxId
+      else pure ("ConsumedTxOut is not active so no Nulls set", 0)
   -- log all the deleted rows in the rollback
   liftIO $ logInfo trce $ mkRollbackSummary (deleteEpochLogs <> blockDeleteLogs) setNullLogs
   pure deleteBlockCount
@@ -357,17 +349,6 @@ deleteDelistedPool poolHash = do
   mapM_ delete keys
   pure $ not (null keys)
 
--- | Delete a block if it exists. Returns 'True' if it did exist and has been
--- deleted and 'False' if it did not exist.
-deleteBlock :: MonadIO m => TxOutTableType -> Block -> ReaderT SqlBackend m Bool
-deleteBlock txOutTableType block = do
-  mBlockId <- queryBlockHash block
-  case mBlockId of
-    Nothing -> pure False
-    Just (blockId, epochN) -> do
-      void $ deleteBlocksBlockId nullTracer txOutTableType blockId epochN Nothing
-      pure True
-
 mkRollbackSummary :: [(Text, Int64)] -> (Text, Int64) -> Text
 mkRollbackSummary logs setNullLogs =
   "\n----------------------- Rollback Summary: ----------------------- \n"
@@ -392,3 +373,25 @@ mkRollbackSummary logs setNullLogs =
         <> if nullCount == 0
           then nullMessage
           else "\n\nSet Null: " <> nullMessage <> " - Count: " <> pack (show nullCount)
+
+-- Tools
+
+deleteBlocksSlotNoNoTrace :: MonadIO m => TxOutTableType -> SlotNo -> ReaderT SqlBackend m Bool
+deleteBlocksSlotNoNoTrace txOutTableType slotNo = deleteBlocksSlotNo nullTracer txOutTableType slotNo True
+
+-- Tests
+
+deleteBlocksForTests :: MonadIO m => TxOutTableType -> BlockId -> Word64 -> ReaderT SqlBackend m ()
+deleteBlocksForTests txOutTableType blockId epochN = do
+  void $ deleteBlocksBlockId nullTracer txOutTableType blockId epochN False
+
+-- | Delete a block if it exists. Returns 'True' if it did exist and has been
+-- deleted and 'False' if it did not exist.
+deleteBlock :: MonadIO m => TxOutTableType -> Block -> ReaderT SqlBackend m Bool
+deleteBlock txOutTableType block = do
+  mBlockId <- queryBlockHash block
+  case mBlockId of
+    Nothing -> pure False
+    Just (blockId, epochN) -> do
+      void $ deleteBlocksBlockId nullTracer txOutTableType blockId epochN False
+      pure True