feat: save function tags in auto trace

includable · Sep 7, 2024 · 65c9578 · 65c9578
1 parent bfee4c4
commit 65c9578
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/packages/api/serverless.yml b/packages/api/serverless.yml
@@ -79,6 +79,7 @@ provider:
             - lambda:ListFunctions
             - lambda:UpdateFunctionConfiguration
             - lambda:GetLayerVersion
+            - lambda:ListTags
           Resource:
             - "*"
         - Effect: Allow

diff --git a/packages/api/src/events/auto-trace.js b/packages/api/src/events/auto-trace.js
@@ -1,6 +1,7 @@
 import {
   LambdaClient,
   ListFunctionsCommand,
+  ListTagsCommand,
   UpdateFunctionConfigurationCommand,
 } from "@aws-sdk/client-lambda";
 import {
@@ -77,7 +78,25 @@ const updateLambda = async (lambda, arnBase, edgeEndpoint) => {
   logger.info(res);
 };
 
+const getLambdaTags = async (lambda) => {
+  const lambdaClient = new LambdaClient();
+  const { Tags } = await lambdaClient.send(
+    new ListTagsCommand({
+      Resource: lambda.FunctionArn,
+    }),
+  );
+
+  return Tags;
+};
+
 const saveFunctionInfo = async (lambda, traceStatus) => {
+  let tags;
+  try {
+    tags = await getLambdaTags(lambda);
+  } catch (e) {
+    logger.warn(`Failed to get tags for ${lambda.FunctionName}`, e);
+  }
+
   await put(
     {
       pk: `function#${process.env.AWS_REGION}#${lambda.FunctionName}`,
@@ -90,6 +109,7 @@ const saveFunctionInfo = async (lambda, traceStatus) => {
       memoryAllocated: lambda.MemorySize,
       timeout: lambda.Timeout * 1000,
       traceStatus,
+      tags: tags || {},
     },
     true,
   );