@@ -25,43 +25,38 @@ module Cardano.Benchmarking.GeneratorTx.SubmissionClient
2525 , txSubmissionClient
2626 ) where
2727
28- import Cardano.Prelude hiding (ByteString , atomically , retry , state , threadDelay )
29- import Prelude (error , fail )
30-
31- import Control.Arrow ((&&&) )
32-
33- import qualified Data.List as L
34- import qualified Data.List.Extra as L
35- import qualified Data.List.NonEmpty as NE
36- import qualified Data.Text as T
28+ import Cardano.Api hiding (Active )
29+ import Cardano.Api.Shelley (fromShelleyTxId , toConsensusGenTx )
3730
31+ import Cardano.Benchmarking.LogTypes
32+ import Cardano.Benchmarking.Types
33+ import Cardano.Logging
34+ import Cardano.Prelude hiding (ByteString , atomically , retry , state , threadDelay )
3835import Cardano.Tracing.OrphanInstances.Byron ()
3936import Cardano.Tracing.OrphanInstances.Common ()
4037import Cardano.Tracing.OrphanInstances.Consensus ()
4138import Cardano.Tracing.OrphanInstances.Network ()
4239import Cardano.Tracing.OrphanInstances.Shelley ()
43-
4440import qualified Ouroboros.Consensus.Cardano as Consensus (CardanoBlock )
41+ import qualified Ouroboros.Consensus.Cardano.Block as Block
42+ (TxId (GenTxIdAllegra , GenTxIdAlonzo , GenTxIdBabbage , GenTxIdConway , GenTxIdMary , GenTxIdShelley ))
4543import Ouroboros.Consensus.Ledger.SupportsMempool (GenTx , GenTxId , txInBlockSize )
4644import qualified Ouroboros.Consensus.Ledger.SupportsMempool as Mempool
4745import Ouroboros.Consensus.Shelley.Eras (StandardCrypto )
4846import qualified Ouroboros.Consensus.Shelley.Ledger.Mempool as Mempool (TxId (ShelleyTxId ))
49-
50- import qualified Ouroboros.Consensus.Cardano.Block as Block
51- (TxId (GenTxIdAllegra , GenTxIdAlonzo , GenTxIdBabbage , GenTxIdConway , GenTxIdMary , GenTxIdShelley ))
52-
5347import Ouroboros.Network.Protocol.TxSubmission2.Client (ClientStIdle (.. ),
5448 ClientStTxIds (.. ), ClientStTxs (.. ), TxSubmissionClient (.. ))
5549import Ouroboros.Network.Protocol.TxSubmission2.Type (BlockingReplyList (.. ),
56- TokBlockingStyle (.. ), TxSizeInBytes )
50+ NumTxIdsToAck (.. ), NumTxIdsToReq (.. ), TokBlockingStyle (.. ))
51+ import Ouroboros.Network.SizeInBytes
5752
58- import Cardano.Api hiding (Active )
59- import Cardano.Api.Shelley (fromShelleyTxId , toConsensusGenTx )
60-
61- import Cardano.Logging
53+ import Prelude (error , fail )
6254
63- import Cardano.Benchmarking.LogTypes
64- import Cardano.Benchmarking.Types
55+ import Control.Arrow ((&&&) )
56+ import qualified Data.List as L
57+ import qualified Data.List.Extra as L
58+ import qualified Data.List.NonEmpty as NE
59+ import qualified Data.Text as T
6560type CardanoBlock = Consensus. CardanoBlock StandardCrypto
6661
6762data SubmissionThreadStats
@@ -129,10 +124,10 @@ txSubmissionClient tr bmtr initialTxSource endOfProtocolCallback =
129124 requestTxIds :: forall blocking .
130125 LocalState era
131126 -> TokBlockingStyle blocking
132- -> Word16
133- -> Word16
127+ -> NumTxIdsToAck
128+ -> NumTxIdsToReq
134129 -> m (ClientStTxIds blocking (GenTxId CardanoBlock ) (GenTx CardanoBlock ) m () )
135- requestTxIds state blocking ackNum reqNum = do
130+ requestTxIds state blocking ( NumTxIdsToAck ackNum) ( NumTxIdsToReq reqNum) = do
136131 let ack = Ack $ fromIntegral ackNum
137132 req = Req $ fromIntegral reqNum
138133 traceWith tr $ reqIdsTrace ack req blocking
@@ -182,8 +177,8 @@ txSubmissionClient tr bmtr initialTxSource endOfProtocolCallback =
182177 , stsUnavailable =
183178 stsUnavailable stats + Unav (length missIds)}))
184179
185- txToIdSize :: tx -> (GenTxId CardanoBlock , TxSizeInBytes )
186- txToIdSize = (Mempool. txId &&& txInBlockSize) . toGenTx
180+ txToIdSize :: tx -> (GenTxId CardanoBlock , SizeInBytes )
181+ txToIdSize = (Mempool. txId &&& ( SizeInBytes . txInBlockSize) ) . toGenTx
187182
188183 toGenTx :: tx -> GenTx CardanoBlock
189184 toGenTx tx = toConsensusGenTx $ TxInMode (shelleyBasedEra @ era ) tx
0 commit comments