feat(batch-alpha): add RuntimePlatform prop

packages/@aws-cdk/aws-batch-alpha/lib/ecs-container-definition.ts

@@ -8,6 +8,7 @@ import { Construct, IConstruct } from 'constructs';
 import { CfnJobDefinition } from 'aws-cdk-lib/aws-batch';
 import { LinuxParameters } from './linux-parameters';
 import { LogGroup } from 'aws-cdk-lib/aws-logs';
+import { RuntimePlatform } from './runtime-platform';
 
 const EFS_VOLUME_SYMBOL = Symbol.for('aws-cdk-lib/aws-batch/lib/container-definition.EfsVolume');
 const HOST_VOLUME_SYMBOL = Symbol.for('aws-cdk-lib/aws-batch/lib/container-definition.HostVolume');
@@ -429,6 +430,15 @@ export interface IEcsContainerDefinition extends IConstruct {
    */
   readonly readonlyRootFilesystem?: boolean;
 
+  /**
+   * The operating system that your task definitions are running on.
+   *
+   * A runtimePlatform is supported only for tasks using the Fargate launch type.
+   *
+   * @default - Undefined.
+   */
+  readonly runtimePlatform?: RuntimePlatform;
+
   /**
    * A map from environment variable names to the secrets for the container. Allows your job definitions
    * to reference the secret by the environment variable name defined in this property.
@@ -546,6 +556,15 @@ export interface EcsContainerDefinitionProps {
    */
   readonly readonlyRootFilesystem?: boolean;
 
+  /**
+   * The operating system that your task definitions are running on.
+   *
+   * A runtimePlatform is supported only for tasks using the Fargate launch type.
+   *
+   * @default - Undefined.
+   */
+  readonly runtimePlatform?: RuntimePlatform;
+
   /**
    * A map from environment variable names to the secrets for the container. Allows your job definitions
    * to reference the secret by the environment variable name defined in this property.
@@ -585,6 +604,7 @@ abstract class EcsContainerDefinitionBase extends Construct implements IEcsConta
   public readonly linuxParameters?: LinuxParameters;
   public readonly logDriverConfig?: ecs.LogDriverConfig;
   public readonly readonlyRootFilesystem?: boolean;
+  public readonly runtimePlatform?: RuntimePlatform;
   public readonly secrets?: { [envVarName: string]: Secret };
   public readonly user?: string;
   public readonly volumes: EcsVolume[];
@@ -618,6 +638,10 @@ abstract class EcsContainerDefinitionBase extends Construct implements IEcsConta
     this.user = props.user;
     this.volumes = props.volumes ?? [];
 
+    if (props.runtimePlatform) {
+      this.runtimePlatform = props.runtimePlatform;
+    }
+
     this.imageConfig = props.image.bind(this, {
       ...this as any,
       taskDefinition: {

packages/@aws-cdk/aws-batch-alpha/lib/index.ts

@@ -10,4 +10,5 @@ export * from './linux-parameters';
 export * from './managed-compute-environment';
 export * from './multinode-job-definition';
 export * from './scheduling-policy';
-export * from './unmanaged-compute-environment';
+export * from './unmanaged-compute-environment';
+export * from './runtime-platform';

packages/@aws-cdk/aws-batch-alpha/lib/runtime-platform.ts

@@ -0,0 +1,110 @@
+/**
+ * The CpuArchitecture for Runtime Platform.
+ */
+export class CpuArchitecture {
+  /**
+  * ARM64
+  * */
+  public static readonly ARM64 = CpuArchitecture.of('ARM64');
+
+  /**
+   * X86_64
+  */
+  public static readonly X86_64 = CpuArchitecture.of('X86_64');
+
+  /**
+   * Other cpu architecture.
+   *
+   * @see https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-ecs-taskdefinition-runtimeplatform.html#cfn-ecs-taskdefinition-runtimeplatform-cpuarchitecture for all available cpu architecture.
+   *
+   * @param cpuArchitecture cpu architecture.
+   *
+  */
+  public static of(cpuArchitecture: string) { return new CpuArchitecture(cpuArchitecture); }
+
+  /**
+   *
+   * @param _cpuArchitecture The CPU architecture.
+   */
+  private constructor(public readonly _cpuArchitecture: string) { }
+}
+
+/**
+ * The operating system for Fargate Runtime Platform.
+ */
+export class OperatingSystemFamily {
+  /**
+   * LINUX
+   */
+  public static readonly LINUX = OperatingSystemFamily.of('LINUX');
+
+  /**
+   * WINDOWS_SERVER_2004_CORE
+   */
+  public static readonly WINDOWS_SERVER_2004_CORE = OperatingSystemFamily.of('WINDOWS_SERVER_2004_CORE');
+
+  /**
+   * WINDOWS_SERVER_2016_FULL
+   */
+  public static readonly WINDOWS_SERVER_2016_FULL = OperatingSystemFamily.of('WINDOWS_SERVER_2016_FULL');
+
+  /**
+   * WINDOWS_SERVER_2019_CORE
+   */
+  public static readonly WINDOWS_SERVER_2019_CORE = OperatingSystemFamily.of('WINDOWS_SERVER_2019_CORE');
+
+  /**
+   * WINDOWS_SERVER_2019_FULL
+   */
+  public static readonly WINDOWS_SERVER_2019_FULL = OperatingSystemFamily.of('WINDOWS_SERVER_2019_FULL');
+
+  /**
+   * WINDOWS_SERVER_2022_CORE
+   */
+  public static readonly WINDOWS_SERVER_2022_CORE = OperatingSystemFamily.of('WINDOWS_SERVER_2022_CORE');
+
+  /**
+   * WINDOWS_SERVER_2022_FULL
+   */
+  public static readonly WINDOWS_SERVER_2022_FULL = OperatingSystemFamily.of('WINDOWS_SERVER_2022_FULL');
+
+  /**
+   * WINDOWS_SERVER_20H2_CORE
+   */
+  public static readonly WINDOWS_SERVER_20H2_CORE = OperatingSystemFamily.of('WINDOWS_SERVER_20H2_CORE');
+
+  /*
+   * Other operating system family.
+   *
+   * @see https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-ecs-taskdefinition-runtimeplatform.html#cfn-ecs-taskdefinition-runtimeplatform-operatingsystemfamily for all available operating system family.
+   *
+   * @param family operating system family.
+   *
+   */
+  public static of(family: string) { return new OperatingSystemFamily(family); }
+
+  /**
+   *
+   * @param _operatingSystemFamily The operating system family.
+   */
+  private constructor(public readonly _operatingSystemFamily: string) { }
+}
+
+/**
+ * The interface for Runtime Platform.
+ */
+export interface RuntimePlatform {
+/**
+ * The CpuArchitecture for Fargate Runtime Platform
+ *
+ * @default - Undefined.
+ */
+  readonly cpuArchitecture?: CpuArchitecture,
+
+  /**
+   * The operating system for Fargate Runtime Platform.
+   *
+   * @default - Undefined.
+   */
+  readonly operatingSystemFamily?: OperatingSystemFamily,
+}

packages/@aws-cdk/aws-batch-alpha/test/ecs-job-definition.test.ts

@@ -1,7 +1,10 @@
 import { Template } from 'aws-cdk-lib/assertions';
 import * as ecs from 'aws-cdk-lib/aws-ecs';
 import { DefaultTokenResolver, Size, StringConcat, Stack, Tokenization } from 'aws-cdk-lib';
-import { Compatibility, EcsEc2ContainerDefinition, EcsFargateContainerDefinition, EcsJobDefinition } from '../lib';
+import * as iam from 'aws-cdk-lib/aws-iam';
+import { Compatibility, CpuArchitecture, EcsEc2ContainerDefinition, EcsFargateContainerDefinition, EcsJobDefinition, OperatingSystemFamily, JobQueue, ManagedEc2EcsComputeEnvironment } from '../lib';
+import { Vpc } from 'aws-cdk-lib/aws-ec2';
+
 
 test('EcsJobDefinition respects propagateTags', () => {
   // GIVEN
@@ -42,6 +45,29 @@ test('EcsJobDefinition uses Compatibility.EC2 for EC2 containers', () => {
   });
 });
 
+test('EcsJobDefinition uses runtimePlatform EC2 containers', () => {
+  // GIVEN
+  const stack = new Stack();
+
+  // WHEN
+  new EcsJobDefinition(stack, 'ECSJobDefn', {
+    container: new EcsEc2ContainerDefinition(stack, 'EcsContainer', {
+      cpu: 256,
+      image: ecs.ContainerImage.fromRegistry('amazon/amazon-ecs-sample'),
+      memory: Size.mebibytes(2048),
+      runtimePlatform: {
+        cpuArchitecture: CpuArchitecture.ARM64,
+        operatingSystemFamily: OperatingSystemFamily.LINUX,
+      },
+    }),
+  });
+
+  // THEN
+  Template.fromStack(stack).hasResourceProperties('AWS::Batch::JobDefinition', {
+    PlatformCapabilities: [Compatibility.EC2],
+  });
+});
+
 test('EcsJobDefinition uses Compatibility.FARGATE for Fargate containers', () => {
   // GIVEN
   const stack = new Stack();

packages/@aws-cdk/aws-batch-alpha/test/integ.ecs-job-definition.ts

@@ -1,5 +1,5 @@
 import { Vpc } from 'aws-cdk-lib/aws-ec2';
-import { ContainerImage, FargatePlatformVersion } from 'aws-cdk-lib/aws-ecs';
+import { ContainerImage, CpuArchitecture, FargatePlatformVersion, OperatingSystemFamily } from 'aws-cdk-lib/aws-ecs';
 import * as efs from 'aws-cdk-lib/aws-efs';
 import * as ecs from 'aws-cdk-lib/aws-ecs';
 import * as ssm from 'aws-cdk-lib/aws-ssm';
@@ -42,6 +42,10 @@ new batch.EcsJobDefinition(stack, 'ECSJobDefn', {
       name: batch.UlimitName.CORE,
       softLimit: 10,
     }],
+    runtimePlatform: {
+      cpuArchitecture: CpuArchitecture.ARM64,
+      operatingSystemFamily: OperatingSystemFamily.LINUX,
+    },
     secrets: {
       MY_SECRET_ENV_VAR: batch.Secret.fromSecretsManager(new secretsmanager.Secret(stack, 'mySecret')),
       ANOTHER_ONE: batch.Secret.fromSecretsManagerVersion(new secretsmanager.Secret(stack, 'anotherSecret'), {