Handle case context is done

abicky · Oct 30, 2024 · 71610e3 · 71610e3
1 parent 63f719c
commit 71610e3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/capacity/autoscalinggroup.go b/internal/capacity/autoscalinggroup.go
@@ -345,6 +345,8 @@ func (asg *AutoScalingGroup) waitUntilInstancesInService(ctx context.Context, ca
 			continue
 		case <-timer.C:
 			return xerrors.Errorf("can't prepare at least %d in-service instances within %v", capacity, timeout)
+		case <-ctx.Done():
+			return ctx.Err()
 		}
 	}
 }

diff --git a/internal/capacity/cluster.go b/internal/capacity/cluster.go
@@ -66,6 +66,8 @@ func (c *cluster) WaitUntilContainerInstancesRegistered(ctx context.Context, cou
 			continue
 		case <-timer.C:
 			return xerrors.Errorf("%d container instances expected to be registered but only %d instances were registered within %v", count, foundCount, timeout)
+		case <-ctx.Done():
+			return ctx.Err()
 		}
 	}
 }