feat(deadline): allow providing subnets for RenderQueue's ALB (#264)

aws · Jan 5, 2021 · 53088fb · 53088fb
1 parent c5d881a
commit 53088fb
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/packages/aws-rfdk/lib/deadline/lib/render-queue-ref.ts b/packages/aws-rfdk/lib/deadline/lib/render-queue-ref.ts
@@ -243,6 +243,13 @@ export interface RenderQueueProps {
    */
   readonly vpcSubnets?: SubnetSelection;
 
+  /**
+   * The subnets into which to place the Application Load Balancer that is deployed.
+   *
+   * @default - One Private subnet from each AZ.
+   */
+  readonly vpcSubnetsAlb?: SubnetSelection;
+
   /**
    * Hostname to use to connect to the RenderQueue.
    *

diff --git a/packages/aws-rfdk/lib/deadline/lib/render-queue.ts b/packages/aws-rfdk/lib/deadline/lib/render-queue.ts
@@ -334,6 +334,7 @@ export class RenderQueue extends RenderQueueBase implements IGrantable {
 
     const loadBalancer = new ApplicationLoadBalancer(this, 'LB', {
       vpc: this.cluster.vpc,
+      vpcSubnets: props.vpcSubnetsAlb ?? { subnetType: SubnetType.PRIVATE, onePerAz: true },
       internetFacing: false,
       deletionProtection: props.deletionProtection ?? true,
     });

diff --git a/packages/aws-rfdk/lib/deadline/test/render-queue.test.ts b/packages/aws-rfdk/lib/deadline/test/render-queue.test.ts
@@ -1696,6 +1696,9 @@ describe('RenderQueue', () => {
       vpcSubnets: {
         subnets,
       },
+      vpcSubnetsAlb: {
+        subnets,
+      },
     };
     const isolatedStack = new Stack(app, 'IsolatedStack');
 
@@ -1708,6 +1711,12 @@ describe('RenderQueue', () => {
         'SubnetID2',
       ),
     }));
+    expectCDK(isolatedStack).to(haveResource('AWS::ElasticLoadBalancingV2::LoadBalancer', {
+      Subnets: [
+        'SubnetID1',
+        'SubnetID2',
+      ],
+    }));
   });
 
   test('can specify instance type', () => {