From 2a4519509df955f9aa8226d15355fe4cb6f6f040 Mon Sep 17 00:00:00 2001
From: Peter Zhu <zhujiaxi@amazon.com>
Date: Fri, 29 Mar 2024 14:37:56 -0400
Subject: [PATCH] Add M54xlarge / M6g4xlarge jenkins agents for integTests
 (#413)

Signed-off-by: Peter Zhu <zhujiaxi@amazon.com>
---
 lib/ci-stack.ts            |  2 ++
 lib/compute/agent-nodes.ts | 36 ++++++++++++++++++++++++++++++++++--
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/lib/ci-stack.ts b/lib/ci-stack.ts
index f09d6c6..76724fd 100644
--- a/lib/ci-stack.ts
+++ b/lib/ci-stack.ts
@@ -158,9 +158,11 @@ export class CIStack extends Stack {
         agentNode.AL2023_X64,
         agentNode.AL2_X64_DOCKER_HOST,
         agentNode.AL2023_X64_DOCKER_HOST,
+        agentNode.AL2023_X64_DOCKER_HOST_EXTRA,
         agentNode.AL2023_ARM64,
         agentNode.AL2_ARM64_DOCKER_HOST,
         agentNode.AL2023_ARM64_DOCKER_HOST,
+        agentNode.AL2023_ARM64_DOCKER_HOST_EXTRA,
         agentNode.AL2023_X64_BENCHMARK_TEST,
         agentNode.UBUNTU2004_X64_GRADLE_CHECK,
         agentNode.UBUNTU2004_X64_DOCKER_BUILDER,
diff --git a/lib/compute/agent-nodes.ts b/lib/compute/agent-nodes.ts
index ba06006..b272946 100644
--- a/lib/compute/agent-nodes.ts
+++ b/lib/compute/agent-nodes.ts
@@ -18,12 +18,16 @@ export class AgentNodes {
 
   readonly AL2023_X64_DOCKER_HOST: AgentNodeProps;
 
+  readonly AL2023_X64_DOCKER_HOST_EXTRA: AgentNodeProps;
+
   readonly AL2023_ARM64: AgentNodeProps;
 
   readonly AL2_ARM64_DOCKER_HOST: AgentNodeProps;
 
   readonly AL2023_ARM64_DOCKER_HOST: AgentNodeProps;
 
+  readonly AL2023_ARM64_DOCKER_HOST_EXTRA: AgentNodeProps;
+
   readonly AL2023_X64_BENCHMARK_TEST: AgentNodeProps;
 
   readonly UBUNTU2004_X64_GRADLE_CHECK: AgentNodeProps;
@@ -78,7 +82,21 @@ export class AgentNodes {
       instanceType: 'C54xlarge',
       remoteUser: 'ec2-user',
       maxTotalUses: -1,
-      minimumNumberOfSpareInstances: 4,
+      minimumNumberOfSpareInstances: 3,
+      numExecutors: 4,
+      amiId: 'ami-0d09563cd5663bdc7',
+      initScript: 'sudo dnf clean all && sudo rm -rf /var/cache/dnf && sudo dnf repolist &&'
+          + ' sudo dnf update --releasever=latest --skip-broken --exclude=openssh* --exclude=docker* --exclude=gh* --exclude=python* -y && docker ps',
+      remoteFs: '/var/jenkins',
+    };
+    this.AL2023_X64_DOCKER_HOST_EXTRA = {
+      agentType: 'unix',
+      customDeviceMapping: '/dev/xvda=:600:true:::encrypted',
+      workerLabelString: 'Jenkins-Agent-AL2023-X64-M54xlarge-Docker-Host',
+      instanceType: 'M54xlarge',
+      remoteUser: 'ec2-user',
+      maxTotalUses: -1,
+      minimumNumberOfSpareInstances: 1,
       numExecutors: 4,
       amiId: 'ami-0d09563cd5663bdc7',
       initScript: 'sudo dnf clean all && sudo rm -rf /var/cache/dnf && sudo dnf repolist &&'
@@ -120,7 +138,21 @@ export class AgentNodes {
       instanceType: 'C6g4xlarge',
       remoteUser: 'ec2-user',
       maxTotalUses: -1,
-      minimumNumberOfSpareInstances: 4,
+      minimumNumberOfSpareInstances: 3,
+      numExecutors: 4,
+      amiId: 'ami-0444fd195657f193f',
+      initScript: 'sudo dnf clean all && sudo rm -rf /var/cache/dnf && sudo dnf repolist &&'
+          + ' sudo dnf update --releasever=latest --skip-broken --exclude=openssh* --exclude=docker* --exclude=gh* --exclude=python* -y && docker ps',
+      remoteFs: '/var/jenkins',
+    };
+    this.AL2023_ARM64_DOCKER_HOST_EXTRA = {
+      agentType: 'unix',
+      customDeviceMapping: '/dev/xvda=:600:true:::encrypted',
+      workerLabelString: 'Jenkins-Agent-AL2023-Arm64-M6g4xlarge-Docker-Host',
+      instanceType: 'M6g4xlarge',
+      remoteUser: 'ec2-user',
+      maxTotalUses: -1,
+      minimumNumberOfSpareInstances: 1,
       numExecutors: 4,
       amiId: 'ami-0444fd195657f193f',
       initScript: 'sudo dnf clean all && sudo rm -rf /var/cache/dnf && sudo dnf repolist &&'