feat: minor polycli enhancements

cmd/loadtest/app.go

@@ -70,6 +70,7 @@ type (
 		ShouldProduceSummary       *bool
 		SummaryOutputMode          *string
 		LegacyTransactionMode      *bool
+		SendOnly                   *bool
 		RecallLength               *uint64
 
 		// Computed
@@ -220,6 +221,7 @@ func initFlags() {
 	ltp.BatchSize = LoadtestCmd.PersistentFlags().Uint64("batch-size", 999, "Number of batches to perform at a time for receipt fetching. Default is 999 requests at a time.")
 	ltp.SummaryOutputMode = LoadtestCmd.PersistentFlags().String("output-mode", "text", "Format mode for summary output (json | text)")
 	ltp.LegacyTransactionMode = LoadtestCmd.PersistentFlags().Bool("legacy", false, "Send a legacy transaction instead of an EIP1559 transaction.")
+	ltp.SendOnly = LoadtestCmd.PersistentFlags().Bool("send-only", false, "Send transactions and load without waiting for it to be mined.")
 
 	// Local flags.
 	ltp.Modes = LoadtestCmd.Flags().StringSliceP("mode", "m", []string{"t"}, `The testing mode to use. It can be multiple like: "t,c,d,f"

cmd/loadtest/loadtest.go

@@ -151,6 +151,7 @@ func initializeLoadTestParams(ctx context.Context, c *ethclient.Client) error {
 		inputLoadTestParams.CurrentGasTipCap = gasTipCap
 	}
 
+	*inputLoadTestParams.PrivateKey = strings.TrimPrefix(*inputLoadTestParams.PrivateKey, "0x")
 	privateKey, err := ethcrypto.HexToECDSA(*inputLoadTestParams.PrivateKey)
 	if err != nil {
 		log.Error().Err(err).Msg("Couldn't process the hex private key")
@@ -164,7 +165,6 @@ func initializeLoadTestParams(ctx context.Context, c *ethclient.Client) error {
 		return err
 	}
 	log.Trace().Uint64("blocknumber", blockNumber).Msg("Current Block Number")
-
 	ethAddress := ethcrypto.PubkeyToAddress(privateKey.PublicKey)
 
 	nonce, err := c.NonceAt(ctx, ethAddress, bigBlockNumber)
@@ -310,6 +310,10 @@ func initNonce(ctx context.Context, c *ethclient.Client, rpc *ethrpc.Client) err
 
 func completeLoadTest(ctx context.Context, c *ethclient.Client, rpc *ethrpc.Client) error {
 	log.Debug().Uint64("startNonce", startNonce).Uint64("lastNonce", currentNonce).Msg("Finished main load test loop")
+	if *inputLoadTestParams.SendOnly {
+		log.Info().Msgf("Send only mode enabled: finished sending %d transactions.", currentNonce-startNonce)
+		return nil
+	}
 	log.Debug().Msg("Waiting for remaining transactions to be completed and mined")
 
 	var err error

doc/polycli_loadtest.md

@@ -145,6 +145,7 @@ The codebase has a contract that used for load testing. It's written in Yul and
   -r, --rpc-url string                         The RPC endpoint url (default "http://localhost:8545")
       --seed int                               A seed for generating random values and addresses (default 123456)
       --send-amount string                     The amount of wei that we'll send every transaction (default "0x38D7EA4C68000")
+      --send-only                              Send transactions and load without waiting for it to be mined.
       --steady-state-tx-pool-size uint         When using adaptive rate limiting, this value sets the target queue size. If the queue is smaller than this value, we'll speed up. If the queue is smaller than this value, we'll back off. (default 1000)
       --summarize                              Should we produce an execution summary after the load test has finished. If you're running a large load test, this can take a long time
   -t, --time-limit int                         Maximum number of seconds to spend for benchmarking. Use this to benchmark within a fixed total amount of time. Per default there is no time limit. (default -1)

doc/polycli_loadtest_uniswapv3.md

@@ -99,6 +99,7 @@ The command also inherits flags from parent commands.
   -r, --rpc-url string                         The RPC endpoint url (default "http://localhost:8545")
       --seed int                               A seed for generating random values and addresses (default 123456)
       --send-amount string                     The amount of wei that we'll send every transaction (default "0x38D7EA4C68000")
+      --send-only                              Send transactions and load without waiting for it to be mined.
       --steady-state-tx-pool-size uint         When using adaptive rate limiting, this value sets the target queue size. If the queue is smaller than this value, we'll speed up. If the queue is smaller than this value, we'll back off. (default 1000)
       --summarize                              Should we produce an execution summary after the load test has finished. If you're running a large load test, this can take a long time
   -t, --time-limit int                         Maximum number of seconds to spend for benchmarking. Use this to benchmark within a fixed total amount of time. Per default there is no time limit. (default -1)