fix timeouts for sa (#2630)

ava-labs · Feb 24, 2025 · 752b23c · 752b23c
1 parent 0cbb0dc
commit 752b23c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/cmd/blockchaincmd/add_validator.go b/cmd/blockchaincmd/add_validator.go
@@ -525,6 +525,8 @@ func CallAddValidator(
 		}
 	}
 
+	aggregatorCtx, aggregatorCancel = sdkutils.GetTimedContext(constants.SignatureAggregatorTimeout)
+	defer aggregatorCancel()
 	if err := validatormanager.FinishValidatorRegistration(
 		aggregatorCtx,
 		app,

diff --git a/cmd/blockchaincmd/remove_validator.go b/cmd/blockchaincmd/remove_validator.go
@@ -318,6 +318,8 @@ func removeValidatorSOV(
 		if !force {
 			return fmt.Errorf("validator %s is not eligible for rewards. Use --force flag to force removal", nodeID)
 		}
+		aggregatorCtx, aggregatorCancel = sdkutils.GetTimedContext(constants.SignatureAggregatorTimeout)
+		defer aggregatorCancel()
 		signedMessage, validationID, err = validatormanager.InitValidatorRemoval(
 			aggregatorCtx,
 			app,
@@ -357,6 +359,8 @@ func removeValidatorSOV(
 		}
 	}
 
+	aggregatorCtx, aggregatorCancel = sdkutils.GetTimedContext(constants.SignatureAggregatorTimeout)
+	defer aggregatorCancel()
 	if err := validatormanager.FinishValidatorRemoval(
 		aggregatorCtx,
 		app,

diff --git a/cmd/nodecmd/local.go b/cmd/nodecmd/local.go
@@ -583,6 +583,8 @@ func addAsValidator(network models.Network,
 		}
 	}
 
+	aggregatorCtx, aggregatorCancel = sdkutils.GetTimedContext(constants.SignatureAggregatorTimeout)
+	defer aggregatorCancel()
 	if err := validatormanager.FinishValidatorRegistration(
 		aggregatorCtx,
 		app,