added capturing s3 otel attributes from lamba S3Event

elastic · Oct 17, 2023 · 1ea7a58 · 1ea7a58
1 parent 69e6414
commit 1ea7a58
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 7 deletions.
diff --git a/...ambda-plugin/src/main/java/co/elastic/apm/agent/awslambda/helper/S3TransactionHelper.java b/...ambda-plugin/src/main/java/co/elastic/apm/agent/awslambda/helper/S3TransactionHelper.java
@@ -78,12 +78,20 @@ protected void setTransactionTriggerData(Transaction transaction, S3Event s3Even
 
             cloudOrigin.withRegion(s3NotificationRecord.getAwsRegion());
 
-            if (null != s3NotificationRecord.getS3() && null != s3NotificationRecord.getS3().getBucket()) {
-                S3EventNotification.S3BucketEntity bucket = s3NotificationRecord.getS3().getBucket();
-                ServiceOrigin serviceOrigin = transaction.getContext().getServiceOrigin();
-                serviceOrigin.withId(bucket.getArn());
-                serviceOrigin.withName(bucket.getName());
-                serviceOrigin.withVersion(s3NotificationRecord.getEventVersion());
+            if (null != s3NotificationRecord.getS3()) {
+                if (null != s3NotificationRecord.getS3().getBucket()) {
+                    S3EventNotification.S3BucketEntity bucket = s3NotificationRecord.getS3().getBucket();
+                    ServiceOrigin serviceOrigin = transaction.getContext().getServiceOrigin();
+                    serviceOrigin.withId(bucket.getArn());
+                    serviceOrigin.withName(bucket.getName());
+                    serviceOrigin.withVersion(s3NotificationRecord.getEventVersion());
+
+                    transaction.withOtelAttribute("aws.s3.bucket", bucket.getName());
+                }
+                if (null != s3NotificationRecord.getS3().getObject()) {
+                    final S3EventNotification.S3ObjectEntity object = s3NotificationRecord.getS3().getObject();
+                    transaction.withOtelAttribute("aws.s3.key", object.getKey());
+                }
             }
         }
     }

diff --git a/.../apm-awslambda-plugin/src/test/java/co/elastic/apm/agent/awslambda/S3EventLambdaTest.java b/.../apm-awslambda-plugin/src/test/java/co/elastic/apm/agent/awslambda/S3EventLambdaTest.java
@@ -66,7 +66,8 @@ protected boolean supportsContextPropagation() {
     private S3EventNotification.S3EventNotificationRecord createS3NotificationRecord() {
         S3EventNotification.ResponseElementsEntity responseElements = new S3EventNotification.ResponseElementsEntity("xAmzId2", S3_REQUEST_ID);
         S3EventNotification.S3BucketEntity bucket = new S3EventNotification.S3BucketEntity(S3_BUCKET_NAME, null, S3_BUCKET_ARN);
-        S3EventNotification.S3Entity s3 = new S3EventNotification.S3Entity("configId", bucket, null, "3.3");
+        S3EventNotification.S3ObjectEntity object = new S3EventNotification.S3ObjectEntity("b21b84d653bb07b05b1e6b33684dc11b", 1305107, "b21b84d653bb07b05b1e6b33684dc11b", "0C0F6F405D6ED209E1");
+        S3EventNotification.S3Entity s3 = new S3EventNotification.S3Entity("configId", bucket, object, "3.3");
         return new S3EventNotification.S3EventNotificationRecord(EVENT_SOURCE_REGION, S3_EVENT_NAME, S3_EVENT_SOURCE, null,
             S3_EVENT_VERSION, null, responseElements, s3, null);
     }
@@ -102,6 +103,8 @@ public void testBasicCall() {
         assertThat(faas.getId()).isEqualTo(TestContext.FUNCTION_ARN);
         assertThat(faas.getTrigger().getType()).isEqualTo("datasource");
         assertThat(faas.getTrigger().getRequestId()).isEqualTo(S3_REQUEST_ID);
+
+        verifyOtelAttributes(transaction);
     }
 
     @Test
@@ -174,4 +177,12 @@ private void validateResultsForUnspecifiedRecord() {
         assertThat(faas.getTrigger().getType()).isEqualTo("datasource");
         assertThat(faas.getTrigger().getRequestId()).isNull();
     }
+
+    private void verifyOtelAttributes(Transaction transaction) {
+        Object s3keyAttribute = transaction.getOtelAttributes().get("aws.s3.key");
+        assertThat(s3keyAttribute).isInstanceOf(String.class).isEqualTo("b21b84d653bb07b05b1e6b33684dc11b");
+
+        Object s3bucketAttribute = transaction.getOtelAttributes().get("aws.s3.bucket");
+        assertThat(s3bucketAttribute).isInstanceOf(String.class).isEqualTo(S3_BUCKET_NAME);
+    }
 }