Skip to content

Commit

Permalink
tx-generator: Use distinct keys for benchmarking phases
Browse files Browse the repository at this point in the history
  • Loading branch information
@MarcFontaine
MarcFontaine committed Sep 27, 2022
1 parent 99d2eaf commit ddf375c
Showing 1 changed file with 67 additions and 15 deletions.
82 changes: 67 additions & 15 deletions bench/tx-generator/src/Cardano/Benchmarking/Compiler.hs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,17 +9,19 @@ import Control.Applicative (liftA2)
import Control.Monad
import Control.Monad.Trans.Except
import Control.Monad.Trans.RWS.CPS

import Data.Aeson
import Data.ByteString.Lazy as BSL (ByteString)
import Data.Dependent.Sum ( (==>) )
import Data.DList (DList)
import qualified Data.DList as DL
import Data.Text (Text)
import qualified Data.Text as Text

import Cardano.Api

import Cardano.Benchmarking.NixOptions
import Cardano.Benchmarking.Script.Setters
import Cardano.Benchmarking.Script.Store (Name(..), WalletName)
import Cardano.Benchmarking.Script.Store (KeyName, Name(..), WalletName)
import Cardano.Benchmarking.Script.Types

data CompileError where
Expand Down Expand Up @@ -53,12 +55,17 @@ compileToScript = do
genesisWallet <- importGenesisFunds
collateralWallet <- addCollaterals genesisWallet
splitWallet <- splittingPhase genesisWallet
benchmarkingPhase splitWallet collateralWallet
void $ benchmarkingPhase splitWallet collateralWallet

initConstants :: Compiler ()
initConstants = do
setN TLocalSocket _nix_localNodeSocketPath
setConst TTTL 1000000
emit $ DefineSigningKey keyNameTxGenFunds keyTxGenFunds
emit $ DefineSigningKey keyNameCollaterals keyCollaterals
emit $ DefineSigningKey keyNameSplitPhase keySplitPhase
emit $ DefineSigningKey keyNameBenchmarkInputs keyBenchmarkInputs
emit $ DefineSigningKey keyNameBenchmarkDone keyBenchmarkDone
where
setConst :: Tag v -> v -> Compiler ()
setConst key val = emit $ Set $ key ==> val
Expand All @@ -72,8 +79,8 @@ importGenesisFunds = do
wallet <- newWallet "genesis_wallet"
era <- askNixOption _nix_era
fee <- askNixOption _nix_tx_fee
cmd1 (ReadSigningKey $ KeyName "pass-partout") _nix_sigKey
emit $ Submit era LocalSocket $ SecureGenesis fee wallet (KeyName "pass-partout") (KeyName "pass-partout")
cmd1 (ReadSigningKey keyNameGenesisInputFund) _nix_sigKey
emit $ Submit era LocalSocket $ SecureGenesis fee wallet keyNameGenesisInputFund keyNameTxGenFunds
delay
logMsg "Importing Genesis Fund. Done."
return wallet
Expand All @@ -89,8 +96,8 @@ addCollaterals src = do
collateralWallet <- newWallet "collateral_wallet"
fee <- askNixOption _nix_tx_fee
let generator = Split fee src
(PayToAddr (KeyName "pass-partout") collateralWallet)
(PayToAddr (KeyName "pass-partout") src)
(PayToAddr keyNameCollaterals collateralWallet)
(PayToAddr keyNameTxGenFunds src)
[ safeCollateral ]
emit $ Submit era LocalSocket generator
logMsg "Create collaterals. Done."
Expand All @@ -106,17 +113,18 @@ splittingPhase srcWallet = do
finalDest <- newWallet "final_split_wallet"
splitSteps <- splitSequenceWalletNames srcWallet finalDest $ unfoldSplitSequence tx_fee minValuePerInput (tx_count * inputs_per_tx)
isPlutus <- isAnyPlutusMode
forM_ (init splitSteps) $ createChange False era
createChange isPlutus era $ last splitSteps
forM_ (init splitSteps) $ createChange False False era
createChange True isPlutus era $ last splitSteps
return finalDest
where
createChange :: Bool -> AnyCardanoEra -> (SrcWallet, DstWallet, Split) -> Compiler ()
createChange isPlutus era (src, dst, split) = do
createChange :: Bool -> Bool -> AnyCardanoEra -> (SrcWallet, DstWallet, Split) -> Compiler ()
createChange isLastStep isPlutus era (src, dst, split) = do
logMsg $ Text.pack $ "Splitting step: " ++ show split
tx_fee <- askNixOption _nix_tx_fee
payMode <- if isPlutus then plutusPayMode dst else return $ PayToAddr (KeyName "pass-partout") dst
let valuePayMode = PayToAddr (if isLastStep then keyNameSplitPhase else keyNameBenchmarkInputs) dst
payMode <- if isPlutus then plutusPayMode dst else return valuePayMode
let generator = case split of
SplitWithChange lovelace count -> Split tx_fee src payMode (PayToAddr (KeyName "pass-partout") src) $ replicate count lovelace
SplitWithChange lovelace count -> Split tx_fee src payMode (PayToAddr keyNameTxGenFunds src) $ replicate count lovelace
FullSplits txCount -> Take txCount $ Cycle $ SplitN tx_fee src payMode maxOutputsPerTx
emit $ Submit era LocalSocket generator
delay
Expand Down Expand Up @@ -167,7 +175,7 @@ unfoldSplitSequence fee value outputs
(x, 0) -> x
(x, _rest) -> x+1

benchmarkingPhase :: WalletName -> Maybe WalletName -> Compiler ()
benchmarkingPhase :: WalletName -> Maybe WalletName -> Compiler WalletName
benchmarkingPhase wallet collateralWallet = do
debugMode <- askNixOption _nix_debugMode
targetNodes <- askNixOption _nix_targetNodes
Expand All @@ -178,15 +186,17 @@ benchmarkingPhase wallet collateralWallet = do
inputs <- askNixOption _nix_inputs_per_tx
outputs <- askNixOption _nix_outputs_per_tx
metadataSize <- askNixOption _nix_add_tx_size
doneWallet <- newWallet "done_wallet"
let
payMode = PayToAddr (KeyName "pass-partout") wallet --todo: used different wallet here !
payMode = PayToAddr keyNameBenchmarkDone doneWallet
submitMode = if debugMode
then LocalSocket
else Benchmark targetNodes (ThreadName "tx-submit-benchmark") tps txCount
generator = Take txCount $ Cycle $ NtoM fee wallet payMode inputs outputs (Just metadataSize) collateralWallet
emit $ Submit era submitMode generator
unless debugMode $ do
emit $ WaitBenchmark $ ThreadName "tx-submit-benchmark"
return doneWallet

data Fees = Fees {
_safeCollateral :: Lovelace
Expand Down Expand Up @@ -250,3 +260,45 @@ newWallet n = do
name <- WalletName <$> newIdentifier n
emit $ InitWallet name
return name

parseKey :: BSL.ByteString -> TextEnvelope
parseKey x = case decode x of
Nothing -> error $ "parsing of key failed : " ++ show x
Just k -> k

keyNameGenesisInputFund :: KeyName
keyNameGenesisInputFund = KeyName "GenesisInputFund"

keyNameTxGenFunds :: KeyName
keyNameTxGenFunds = KeyName "TxGenFunds"

-- addr_test1vzd3muund27y5nw83vymqj3a83pcuzkkejej6s75e5lfjcc85nc3p
keyTxGenFunds :: TextEnvelope
keyTxGenFunds = parseKey "{\n \"type\": \"PaymentSigningKeyShelley_ed25519\",\n \"description\": \"Payment Signing Key\",\n \"cborHex\": \"5820617f846fc8b0e753bd51790de5f5a916de500175c6f5a0e27dde9da7879e1d35\"\n}\n"

keyNameSplitPhase :: KeyName
keyNameSplitPhase = KeyName "SplitPhase"

-- addr_test1vz45dtkyzk6s3245qw8hmaddaatcx8td3pvmntl8ty7q99c22eahm
keySplitPhase :: TextEnvelope
keySplitPhase = parseKey "{\n \"type\": \"PaymentSigningKeyShelley_ed25519\",\n \"description\": \"Payment Signing Key\",\n \"cborHex\": \"5820cf0083c2a5d4c90ab255bc8e68f407d52eebd9408de60a0b9e4c468f9714f076\"\n}\n"

-- addr_test1vzj7zv9msmdasvy5nc9jhnn2gqvrvu33v5rlg332zdfrkugklxkau
keyNameBenchmarkInputs :: KeyName
keyNameBenchmarkInputs = KeyName "BenchmarkInputs"

keyBenchmarkInputs :: TextEnvelope
keyBenchmarkInputs = parseKey "{\n \"type\": \"PaymentSigningKeyShelley_ed25519\",\n \"description\": \"Payment Signing Key\",\n \"cborHex\": \"58205b7f272602661d4ad3d9a4081f25fdcdcdf64fdc4892107de50e50937b77ea42\"\n}\n"

keyNameBenchmarkDone :: KeyName
keyNameBenchmarkDone = KeyName "BenchmarkingDone"

-- addr_test1vz4qz2ayucp7xvnthrx93uhha7e04gvxttpnuq4e6mx2n5gzfw23z
keyBenchmarkDone :: TextEnvelope
keyBenchmarkDone = parseKey "{\n \"type\": \"PaymentSigningKeyShelley_ed25519\",\n \"description\": \"Payment Signing Key\",\n \"cborHex\": \"582016ca4f13fa17557e56a7d0dd3397d747db8e1e22fdb5b9df638abdb680650d50\"\n}\n"

keyNameCollaterals :: KeyName
keyNameCollaterals = KeyName "Collaterals"

keyCollaterals :: TextEnvelope
keyCollaterals = parseKey "{\n \"type\": \"PaymentSigningKeyShelley_ed25519\",\n \"description\": \"Payment Signing Key\",\n \"cborHex\": \"58204babdb63537ccdac393ea23d042af3b7c3587d7dc88ed3b66c959f198ad358fa\"\n}\n"

0 comments on commit ddf375c

Please sign in to comment.