db: Add an epochNo column to StakeDeregistration table

IntersectMBO · Jul 29, 2021 · f4bf5bf · f4bf5bf
1 parent 960a546
commit f4bf5bf
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/cardano-db-sync/src/Cardano/DbSync/Era/Shelley/Insert.hs b/cardano-db-sync/src/Cardano/DbSync/Era/Shelley/Insert.hs
@@ -309,7 +309,7 @@ insertDelegCert
 insertDelegCert tracer network txId idx epochNo slotNo dCert =
   case dCert of
     Shelley.RegKey cred -> insertStakeRegistration tracer txId idx $ Generic.annotateStakingCred network cred
-    Shelley.DeRegKey cred -> insertStakeDeregistration tracer network txId idx cred
+    Shelley.DeRegKey cred -> insertStakeDeregistration tracer network epochNo txId idx cred
     Shelley.Delegate (Shelley.Delegation cred poolkh) -> insertDelegation tracer network txId idx epochNo slotNo cred poolkh
 
 insertPoolRegister
@@ -476,14 +476,15 @@ insertStakeRegistration _tracer txId idx rewardAccount = do
 
 insertStakeDeregistration
     :: (MonadBaseControl IO m, MonadIO m)
-    => Trace IO Text -> Ledger.Network -> DB.TxId -> Word16 -> Ledger.StakeCredential StandardCrypto
+    => Trace IO Text -> Ledger.Network -> EpochNo -> DB.TxId -> Word16 -> Ledger.StakeCredential StandardCrypto
     -> ExceptT SyncNodeError (ReaderT SqlBackend m) ()
-insertStakeDeregistration _tracer network txId idx cred = do
+insertStakeDeregistration _tracer network (EpochNo epochNo) txId idx cred = do
   scId <- liftLookupFail "insertStakeDeregistration" $ queryStakeAddress (Generic.stakingCredHash network cred)
   void . lift . DB.insertStakeDeregistration $
     DB.StakeDeregistration
       { DB.stakeDeregistrationAddrId = scId
       , DB.stakeDeregistrationCertIndex = idx
+      , DB.stakeDeregistrationEpochNo = epochNo
       , DB.stakeDeregistrationTxId = txId
       }
 

diff --git a/cardano-db/src/Cardano/Db/Schema.hs b/cardano-db/src/Cardano/Db/Schema.hs
@@ -254,6 +254,7 @@ share
   StakeDeregistration
     addrId              StakeAddressId      OnDeleteCascade
     certIndex           Word16
+    epochNo             Word64              sqltype=uinteger
     txId                TxId                OnDeleteCascade
     UniqueStakeDeregistration addrId txId
 

diff --git a/schema/migration-2-0014-20210729.sql b/schema/migration-2-0014-20210729.sql
@@ -0,0 +1,19 @@
+-- Persistent generated migration.
+
+CREATE FUNCTION migrate() RETURNS void AS $$
+DECLARE
+  next_version int ;
+BEGIN
+  SELECT stage_two + 1 INTO next_version FROM schema_version ;
+  IF next_version = 14 THEN
+    EXECUTE 'ALTER TABLE "stake_deregistration" ADD COLUMN "epoch_no" uinteger NOT NULL' ;
+    -- Hand written SQL statements can be added here.
+    UPDATE schema_version SET stage_two = next_version ;
+    RAISE NOTICE 'DB has been migrated to stage_two version %', next_version ;
+  END IF ;
+END ;
+$$ LANGUAGE plpgsql ;
+
+SELECT migrate() ;
+
+DROP FUNCTION migrate() ;