make num blocks to write a config

Author: udpatil

config/settings.go

@@ -26,6 +26,7 @@ type Settings struct {
 	ReportPath       string   `json:"reportPath,omitempty"`
 	TxsDir           string   `json:"txsDir,omitempty"`
 	TargetGas        uint64   `json:"targetGas,omitempty"`
+	NumBlocksToWrite int      `json:"numBlocksToWrite,omitempty"`
 }
 
 // DefaultSettings returns the default configuration values
@@ -45,6 +46,7 @@ func DefaultSettings() Settings {
 		ReportPath:       "",
 		TxsDir:           "",
 		TargetGas:        10_000_000,
+		NumBlocksToWrite: 100,
 	}
 }
 
@@ -66,6 +68,7 @@ func InitializeViper(cmd *cobra.Command) error {
 		"reportPath":       "report-path",
 		"txsDir":           "txs-dir",
 		"targetGas":        "target-gas",
+		"numBlocksToWrite": "num-blocks-to-write",
 	}
 
 	for viperKey, flagName := range flagBindings {
@@ -90,6 +93,7 @@ func InitializeViper(cmd *cobra.Command) error {
 	viper.SetDefault("reportPath", defaults.ReportPath)
 	viper.SetDefault("txsDir", defaults.TxsDir)
 	viper.SetDefault("targetGas", defaults.TargetGas)
+	viper.SetDefault("numBlocksToWrite", defaults.NumBlocksToWrite)
 	return nil
 }
 
@@ -130,5 +134,6 @@ func ResolveSettings() Settings {
 		ReportPath:       viper.GetString("reportPath"),
 		TxsDir:           viper.GetString("txsDir"),
 		TargetGas:        viper.GetUint64("targetGas"),
+		NumBlocksToWrite: viper.GetInt("numBlocksToWrite"),
 	}
 }

main.go

@@ -68,6 +68,7 @@ func init() {
 	rootCmd.Flags().String("report-path", "", "Path to save the report")
 	rootCmd.Flags().String("txs-dir", "", "Path to save the transactions")
 	rootCmd.Flags().Uint64("target-gas", 10_000_000, "Target gas per block")
+	rootCmd.Flags().Int("num-blocks-to-write", 100, "Number of blocks to write")
 
 	// Initialize Viper with proper error handling
 	if err := config.InitializeViper(rootCmd); err != nil {
@@ -239,9 +240,10 @@ func runLoadTest(ctx context.Context, cmd *cobra.Command, args []string) error {
 			if err != nil {
 				return fmt.Errorf("failed to get latest height: %w", err)
 			}
+			numBlocksToWrite := settings.NumBlocksToWrite
 			writerHeight := latestHeight + 10 // some buffer
 			log.Printf("🔍 Latest height: %d, writer start height: %d", latestHeight, writerHeight)
-			writer := sender.NewTxsWriter(settings.TargetGas, settings.TxsDir, writerHeight, 100)
+			writer := sender.NewTxsWriter(settings.TargetGas, settings.TxsDir, writerHeight, uint64(numBlocksToWrite))
 			dispatcher = sender.NewDispatcher(gen, writer)
 		} else {
 			dispatcher = sender.NewDispatcher(gen, snd)

sender/writer_test.go

@@ -13,7 +13,7 @@ import (
 
 func TestTxsWriter_Flush(t *testing.T) {
 	// two evm transfer txs
-	writer := NewTxsWriter(42000, "/tmp", 1)
+	writer := NewTxsWriter(42000, "/tmp", 1, 1)
 
 	loadConfig := &config.LoadConfig{
 		ChainID: 7777,