fix timeouts for sa

ava-labs · Feb 24, 2025 · b648757 · b648757
1 parent 394ce8e
commit b648757
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
@@ -544,6 +544,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
@@ -317,6 +317,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,
@@ -356,6 +358,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
@@ -585,6 +585,8 @@ func addAsValidator(network models.Network,
 		}
 	}
 
+	aggregatorCtx, aggregatorCancel = sdkutils.GetTimedContext(constants.SignatureAggregatorTimeout)
+	defer aggregatorCancel()
 	if err := validatormanager.FinishValidatorRegistration(
 		aggregatorCtx,
 		app,