fund staker faster

harness/app.go

@@ -90,6 +90,9 @@ func startHarness(cmdCtx context.Context, cfg config.Config) error {
 		stakers = append(stakers, NewBTCStaker(tm, stakerSender, fpMgr.randomFp().btcPk.MustToBTCPK(), tm.fundingRequests))
 	}
 
+	// periodically check if we need to fund the staker
+	go tm.fundForever(ctx)
+
 	// fund all stakers
 	if err := tm.fundAllParties(ctx, senders(stakers)); err != nil {
 		return err
@@ -117,7 +120,6 @@ func startHarness(cmdCtx context.Context, cfg config.Config) error {
 	fpMgr.Start(ctx)
 
 	go tm.listBlocksForever(ctx)
-	go tm.fundForever(ctx)
 
 	select {
 	case <-ctx.Done():

harness/btcstaker.go

@@ -88,8 +88,13 @@ func (s *BTCStaker) runForever(ctx context.Context, stakerAddress btcutil.Addres
 			err = s.buildAndSendStakingTransaction(ctx, stakerAddress, stakerPk, &paramsResp.Params)
 			if err != nil {
 				fmt.Printf("🚫 Err in BTC Staker (%s), err: %v\n", s.client.BabylonAddress.String(), err)
-				if strings.Contains(err.Error(), "Insufficient funds") {
-					s.fundingRequest <- s.client.BabylonAddress
+				if strings.Contains(strings.ToLower(err.Error()), "insufficient funds") {
+					select {
+					case s.fundingRequest <- s.client.BabylonAddress:
+						time.Sleep(5 * time.Second)
+					default:
+						fmt.Println("fundingRequest channel is full or closed")
+					}
 				}
 			}
 		}

harness/manager.go

@@ -216,7 +216,7 @@ func StartManager(ctx context.Context, outputsInWallet uint32, epochInterval uin
 		manger:             manager,
 		babylonDir:         babylonDir,
 		benchConfig:        runCfg,
-		fundingRequests:    make(chan sdk.AccAddress),
+		fundingRequests:    make(chan sdk.AccAddress, 100),
 	}, nil
 }
 
@@ -366,6 +366,7 @@ func (tm *TestManager) fundAllParties(
 
 	return nil
 }
+
 func (tm *TestManager) fundBnnAddress(
 	ctx context.Context,
 	addr sdk.AccAddress,
@@ -425,14 +426,19 @@ func (tm *TestManager) listBlocksForever(ctx context.Context) {
 }
 
 func (tm *TestManager) fundForever(ctx context.Context) {
+	ticker := time.NewTicker(3 * time.Second)
+	defer ticker.Stop()
 	for {
 		select {
 		case <-ctx.Done():
 			return
+		case <-ticker.C:
 		case addr := <-tm.fundingRequests:
-			if err := tm.fundBnnAddress(ctx, addr); err != nil {
-				fmt.Printf("🚫 Failed to fund addr %s, err %v\n", addr.String(), err)
-			}
+			go func() {
+				if err := tm.fundBnnAddress(ctx, addr); err != nil {
+					fmt.Printf("🚫 Failed to fund addr %s, err %v\n", addr.String(), err)
+				}
+			}()
 		}
 	}
 }