fix(ecs-patterns): expose service on queue worker services (#2780)

aws · Jun 7, 2019 · 6d83cb9 · 6d83cb9
1 parent a610017
commit 6d83cb9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 8 deletions.
diff --git a/packages/@aws-cdk/aws-ecs-patterns/lib/base/queue-worker-service-base.ts b/packages/@aws-cdk/aws-ecs-patterns/lib/base/queue-worker-service-base.ts
@@ -4,7 +4,7 @@ import sqs = require('@aws-cdk/aws-sqs');
 import cdk = require('@aws-cdk/cdk');
 
 /**
- * Properties to define a Query Worker service
+ * Properties to define a queue worker service
  */
 export interface QueueWorkerServiceBaseProps {
   /**

diff --git a/packages/@aws-cdk/aws-ecs-patterns/lib/ecs/ecs-queue-worker-service.ts b/packages/@aws-cdk/aws-ecs-patterns/lib/ecs/ecs-queue-worker-service.ts
@@ -3,7 +3,7 @@ import cdk = require('@aws-cdk/cdk');
 import { QueueWorkerServiceBase, QueueWorkerServiceBaseProps } from '../base/queue-worker-service-base';
 
 /**
- * Properties to define an Ec2 query worker service
+ * Properties to define an Ec2 queue worker service
  */
 export interface Ec2QueueWorkerServiceProps extends QueueWorkerServiceBaseProps {
   /**
@@ -41,9 +41,15 @@ export interface Ec2QueueWorkerServiceProps extends QueueWorkerServiceBaseProps
 }
 
 /**
- * Class to create an Ec2 query worker service
+ * Class to create an Ec2 queue worker service
  */
 export class Ec2QueueWorkerService extends QueueWorkerServiceBase {
+
+  /**
+   * The ECS service in this construct
+   */
+  public readonly service: ecs.Ec2Service;
+
   constructor(scope: cdk.Construct, id: string, props: Ec2QueueWorkerServiceProps) {
     super(scope, id, props);
 
@@ -61,11 +67,11 @@ export class Ec2QueueWorkerService extends QueueWorkerServiceBase {
 
     // Create an ECS service with the previously defined Task Definition and configure
     // autoscaling based on cpu utilization and number of messages visible in the SQS queue.
-    const ecsService = new ecs.Ec2Service(this, 'QueueWorkerService', {
+    this.service = new ecs.Ec2Service(this, 'QueueWorkerService', {
       cluster: props.cluster,
       desiredCount: this.desiredCount,
       taskDefinition
     });
-    this.configureAutoscalingForService(ecsService);
+    this.configureAutoscalingForService(this.service);
   }
 }
diff --git a/packages/@aws-cdk/aws-ecs-patterns/lib/fargate/fargate-queue-worker-service.ts b/packages/@aws-cdk/aws-ecs-patterns/lib/fargate/fargate-queue-worker-service.ts
@@ -45,9 +45,14 @@ export interface FargateQueueWorkerServiceProps extends QueueWorkerServiceBasePr
 }
 
 /**
- * Class to create a Fargate query worker service
+ * Class to create a Fargate queue worker service
  */
 export class FargateQueueWorkerService extends QueueWorkerServiceBase {
+  /**
+   * The Fargate service in this construct
+   */
+  public readonly service: ecs.FargateService;
+
   constructor(scope: cdk.Construct, id: string, props: FargateQueueWorkerServiceProps) {
     super(scope, id, props);
 
@@ -65,11 +70,11 @@ export class FargateQueueWorkerService extends QueueWorkerServiceBase {
 
     // Create a Fargate service with the previously defined Task Definition and configure
     // autoscaling based on cpu utilization and number of messages visible in the SQS queue.
-    const fargateService = new ecs.FargateService(this, 'FargateQueueWorkerService', {
+    this.service = new ecs.FargateService(this, 'FargateQueueWorkerService', {
       cluster: props.cluster,
       desiredCount: this.desiredCount,
       taskDefinition
     });
-    this.configureAutoscalingForService(fargateService);
+    this.configureAutoscalingForService(this.service);
   }
 }