ENGOPS-1634: Custom bk-agent-scaller that runs on funnel agent medium…

README.md

@@ -1,5 +1,10 @@
 # Buildkite Agent Scaler
 
+# Nestio ORG changes 
+Please refer to Notion documentation: 
+[Buildkite Agent Scaler Nestio](https://www.notion.so/funnelleasing/Buildkite-Agent-Scaler-Lambda-Function-Update-Version-1-9-0-11abeed2aafd8094b76df77632a69d68)
+
+# Offiacial Readme
 An AWS lambda function that handles the scaling of an
 [Amazon Autoscaling Group](https://docs.aws.amazon.com/autoscaling/ec2/userguide/AutoScalingGroup.html)
 (ASG) based on metrics provided by the Buildkite Agent Metrics API.

main.go

@@ -13,13 +13,13 @@ import (
 func main() {
 	var (
 		// aws params
-		asgName           = flag.String("asg-name", "", "The name of the autoscaling group")
+		asgName           = flag.String("asg-name", "ci-docker-medium-agent", "The name of the autoscaling group")
 		agentsPerInstance = flag.Int("agents-per-instance", 1, "The number of agents per instance")
 		cwMetrics         = flag.Bool("cloudwatch-metrics", false, "Whether to publish cloudwatch metrics")
 		ssmTokenKey       = flag.String("agent-token-ssm-key", "", "The AWS SSM Parameter Store key for the agent token")
 
 		// buildkite params
-		buildkiteQueue      = flag.String("queue", "default", "The queue to watch in the metrics")
+		buildkiteQueue      = flag.String("queue", "docker-medium", "The queue to watch in the metrics")
 		buildkiteAgentToken = flag.String("agent-token", "", "A buildkite agent registration token")
 		includeWaiting      = flag.Bool("include-waiting", false, "Whether to include jobs behind a wait step for scaling")
 

scaler/scaler.go

@@ -134,6 +134,8 @@ func (s *Scaler) Run() (time.Duration, error) {
 }
 
 func (s *Scaler) scaleIn(desired int64, current AutoscaleGroupDetails) error {
+	s.scaleInParams.Disable = true
+	log.Print("Scale in disable is set to:", s.scaleInParams.Disable)
 	if s.scaleInParams.Disable {
 		return nil
 	}