convert persistent to hasql

Author: Cmdv

cardano-chain-gen/src/Cardano/Mock/Query.hs

@@ -36,7 +36,7 @@ import Prelude ()
 queryVersionMajorFromEpoch ::
   MonadIO io =>
   Word64 ->
-  ReaderT SqlBackend io (Maybe Word16)
+  DB.DbAction io (Maybe Word16)
 queryVersionMajorFromEpoch epochNo = do
   res <- selectOne $ do
     prop <- from $ table @Db.EpochParam
@@ -48,7 +48,7 @@ queryVersionMajorFromEpoch epochNo = do
 queryParamProposalFromEpoch ::
   MonadIO io =>
   Word64 ->
-  ReaderT SqlBackend io (Maybe Db.ParamProposal)
+  DB.DbAction io (Maybe Db.ParamProposal)
 queryParamProposalFromEpoch epochNo = do
   res <- selectOne $ do
     prop <- from $ table @Db.ParamProposal
@@ -59,7 +59,7 @@ queryParamProposalFromEpoch epochNo = do
 queryParamFromEpoch ::
   MonadIO io =>
   Word64 ->
-  ReaderT SqlBackend io (Maybe Db.EpochParam)
+  DB.DbAction io (Maybe Db.EpochParam)
 queryParamFromEpoch epochNo = do
   res <- selectOne $ do
     param <- from $ table @Db.EpochParam
@@ -68,22 +68,22 @@ queryParamFromEpoch epochNo = do
   pure (entityVal <$> res)
 
 -- | Query whether there any null tx deposits?
-queryNullTxDepositExists :: MonadIO io => ReaderT SqlBackend io Bool
+queryNullTxDepositExists :: MonadIO io => DB.DbAction io Bool
 queryNullTxDepositExists = do
   res <- select $ do
     tx <- from $ table @Db.Tx
     where_ $ isNothing_ (tx ^. Db.TxDeposit)
   pure $ not (null res)
 
-queryMultiAssetCount :: MonadIO io => ReaderT SqlBackend io Word
+queryMultiAssetCount :: MonadIO io => DB.DbAction io Word
 queryMultiAssetCount = do
   res <- select $ do
     _ <- from (table @Db.MultiAsset)
     pure countRows
 
   pure $ maybe 0 unValue (listToMaybe res)
 
-queryTxMetadataCount :: MonadIO io => ReaderT SqlBackend io Word
+queryTxMetadataCount :: MonadIO io => DB.DbAction io Word
 queryTxMetadataCount = do
   res <- selectOne $ do
     _ <- from (table @Db.TxMetadata)
@@ -95,7 +95,7 @@ queryDRepDistrAmount ::
   MonadIO io =>
   ByteString ->
   Word64 ->
-  ReaderT SqlBackend io Word64
+  DB.DbAction io Word64
 queryDRepDistrAmount drepHash epochNo = do
   res <- selectOne $ do
     (distr :& hash) <-
@@ -113,7 +113,7 @@ queryDRepDistrAmount drepHash epochNo = do
 
 queryGovActionCounts ::
   MonadIO io =>
-  ReaderT SqlBackend io (Word, Word, Word, Word)
+  DB.DbAction io (Word, Word, Word, Word)
 queryGovActionCounts = do
   ratified <- countNonNulls Db.GovActionProposalRatifiedEpoch
   enacted <- countNonNulls Db.GovActionProposalEnactedEpoch
@@ -125,7 +125,7 @@ queryGovActionCounts = do
     countNonNulls ::
       (MonadIO io, PersistField field) =>
       EntityField Db.GovActionProposal (Maybe field) ->
-      ReaderT SqlBackend io Word
+      DB.DbAction io Word
     countNonNulls field = do
       res <- selectOne $ do
         e <- from $ table @Db.GovActionProposal
@@ -137,7 +137,7 @@ queryGovActionCounts = do
 queryConstitutionAnchor ::
   MonadIO io =>
   Word64 ->
-  ReaderT SqlBackend io (Maybe (Text, ByteString))
+  DB.DbAction io (Maybe (Text, ByteString))
 queryConstitutionAnchor epochNo = do
   res <- selectOne $ do
     (_ :& anchor :& epochState) <-
@@ -160,7 +160,7 @@ queryConstitutionAnchor epochNo = do
 
 queryRewardRests ::
   MonadIO io =>
-  ReaderT SqlBackend io [(Db.RewardSource, Word64)]
+  DB.DbAction io [(Db.RewardSource, Word64)]
 queryRewardRests = do
   res <- select $ do
     reward <- from $ table @Db.RewardRest
@@ -170,7 +170,7 @@ queryRewardRests = do
 
 queryTreasuryDonations ::
   MonadIO io =>
-  ReaderT SqlBackend io Word64
+  DB.DbAction io Word64
 queryTreasuryDonations = do
   res <- selectOne $ do
     txs <- from $ table @Db.Tx
@@ -183,7 +183,7 @@ queryVoteCounts ::
   MonadIO io =>
   ByteString ->
   Word16 ->
-  ReaderT SqlBackend io (Word64, Word64, Word64)
+  DB.DbAction io (Word64, Word64, Word64)
 queryVoteCounts txHash idx = do
   yes <- countVotes Db.VoteYes
   no <- countVotes Db.VoteNo
@@ -210,7 +210,7 @@ queryVoteCounts txHash idx = do
 queryEpochStateCount ::
   MonadIO io =>
   Word64 ->
-  ReaderT SqlBackend io Word64
+  DB.DbAction io Word64
 queryEpochStateCount epochNo = do
   res <- selectOne $ do
     epochState <- from (table @Db.EpochState)
@@ -222,7 +222,7 @@ queryEpochStateCount epochNo = do
 queryCommitteeByTxHash ::
   MonadIO io =>
   ByteString ->
-  ReaderT SqlBackend io (Maybe Db.Committee)
+  DB.DbAction io (Maybe Db.Committee)
 queryCommitteeByTxHash txHash = do
   res <- selectOne $ do
     (committee :& _ :& tx) <-
@@ -244,7 +244,7 @@ queryCommitteeByTxHash txHash = do
 queryCommitteeMemberCountByTxHash ::
   MonadIO io =>
   Maybe ByteString ->
-  ReaderT SqlBackend io Word64
+  DB.DbAction io Word64
 queryCommitteeMemberCountByTxHash txHash = do
   res <- selectOne $ do
     (_ :& committee :& _ :& tx) <-

cardano-chain-gen/test/Test/Cardano/Db/Mock/Config.hs

@@ -58,7 +58,7 @@ module Test.Cardano.Db.Mock.Config (
   withCustomConfigAndLogs,
   withFullConfig',
   replaceConfigFile,
-  txOutTableTypeFromConfig,
+  txOutVariantTypeFromConfig,
 ) where
 
 import Cardano.Api (NetworkMagic (..))
@@ -226,7 +226,7 @@ withDBSyncEnv mkEnv = bracket mkEnv stopDBSyncIfRunning
 getDBSyncPGPass :: DBSyncEnv -> DB.PGPassSource
 getDBSyncPGPass = enpPGPassSource . dbSyncParams
 
-queryDBSync :: DBSyncEnv -> ReaderT SqlBackend (NoLoggingT IO) a -> IO a
+queryDBSync :: DBSyncEnv -> DB.DbAction (NoLoggingT IO) a -> IO a
 queryDBSync env = DB.runWithConnectionNoLogging (getDBSyncPGPass env)
 
 getPoolLayer :: DBSyncEnv -> IO PoolDataLayer
@@ -601,14 +601,14 @@ replaceConfigFile newFilename dbSync@DBSyncEnv {..} = do
     newParams =
       dbSyncParams {enpConfigFile = ConfigFile $ configDir </> newFilename}
 
-txOutTableTypeFromConfig :: DBSyncEnv -> DB.TxOutTableType
-txOutTableTypeFromConfig dbSyncEnv =
+txOutVariantTypeFromConfig :: DBSyncEnv -> DB.TxOutVariantType
+txOutVariantTypeFromConfig dbSyncEnv =
   case sioTxOut $ dncInsertOptions $ dbSyncConfig dbSyncEnv of
-    TxOutDisable -> DB.TxOutCore
+    TxOutDisable -> DB.TxOutVariantCore
     TxOutEnable useTxOutAddress -> getTxOutTT useTxOutAddress
     TxOutConsumed _ useTxOutAddress -> getTxOutTT useTxOutAddress
     TxOutConsumedPrune _ useTxOutAddress -> getTxOutTT useTxOutAddress
     TxOutConsumedBootstrap _ useTxOutAddress -> getTxOutTT useTxOutAddress
   where
-    getTxOutTT :: UseTxOutAddress -> DB.TxOutTableType
-    getTxOutTT value = if unUseTxOutAddress value then DB.TxOutVariantAddress else DB.TxOutCore
+    getTxOutTT :: UseTxOutAddress -> DB.TxOutVariantType
+    getTxOutTT value = if unUseTxOutAddress value then DB.TxOutVariantAddress else DB.TxOutVariantCore

cardano-chain-gen/test/Test/Cardano/Db/Mock/Unit/Alonzo/Plutus.hs

@@ -29,10 +29,10 @@ module Test.Cardano.Db.Mock.Unit.Alonzo.Plutus (
 ) where
 
 import qualified Cardano.Crypto.Hash as Crypto
-import Cardano.Db (TxOutTableType (..))
+import Cardano.Db (TxOutVariantType (..))
 import qualified Cardano.Db as DB
-import qualified Cardano.Db.Schema.Variant.TxOutAddress as V
-import qualified Cardano.Db.Schema.Variant.TxOutCore as C
+import qualified Cardano.Db.Schema.Variants.TxOutAddress as VA
+import qualified Cardano.Db.Schema.Variants.TxOutCore as VC
 import Cardano.DbSync.Era.Shelley.Generic.Util (renderAddress)
 import Cardano.Ledger.Coin
 import Cardano.Ledger.Mary.Value (MaryValue (..), MultiAsset (..), PolicyID (..))
@@ -97,18 +97,18 @@ simpleScript =
   where
     testLabel = "simpleScript-alonzo"
     getOutFields txOutW = case txOutW of
-      DB.CTxOutW txOut ->
-        ( C.txOutAddress txOut
-        , C.txOutAddressHasScript txOut
-        , C.txOutValue txOut
-        , C.txOutDataHash txOut
+      DB.VCTxOutW txOut ->
+        ( VC.txOutAddress txOut
+        , VC.txOutAddressHasScript txOut
+        , VC.txOutValue txOut
+        , VC.txOutDataHash txOut
         )
-      DB.VTxOutW txout mAddress -> case mAddress of
+      DB.VATxOutW txout mAddress -> case mAddress of
         Just address ->
-          ( V.addressAddress address
-          , V.addressHasScript address
-          , V.txOutValue txout
-          , V.txOutDataHash txout
+          ( VA.addressAddress address
+          , VA.addressHasScript address
+          , VA.txOutValue txout
+          , VA.txOutDataHash txout
           )
         Nothing -> error "AlonzoSimpleScript: expected an address"
     expectedFields =

cardano-chain-gen/test/Test/Cardano/Db/Mock/Unit/Babbage/Plutus.hs

@@ -32,8 +32,8 @@ module Test.Cardano.Db.Mock.Unit.Babbage.Plutus (
 
 import qualified Cardano.Crypto.Hash as Crypto
 import qualified Cardano.Db as DB
-import qualified Cardano.Db.Schema.Variant.TxOutAddress as V
-import qualified Cardano.Db.Schema.Variant.TxOutCore as C
+import qualified Cardano.Db.Schema.Variants.TxOutAddress as VA
+import qualified Cardano.Db.Schema.Variants.TxOutCore as VC
 import Cardano.DbSync.Era.Shelley.Generic.Util (renderAddress)
 import Cardano.Ledger.Coin
 import Cardano.Ledger.Mary.Value (MaryValue (..), MultiAsset (..), PolicyID (..))
@@ -63,7 +63,7 @@ import qualified Data.Map as Map
 import Data.Text (Text)
 import Ouroboros.Consensus.Cardano.Block (StandardBabbage)
 import Ouroboros.Network.Block (genesisPoint)
-import Test.Cardano.Db.Mock.Config (babbageConfigDir, startDBSync, txOutTableTypeFromConfig, withFullConfig, withFullConfigAndDropDB)
+import Test.Cardano.Db.Mock.Config (babbageConfigDir, startDBSync, txOutVariantTypeFromConfig, withFullConfig, withFullConfigAndDropDB)
 import Test.Cardano.Db.Mock.UnifiedApi (
   fillUntilNextEpoch,
   forgeNextAndSubmit,
@@ -91,7 +91,7 @@ simpleScript =
   withFullConfigAndDropDB babbageConfigDir testLabel $ \interpreter mockServer dbSync -> do
     startDBSync dbSync
 
-    let txOutTableType = txOutTableTypeFromConfig dbSync
+    let txOutVariantType = txOutVariantTypeFromConfig dbSync
     void $ registerAllStakeCreds interpreter mockServer
 
     a <- fillUntilNextEpoch interpreter mockServer
@@ -101,23 +101,23 @@ simpleScript =
         Babbage.mkLockByScriptTx (UTxOIndex 0) [Babbage.TxOutNoInline True] 20000 20000
 
     assertBlockNoBackoff dbSync (fromIntegral $ length a + 2)
-    assertEqQuery dbSync (fmap getOutFields <$> DB.queryScriptOutputs txOutTableType) [expectedFields] "Unexpected script outputs"
+    assertEqQuery dbSync (fmap getOutFields <$> DB.queryScriptOutputs txOutVariantType) [expectedFields] "Unexpected script outputs"
   where
     testLabel = "simpleScript"
     getOutFields txOutW =
       case txOutW of
-        DB.CTxOutW txOut ->
-          ( C.txOutAddress txOut
-          , C.txOutAddressHasScript txOut
-          , C.txOutValue txOut
-          , C.txOutDataHash txOut
+        DB.VCTxOutW txOut ->
+          ( VC.txOutAddress txOut
+          , VC.txOutAddressHasScript txOut
+          , VC.txOutValue txOut
+          , VC.txOutDataHash txOut
           )
-        DB.VTxOutW txOut mAddress -> case mAddress of
+        DB.VATxOutW txOut mAddress -> case mAddress of
           Just address ->
-            ( V.addressAddress address
-            , V.addressHasScript address
-            , V.txOutValue txOut
-            , V.txOutDataHash txOut
+            ( VA.addressAddress address
+            , VA.addressHasScript address
+            , VA.txOutValue txOut
+            , VA.txOutDataHash txOut
             )
           Nothing -> error "BabbageSimpleScript: expected an address"