feat(RD-12769): add execution tags

src/main/java/io/lumigo/core/SpansContainer.java

@@ -9,13 +9,16 @@
 import io.lumigo.core.parsers.v1.AwsSdkV1ParserFactory;
 import io.lumigo.core.parsers.v2.AwsSdkV2ParserFactory;
 import io.lumigo.core.utils.AwsUtils;
+import io.lumigo.core.utils.ExecutionTags;
 import io.lumigo.core.utils.JsonUtils;
 import io.lumigo.core.utils.StringUtils;
 import io.lumigo.models.HttpSpan;
 import io.lumigo.models.Span;
 import java.io.*;
 import java.util.*;
 import java.util.concurrent.Callable;
+import java.util.stream.Collectors;
+
 import org.apache.http.Header;
 import org.apache.http.HttpEntity;
 import org.apache.http.HttpResponse;
@@ -92,8 +95,8 @@ public void init(Map<String, String> env, Reporter reporter, Context context, Ob
                         .maxFinishTime(
                                 startTime
                                         + ((context.getRemainingTimeInMillis() > 0)
-                                                ? context.getRemainingTimeInMillis()
-                                                : MAX_LAMBDA_TIME))
+                                        ? context.getRemainingTimeInMillis()
+                                        : MAX_LAMBDA_TIME))
                         .transactionId(AwsUtils.extractAwsTraceTransactionId(awsTracerId))
                         .info(
                                 Span.Info.builder()
@@ -148,7 +151,7 @@ public void init(Map<String, String> env, Reporter reporter, Context context, Ob
                         .event(
                                 Configuration.getInstance().isLumigoVerboseMode()
                                         ? JsonUtils.getObjectAsJsonString(
-                                                EventParserFactory.parseEvent(event))
+                                        EventParserFactory.parseEvent(event))
                                         : null)
                         .build();
     }
@@ -198,12 +201,17 @@ public void end() throws IOException {
     }
 
     private void end(Span endFunctionSpan) throws IOException {
+        List<Map<String, String>> executionTags = ExecutionTags.getTags();
         this.endFunctionSpan =
                 endFunctionSpan
                         .toBuilder()
                         .reporter_rtt(rttDuration)
                         .ended(System.currentTimeMillis())
                         .id(this.baseSpan.getId())
+                        .info(
+                                endFunctionSpan.getInfo().toBuilder()
+                                        .tags(executionTags)
+                                        .build())
                         .build();
         reporter.reportSpans(
                 prepareToSend(getAllCollectedSpans(), endFunctionSpan.getError() != null),
@@ -421,8 +429,7 @@ public void addHttpSpan(
                                                                                 context
                                                                                         .response())))
                                                 .statusCode(context.httpResponse().statusCode())
-                                                .build())
-                                .build());
+                                                .build());
 
         Logger.debug(
                 "Trying to extract aws custom properties for service: "
@@ -577,4 +584,16 @@ public Object reduceSpanSize(Object span, boolean hasError) {
         }
         return span;
     }
+
+    public void addExecutionTag(String key, String value) {
+        getInstance().addExecutionTag(key, value);
+    }
+
+    public void clearExecutionTags() {
+        getInstance().clearExecutionTags();
+    }
+
+    public Map<String, String> getExecutionTags() {
+        return getInstance().getExecutionTags();
+    }
 }

src/main/java/io/lumigo/core/utils/ExecutionTags.java

@@ -0,0 +1,73 @@
+package io.lumigo.core.utils;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.HashMap;
+import org.pmw.tinylog.Logger;
+
+public class ExecutionTags {
+    private static final int MAX_TAG_KEY_LEN = 100;
+    private static final int MAX_TAG_VALUE_LEN = 100;
+    private static final int MAX_TAGS = 50;
+    private static final String ADD_TAG_ERROR_MSG_PREFIX = "Error adding tag";
+
+    private static final List<Map<String, String>> tags = new ArrayList<>();
+
+    private static boolean validateTag(String key, String value, boolean shouldLogErrors) {
+        key = String.valueOf(key);
+        value = String.valueOf(value);
+        if (key.isEmpty() || key.length() > MAX_TAG_KEY_LEN) {
+            if (shouldLogErrors) {
+                Logger.error(String.format("%s: key length should be between 1 and %d: %s - %s",
+                        ADD_TAG_ERROR_MSG_PREFIX, MAX_TAG_KEY_LEN, key, value));
+            }
+            return false;
+        }
+        if (value.isEmpty() || value.length() > MAX_TAG_VALUE_LEN) {
+            if (shouldLogErrors) {
+                Logger.error(String.format("%s: value length should be between 1 and %d: %s - %s",
+                        ADD_TAG_ERROR_MSG_PREFIX, MAX_TAG_VALUE_LEN, key, value));
+            }
+            return false;
+        }
+        if (tags.size() >= MAX_TAGS) {
+            if (shouldLogErrors) {
+                Logger.error(String.format("%s: maximum number of tags is %d: %s - %s",
+                        ADD_TAG_ERROR_MSG_PREFIX, MAX_TAGS, key, value));
+            }
+            return false;
+        }
+        return true;
+    }
+
+    private static String normalizeTag(Object val) {
+        return (val == null) ? null : String.valueOf(val);
+    }
+
+    public static void addTag(String key, String value, boolean shouldLogErrors) {
+        try {
+            Logger.info(String.format("Adding tag: %s - %s", key, value));
+            if (!validateTag(key, value, shouldLogErrors)) {
+                return;
+            }
+            Map<String, String> tag = new HashMap<>();
+            tag.put("key", normalizeTag(key));
+            tag.put("value", normalizeTag(value));
+            tags.add(tag);
+        } catch (Exception err) {
+            if (shouldLogErrors) {
+                Logger.error(ADD_TAG_ERROR_MSG_PREFIX);
+            }
+            Logger.error(err.getMessage());
+        }
+    }
+
+    public static List<Map<String, String>> getTags() {
+        return new ArrayList<>(tags);
+    }
+
+    public static void clear() {
+        tags.clear();
+    }
+}

src/main/java/io/lumigo/handlers/LumigoRequestHandler.java

@@ -7,6 +7,8 @@
 import io.lumigo.core.instrumentation.agent.Installer;
 import io.lumigo.core.network.Reporter;
 import io.lumigo.core.utils.EnvUtil;
+
+import java.util.Map;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.Future;
@@ -73,4 +75,16 @@ public OUTPUT handleRequest(INPUT input, Context context) {
     }
 
     public abstract OUTPUT doHandleRequest(INPUT input, Context context);
+
+    public void addExecutionTag(String key, String value) {
+        spansContainer.addExecutionTag(key, value);
+    }
+
+    public void clearExecutionTags() {
+        spansContainer.clearExecutionTags();
+    }
+
+    public Map<String, String> getExecutionTags() {
+        return spansContainer.getExecutionTags();
+    }
 }

src/main/java/io/lumigo/models/Span.java

@@ -3,6 +3,8 @@
 import com.fasterxml.jackson.annotation.JsonProperty;
 import java.util.List;
 import java.util.Locale;
+import java.util.Map;
+
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
@@ -49,6 +51,7 @@ public static class Info {
         private String stage;
         private String messageId;
         private List<String> messageIds;
+        private List<Map<String, String>> tags;
         private long approxEventCreationTime;
     }