feat(lambda-event-sources): expose eventSourceMappingId

packages/@aws-cdk/aws-lambda-event-sources/lib/dynamodb.ts

@@ -9,6 +9,8 @@ export interface DynamoEventSourceProps extends StreamEventSourceProps {
  * Use an Amazon DynamoDB stream as an event source for AWS Lambda.
  */
 export class DynamoEventSource extends StreamEventSource {
+  private _eventSourceMappingId: string | undefined = undefined;
+
   constructor(private readonly table: dynamodb.Table, props: DynamoEventSourceProps) {
     super(props);
 
@@ -22,10 +24,18 @@ export class DynamoEventSource extends StreamEventSource {
       throw new Error(`DynamoDB Streams must be enabled on the table ${this.table.node.path}`);
     }
 
-    target.addEventSourceMapping(`DynamoDBEventSource:${this.table.node.uniqueId}`,
+    const eventSourceMapping = target.addEventSourceMapping(`DynamoDBEventSource:${this.table.node.uniqueId}`,
       this.enrichMappingOptions({eventSourceArn: this.table.tableStreamArn})
     );
+    this._eventSourceMappingId = eventSourceMapping.eventSourceMappingId;
 
     this.table.grantStreamRead(target);
   }
+
+  /**
+   * The Ref of the EventSourceMapping
+   */
+  public get eventSourceMappingId(): string | undefined {
+    return this._eventSourceMappingId;
+  }
 }

packages/@aws-cdk/aws-lambda-event-sources/lib/kinesis.ts

@@ -9,6 +9,8 @@ export interface KinesisEventSourceProps extends StreamEventSourceProps {
  * Use an Amazon Kinesis stream as an event source for AWS Lambda.
  */
 export class KinesisEventSource extends StreamEventSource {
+  private _eventSourceMappingId: string | undefined = undefined;
+
   constructor(readonly stream: kinesis.IStream, props: KinesisEventSourceProps) {
     super(props);
 
@@ -18,10 +20,18 @@ export class KinesisEventSource extends StreamEventSource {
   }
 
   public bind(target: lambda.IFunction) {
-    target.addEventSourceMapping(`KinesisEventSource:${this.stream.node.uniqueId}`,
+    const eventSourceMapping = target.addEventSourceMapping(`KinesisEventSource:${this.stream.node.uniqueId}`,
       this.enrichMappingOptions({eventSourceArn: this.stream.streamArn})
     );
+    this._eventSourceMappingId = eventSourceMapping.eventSourceMappingId;
 
     this.stream.grantRead(target);
   }
+
+  /**
+   * The Ref of the EventSourceMapping
+   */
+  public get eventSourceMappingId(): string | undefined {
+    return this._eventSourceMappingId;
+  }
 }

packages/@aws-cdk/aws-lambda-event-sources/lib/sqs.ts

@@ -18,18 +18,28 @@ export interface SqsEventSourceProps {
  * Use an Amazon SQS queue as an event source for AWS Lambda.
  */
 export class SqsEventSource implements lambda.IEventSource {
+  private _eventSourceMappingId: string | undefined = undefined;
+
   constructor(readonly queue: sqs.IQueue, private readonly props: SqsEventSourceProps = { }) {
     if (this.props.batchSize !== undefined && (this.props.batchSize < 1 || this.props.batchSize > 10)) {
       throw new Error(`Maximum batch size must be between 1 and 10 inclusive (given ${this.props.batchSize})`);
     }
   }
 
   public bind(target: lambda.IFunction) {
-    target.addEventSourceMapping(`SqsEventSource:${this.queue.node.uniqueId}`, {
+    const eventSourceMapping = target.addEventSourceMapping(`SqsEventSource:${this.queue.node.uniqueId}`, {
       batchSize: this.props.batchSize,
       eventSourceArn: this.queue.queueArn,
     });
+    this._eventSourceMappingId = eventSourceMapping.eventSourceMappingId;
 
     this.queue.grantConsumeMessages(target);
   }
+
+  /**
+   * The Ref of the EventSourceMapping
+   */
+  public get eventSourceMappingId(): string | undefined {
+    return this._eventSourceMappingId;
+  }
 }

packages/@aws-cdk/aws-lambda-event-sources/test/test.dynamo.ts

@@ -231,4 +231,46 @@ export = {
     test.done();
   },
 
+  'contains eventSourceMappingId after lambda binding'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const fn = new TestFunction(stack, 'Fn');
+    const table = new dynamodb.Table(stack, 'T', {
+      partitionKey: {
+        name: 'id',
+        type: dynamodb.AttributeType.STRING
+      },
+      stream: dynamodb.StreamViewType.NEW_IMAGE
+    });
+    const eventSource = new sources.DynamoEventSource(table, {
+      startingPosition: lambda.StartingPosition.TRIM_HORIZON
+    });
+
+    // WHEN
+    fn.addEventSource(eventSource);
+
+    // THEN
+    test.notEqual(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
+
+  'eventSourceMappingId is undefined before binding to lambda'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const table = new dynamodb.Table(stack, 'T', {
+      partitionKey: {
+        name: 'id',
+        type: dynamodb.AttributeType.STRING
+      },
+      stream: dynamodb.StreamViewType.NEW_IMAGE
+    });
+    const eventSource = new sources.DynamoEventSource(table, {
+      startingPosition: lambda.StartingPosition.TRIM_HORIZON
+    });
+
+    // WHEN/THEN
+    test.equal(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
+
 };

packages/@aws-cdk/aws-lambda-event-sources/test/test.kinesis.ts

@@ -153,4 +153,34 @@ export = {
 
     test.done();
   },
+
+  'contains eventSourceMappingId after lambda binding'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const fn = new TestFunction(stack, 'Fn');
+    const stream = new kinesis.Stream(stack, 'S');
+    const eventSource = new sources.KinesisEventSource(stream, {
+      startingPosition: lambda.StartingPosition.TRIM_HORIZON
+    });
+
+    // WHEN
+    fn.addEventSource(eventSource);
+
+    // THEN
+    test.notEqual(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
+
+  'eventSourceMappingId is undefined before binding to lambda'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const stream = new kinesis.Stream(stack, 'S');
+    const eventSource = new sources.KinesisEventSource(stream, {
+      startingPosition: lambda.StartingPosition.TRIM_HORIZON
+    });
+
+    // WHEN/THEN
+    test.equal(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
 };

packages/@aws-cdk/aws-lambda-event-sources/test/test.sqs.ts

@@ -112,4 +112,30 @@ export = {
 
     test.done();
   },
+
+  'contains eventSourceMappingId after lambda binding'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const fn = new TestFunction(stack, 'Fn');
+    const q = new sqs.Queue(stack, 'Q');
+    const eventSource = new sources.SqsEventSource(q);
+
+    // WHEN
+    fn.addEventSource(eventSource);
+
+    // THEN
+    test.notEqual(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
+
+  'eventSourceMappingId is undefined before binding to lambda'(test: Test) {
+    // GIVEN
+    const stack = new cdk.Stack();
+    const q = new sqs.Queue(stack, 'Q');
+    const eventSource = new sources.SqsEventSource(q);
+
+    // WHEN/THEN
+    test.equal(eventSource.eventSourceMappingId, undefined);
+    test.done();
+  },
 };

packages/@aws-cdk/aws-lambda/lib/event-source-mapping.ts

@@ -67,21 +67,28 @@ export interface EventSourceMappingProps extends EventSourceMappingOptions {
  * modify the Lambda's execution role so it can consume messages from the queue.
  */
 export class EventSourceMapping extends cdk.Resource {
+  /**
+   * The Ref of the EventSourceMapping
+   * @attribute
+   */
+  public readonly eventSourceMappingId: string;
+
   constructor(scope: cdk.Construct, id: string, props: EventSourceMappingProps) {
     super(scope, id);
 
     if (props.maxBatchingWindow && props.maxBatchingWindow.toSeconds() > 300) {
       throw new Error(`maxBatchingWindow cannot be over 300 seconds, got ${props.maxBatchingWindow.toSeconds()}`);
     }
 
-    new CfnEventSourceMapping(this, 'Resource', {
+    const cfnEventSourceMapping = new CfnEventSourceMapping(this, 'Resource', {
       batchSize: props.batchSize,
       enabled: props.enabled,
       eventSourceArn: props.eventSourceArn,
       functionName: props.target.functionName,
       startingPosition: props.startingPosition,
       maximumBatchingWindowInSeconds: props.maxBatchingWindow && props.maxBatchingWindow.toSeconds(),
     });
+    this.eventSourceMappingId = cfnEventSourceMapping.ref;
   }
 }