Skip to content

Commit

Permalink
feat(servergroup): Allow users to opt-out of the target desired size …
Browse files Browse the repository at this point in the history
…check when verifying if the instances scaled up or down successfully (#4649) (#4650)

* feat(servergroup): Allow users to opt-out of the target desired size check when verifying if the instances scaled up or down successfully

Some of our users prefer to use the actual instance size list to verify if everything was scaled up or down accordingly instead of relying on the target percentage. They noticed this is causing instability on their end

* chore: Update copyright

* refactor: Introduce the ServerGroupProperties which will be used to define all server group related properties

(cherry picked from commit f5fa468)

Co-authored-by: ovidiupopa07 <[email protected]>
  • Loading branch information
mergify[bot] and ovidiupopa07 committed Feb 22, 2024
1 parent 0bf1cb4 commit 6a37596
Show file tree
Hide file tree
Showing 3 changed files with 137 additions and 20 deletions.
@@ -0,0 +1,47 @@
/*
* Copyright 2024 Harness, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package com.netflix.spinnaker.orca.clouddriver.tasks.servergroup;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.stereotype.Component;

@Component
@ConfigurationProperties(prefix = "server-group")
public class ServerGroupProperties {

private Resize resize = new Resize();

public static class Resize {
private boolean matchInstancesSize;

public void setMatchInstancesSize(boolean matchInstancesSize) {
this.matchInstancesSize = matchInstancesSize;
}

public boolean isMatchInstancesSize() {
return matchInstancesSize;
}
}

public void setResize(Resize resize) {
this.resize = resize;
}

public Resize getResize() {
return resize;
}
}
@@ -1,11 +1,11 @@
/*
* Copyright 2014 Netflix, Inc.
* Copyright 2024 Netflix, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
Expand All @@ -27,6 +27,12 @@
@Component
public class WaitForCapacityMatchTask extends AbstractInstancesCheckTask {

private final ServerGroupProperties serverGroupProperties;

public WaitForCapacityMatchTask(ServerGroupProperties serverGroupProperties) {
this.serverGroupProperties = serverGroupProperties;
}

@Override
protected Map<String, List<String>> getServerGroups(StageExecution stage) {
return (Map<String, List<String>>) stage.getContext().get("deploy.server.groups");
Expand Down Expand Up @@ -88,27 +94,37 @@ protected boolean hasSucceeded(
desired = capacity.getDesired();
}

Integer targetDesiredSize =
Optional.ofNullable((Number) context.get("targetDesiredSize"))
.map(Number::intValue)
.orElse(null);

splainer.add(
String.format(
"checking if capacity matches (desired=%s, target=%s current=%s)",
desired, targetDesiredSize == null ? "none" : targetDesiredSize, instances.size()));
if (targetDesiredSize != null && targetDesiredSize != 0) {
// `targetDesiredSize` is derived from `targetHealthyDeployPercentage` and if present,
// then scaling has succeeded if the number of instances is greater than this value.
if (instances.size() < targetDesiredSize) {
if (serverGroupProperties.getResize().isMatchInstancesSize()) {
splainer.add(
"checking if capacity matches (desired=${desired}, instances.size()=${instances.size()}) ");
if (desired == null || desired != instances.size()) {
splainer.add(
"short-circuiting out of WaitForCapacityMatchTask because targetDesired and current capacity don't match");
"short-circuiting out of WaitForCapacityMatchTask because expected and current capacity don't match}");
return false;
}
} else if (desired == null || desired != instances.size()) {
} else {
Integer targetDesiredSize =
Optional.ofNullable((Number) context.get("targetDesiredSize"))
.map(Number::intValue)
.orElse(null);

splainer.add(
"short-circuiting out of WaitForCapacityMatchTask because expected and current capacity don't match");
return false;
String.format(
"checking if capacity matches (desired=%s, target=%s current=%s)",
desired, targetDesiredSize == null ? "none" : targetDesiredSize, instances.size()));
if (targetDesiredSize != null && targetDesiredSize != 0) {
// `targetDesiredSize` is derived from `targetHealthyDeployPercentage` and if present,
// then scaling has succeeded if the number of instances is greater than this value.
if (instances.size() < targetDesiredSize) {
splainer.add(
"short-circuiting out of WaitForCapacityMatchTask because targetDesired and current capacity don't match");
return false;
}
} else if (desired == null || desired != instances.size()) {
splainer.add(
"short-circuiting out of WaitForCapacityMatchTask because expected and current capacity don't match");
return false;
}
}

boolean disabled = Boolean.TRUE.equals(serverGroup.getDisabled());
Expand Down
Expand Up @@ -30,7 +30,7 @@ import spock.lang.Unroll
class WaitForCapacityMatchTaskSpec extends Specification {

CloudDriverService cloudDriverService = Mock()
@Subject WaitForCapacityMatchTask task = new WaitForCapacityMatchTask() {
@Subject WaitForCapacityMatchTask task = new WaitForCapacityMatchTask(new ServerGroupProperties()) {
@Override
void verifyServerGroupsExist(StageExecution stage) {
// do nothing
Expand Down Expand Up @@ -264,6 +264,60 @@ class WaitForCapacityMatchTaskSpec extends Specification {
true || 4 | [min: 3, max: 10, desired: 4] | [min: "1", max: "50", desired: "5"]
}
@Unroll
void 'should use number of instances when determining if scaling has succeeded even if targetHealthyDeployPercentage is defined'() {
def serverGroupProperties = new ServerGroupProperties()
def resize = new ServerGroupProperties.Resize()
resize.setMatchInstancesSize(true)
serverGroupProperties.setResize(resize)
WaitForCapacityMatchTask task = new WaitForCapacityMatchTask(serverGroupProperties) {
@Override
void verifyServerGroupsExist(StageExecution stage) {
// do nothing
}
}
when:
def context = [
capacity: [
min: configured.min,
max: configured.max,
desired: configured.desired
],
targetHealthyDeployPercentage: targetHealthyDeployPercentage,
targetDesiredSize: targetHealthyDeployPercentage
? Math.round(targetHealthyDeployPercentage * configured.desired / 100) : null
]
def serverGroup = ModelUtils.serverGroup([
asg: [
desiredCapacity: asg.desired
],
capacity: [
min: asg.min,
max: asg.max,
desired: asg.desired
]
])
def instances = []
(1..healthy).each {
instances << ModelUtils.instance([health: [[state: 'Up']]])
}
then:
result == task.hasSucceeded(
new StageExecutionImpl(PipelineExecutionImpl.newPipeline("orca"), "", "", context),
serverGroup, instances, null
)
where:
result || healthy | asg | configured | targetHealthyDeployPercentage
false || 5 | [min: 10, max: 15, desired: 15] | [min: 10, max: 15, desired: 15] | 85
false || 12 | [min: 10, max: 15, desired: 15] | [min: 10, max: 15, desired: 15] | 85
false || 13 | [min: 10, max: 15, desired: 15] | [min: 10, max: 15, desired: 15] | 85
true || 15 | [min: 10, max: 15, desired: 15] | [min: 10, max: 15, desired: 15] | 100
}
private static Instance makeInstance(id, healthState = 'Up') {
ModelUtils.instance([instanceId: id, health: [ [ state: healthState ] ]])
}
Expand Down

0 comments on commit 6a37596

Please sign in to comment.