IPublisher methods for accessing env vars

aws · Sep 1, 2021 · 25a6359 · 25a6359
1 parent 8564fdf
commit 25a6359
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/packages/@aws-cdk-containers/ecs-service-extensions/lib/extensions/publisher.ts b/packages/@aws-cdk-containers/ecs-service-extensions/lib/extensions/publisher.ts
@@ -14,6 +14,10 @@ import { Construct } from '@aws-cdk/core';
  */
 export interface IPublisher {
   publish(taskDefinition: ecs.TaskDefinition): void;
+
+  envVarKey(): string;
+
+  envVarValue(): string;
 }
 
 /**
@@ -60,6 +64,14 @@ export class PublisherTopic implements IPublisher {
   public publish(taskDefinition: ecs.TaskDefinition) {
     this.topic.grantPublish(taskDefinition.taskRole);
   }
+
+  public envVarKey(): string {
+    return this.topic.node.id;
+  }
+
+  public envVarValue(): string {
+    return this.topic.topicArn;
+  }
 }
 
 /**
@@ -124,9 +136,7 @@ export class PublisherExtension extends ServiceExtension {
     this.parentService = service;
 
     for (const resource of this.props.publishers) {
-      if (resource instanceof PublisherTopic) {
-        this.environment[`${service.id.toUpperCase()}_${resource.topic.node.id.toUpperCase()}_ARN`] = resource.topic.topicArn;
-      }
+      this.environment[`${service.id.toUpperCase()}_${resource.envVarKey().toUpperCase()}_ARN`] = resource.envVarValue();
     }
   }