feat: added support for batch write dynamodb (#858)

instana · Aug 31, 2023 · a276b84 · a276b84
1 parent f7ff4ad
commit a276b84
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 1 deletion.
diff --git a/packages/collector/test/tracing/cloud/aws-sdk/v3/dynamodb/app.js b/packages/collector/test/tracing/cloud/aws-sdk/v3/dynamodb/app.js
@@ -120,6 +120,36 @@ const operationParams = {
         }
       }
     }
+  },
+  batchWriteItem: {
+    RequestItems: {
+      [tableName]: [
+        {
+          PutRequest: {
+            Item: {
+              year: {
+                N: '2002'
+              },
+              title: {
+                S: 'batch record 1'
+              }
+            }
+          }
+        },
+        {
+          PutRequest: {
+            Item: {
+              year: {
+                N: '2003'
+              },
+              title: {
+                S: 'batch record 2'
+              }
+            }
+          }
+        }
+      ]
+    }
   }
 };
 

diff --git a/packages/collector/test/tracing/cloud/aws-sdk/v3/dynamodb/test_definition.js b/packages/collector/test/tracing/cloud/aws-sdk/v3/dynamodb/test_definition.js
@@ -32,7 +32,8 @@ const availableOperations = [
   'getItem',
   'scan',
   'query',
-  'deleteItem'
+  'deleteItem',
+  'batchWriteItem'
 ];
 
 const retryTime = config.getTestTimeout() * 2;

diff --git a/packages/core/src/tracing/instrumentation/cloud/aws-sdk/v3/dynamodb.js b/packages/core/src/tracing/instrumentation/cloud/aws-sdk/v3/dynamodb.js
@@ -65,6 +65,8 @@ class InstanaAWSDynamoDB extends InstanaAWSProduct {
 
     if (params && params.TableName) {
       spanData.table = params.TableName;
+    } else if (params && params.RequestItems) {
+      spanData.table = Object.keys(params.RequestItems).toString();
     }
 
     return spanData;