Collect process tags for profiling upload requests (#8780)

Author: amarziali

dd-java-agent/agent-profiling/profiling-uploader/src/main/java/com/datadog/profiling/uploader/ProfileUploader.java

@@ -27,6 +27,7 @@
 import datadog.trace.api.Config;
 import datadog.trace.api.DDTags;
 import datadog.trace.api.Platform;
+import datadog.trace.api.ProcessTags;
 import datadog.trace.api.git.GitInfo;
 import datadog.trace.api.git.GitInfoProvider;
 import datadog.trace.api.profiling.RecordingData;
@@ -469,13 +470,18 @@ public void toJson(JsonWriter writer, RecordingData recordingData) throws IOExce
       if (recordingData == null) {
         return;
       }
+      final CharSequence processTags = ProcessTags.getTagsForSerialization();
       writer.beginObject();
       writer.name("attachments");
       writer.beginArray();
       writer.value(V4_ATTACHMENT_FILENAME);
       writer.endArray();
       writer.name(V4_PROFILE_TAGS_PARAM);
       writer.value(tags + ",snapshot:" + recordingData.getKind().name().toLowerCase(Locale.ROOT));
+      if (processTags != null) {
+        writer.name("process_tags");
+        writer.value(processTags.toString());
+      }
       writer.name(V4_PROFILE_START_PARAM);
       writer.value(recordingData.getStart().toString());
       writer.name(V4_PROFILE_END_PARAM);

dd-java-agent/agent-profiling/profiling-uploader/src/test/java/com/datadog/profiling/uploader/ProfileUploaderTest.java

@@ -43,6 +43,7 @@
 import datadog.trace.api.Config;
 import datadog.trace.api.DDTags;
 import datadog.trace.api.Platform;
+import datadog.trace.api.ProcessTags;
 import datadog.trace.api.profiling.ProfilingSnapshot;
 import datadog.trace.api.profiling.RecordingData;
 import datadog.trace.api.profiling.RecordingInputStream;
@@ -830,6 +831,34 @@ public void testShutdown() throws Exception {
     verify(recording).release();
   }
 
+  @ParameterizedTest(name = "process tags enabled ''{0}''")
+  @ValueSource(booleans = {true, false})
+  public void testRequestWithProcessTags(boolean processTagsEnabled) throws Exception {
+    when(config.isExperimentalPropagateProcessTagsEnabled()).thenReturn(processTagsEnabled);
+    ProcessTags.reset(config);
+    uploader =
+        new ProfileUploader(
+            config, configProvider, ioLogger, (int) TERMINATION_TIMEOUT.getSeconds());
+
+    server.enqueue(new MockResponse().setResponseCode(200));
+    uploadAndWait(RECORDING_TYPE, mockRecordingData());
+
+    final RecordedRequest recordedRequest = server.takeRequest(5, TimeUnit.SECONDS);
+    assertNotNull(recordedRequest);
+    final List<FileItem> multiPartItems =
+        FileUpload.parse(
+            recordedRequest.getBody().readByteArray(), recordedRequest.getHeader("Content-Type"));
+
+    final FileItem rawEvent = multiPartItems.get(0);
+    final Map<String, ?> parsed = new ObjectMapper().readValue(rawEvent.get(), Map.class);
+    if (processTagsEnabled) {
+      assertNotNull(ProcessTags.getTagsForSerialization());
+      assertEquals(ProcessTags.getTagsForSerialization().toString(), parsed.get("process_tags"));
+    } else {
+      assertNull(parsed.get("process_tags"));
+    }
+  }
+
   private RecordingData mockRecordingData() throws IOException {
     return mockRecordingData(false, ProfilingSnapshot.Kind.PERIODIC);
   }

internal-api/src/main/java/datadog/trace/api/ProcessTags.java

@@ -147,9 +147,14 @@ static void empty() {
 
   /** Visible for testing. */
   static void reset() {
+    reset(Config.get());
+  }
+
+  /** Visible for testing. */
+  public static void reset(Config config) {
     synchronized (Lazy.TAGS) {
       empty();
-      enabled = Config.get().isExperimentalPropagateProcessTagsEnabled();
+      enabled = config.isExperimentalPropagateProcessTagsEnabled();
       Lazy.TAGS.putAll(Lazy.loadTags());
     }
   }