Fix an issue where sequential scans were used

Author: kderme

cardano-db-sync/src/Cardano/DbSync.hs

@@ -28,7 +28,7 @@ import qualified Cardano.Crypto as Crypto
 import Cardano.Db (textShow)
 import qualified Cardano.Db as Db
 import Cardano.DbSync.Api
-import Cardano.DbSync.Api.Types (InsertOptions (..), RunMigration, SyncOptions (..), envLedgerEnv)
+import Cardano.DbSync.Api.Types (InsertOptions (..), RunMigration, SyncOptions (..), SyncEnv (..), envLedgerEnv)
 import Cardano.DbSync.Config (configureLogging, readSyncNodeConfig)
 import Cardano.DbSync.Config.Cardano
 import Cardano.DbSync.Config.Types (
@@ -51,6 +51,7 @@ import Cardano.DbSync.Rollback (unsafeRollback)
 import Cardano.DbSync.Sync (runSyncNodeClient)
 import Cardano.DbSync.Tracing.ToObjectOrphans ()
 import Cardano.DbSync.Types
+import Cardano.DbSync.Util.Constraint (addRewardConstraintsIfNotExist)
 import Cardano.Prelude hiding (Nat, (%))
 import Cardano.Slotting.Slot (EpochNo (..))
 import Control.Concurrent.Async
@@ -178,6 +179,7 @@ runSyncNode metricsSetters trce iomgr dbConnString ranMigrations runMigrationFnc
                 syncNodeParams
                 ranMigrations
                 runMigrationFnc
+          liftIO $ runReaderT (addRewardConstraintsIfNotExist syncEnv trce) (envBackend syncEnv)
           liftIO $ runExtraMigrationsMaybe syncEnv
           unless (enpShouldUseLedger syncNodeParams) $ liftIO $ do
             logInfo trce "Migrating to a no ledger schema"

cardano-db-sync/src/Cardano/DbSync/Util/Constraint.hs

@@ -9,6 +9,8 @@ module Cardano.DbSync.Util.Constraint (
   constraintNameReward,
   dbConstraintNamesExists,
   addConstraintsIfNotExist,
+  addStakeConstraintsIfNotExist,
+  addRewardConstraintsIfNotExist,
   addRewardTableConstraint,
   addEpochStakeTableConstraint,
 ) where
@@ -58,12 +60,34 @@ addConstraintsIfNotExist ::
   Trace IO Text ->
   ReaderT SqlBackend m ()
 addConstraintsIfNotExist syncEnv trce = do
-  ManualDbConstraints {..} <- liftIO . readTVarIO $ envDbConstraints syncEnv
-  unless dbConstraintRewards (addRewardTableConstraint trce)
-  unless dbConstraintEpochStake (addEpochStakeTableConstraint trce)
+  addStakeConstraintsIfNotExist syncEnv trce
+  addRewardConstraintsIfNotExist syncEnv trce
+
+addStakeConstraintsIfNotExist ::
+  forall m.
+  (MonadBaseControl IO m, MonadIO m) =>
+  SyncEnv ->
+  Trace IO Text ->
+  ReaderT SqlBackend m ()
+addStakeConstraintsIfNotExist syncEnv trce = do
+  mdbc <- liftIO . readTVarIO $ envDbConstraints syncEnv
+  unless (dbConstraintEpochStake mdbc) (addEpochStakeTableConstraint trce)
+  liftIO
+    . atomically
+    $ writeTVar (envDbConstraints syncEnv) (mdbc { dbConstraintEpochStake = True})
+
+addRewardConstraintsIfNotExist ::
+  forall m.
+  (MonadBaseControl IO m, MonadIO m) =>
+  SyncEnv ->
+  Trace IO Text ->
+  ReaderT SqlBackend m ()
+addRewardConstraintsIfNotExist syncEnv trce = do
+  mdbc <- liftIO . readTVarIO $ envDbConstraints syncEnv
+  unless (dbConstraintRewards mdbc) (addRewardTableConstraint trce)
   liftIO
     . atomically
-    $ writeTVar (envDbConstraints syncEnv) (DB.ManualDbConstraints True True)
+    $ writeTVar (envDbConstraints syncEnv) (mdbc {dbConstraintRewards = True})
 
 addRewardTableConstraint ::
   forall m.

cardano-db/src/Cardano/Db/Insert.hs

@@ -203,7 +203,7 @@ insertManyDrepDistr ::
   (MonadBaseControl IO m, MonadIO m) =>
   [DrepDistr] ->
   ReaderT SqlBackend m ()
-insertManyDrepDistr = insertManyUncheckedUnique "Many DrepDistr"
+insertManyDrepDistr = insertManyCheckUnique "Many DrepDistr"
 
 insertManyTxIn :: (MonadBaseControl IO m, MonadIO m) => [TxIn] -> ReaderT SqlBackend m [TxInId]
 insertManyTxIn = insertMany' "Many TxIn"
@@ -341,7 +341,7 @@ insertExtraMigration token = void . insert $ ExtraMigrations (textShow token) (J
 
 insertEpochStakeProgress :: (MonadBaseControl IO m, MonadIO m) => [EpochStakeProgress] -> ReaderT SqlBackend m ()
 insertEpochStakeProgress =
-  insertManyUncheckedUnique "Many EpochStakeProgress"
+  insertManyCheckUnique "Many EpochStakeProgress"
 
 updateSetComplete :: MonadIO m => Word64 -> ReaderT SqlBackend m ()
 updateSetComplete epoch = do
@@ -527,7 +527,7 @@ insertManyWithManualUnique ::
   ReaderT SqlBackend m ()
 insertManyWithManualUnique = insertManyUnique
 
-insertManyUncheckedUnique ::
+insertManyCheckUnique ::
   forall m record.
   ( MonadBaseControl IO m
   , MonadIO m
@@ -536,7 +536,7 @@ insertManyUncheckedUnique ::
   String ->
   [record] ->
   ReaderT SqlBackend m ()
-insertManyUncheckedUnique vtype records = do
+insertManyCheckUnique vtype records = do
   let constraintName = uniqueDBName $ onlyOneUniqueDef (Proxy @record)
   insertManyUnique vtype True constraintName records