Extract git tags from embedded git.properties and datadog_git.properties

Author: wmouchere

dd-java-agent/agent-bootstrap/src/main/java/datadog/trace/bootstrap/Agent.java

@@ -30,6 +30,8 @@
 import datadog.trace.api.config.UsmConfig;
 import datadog.trace.api.gateway.RequestContextSlot;
 import datadog.trace.api.gateway.SubscriptionService;
+import datadog.trace.api.git.EmbeddedGitInfoBuilder;
+import datadog.trace.api.git.GitInfoProvider;
 import datadog.trace.api.profiling.ProfilingEnablement;
 import datadog.trace.api.scopemanager.ScopeListener;
 import datadog.trace.bootstrap.benchmark.StaticEventLogger;
@@ -223,6 +225,12 @@ public static void start(
       }
     }
 
+    // Enable automatic fetching of git tags from datadog_git.properties only if CI Visibility is
+    // not enabled
+    if (!ciVisibilityEnabled) {
+      GitInfoProvider.INSTANCE.registerGitInfoBuilder(new EmbeddedGitInfoBuilder());
+    }
+
     boolean dataJobsEnabled = isFeatureEnabled(AgentFeature.DATA_JOBS);
     if (dataJobsEnabled) {
       log.info("Data Jobs Monitoring enabled, enabling spark integrations");

internal-api/src/main/java/datadog/trace/api/git/EmbeddedGitInfoBuilder.java

@@ -2,6 +2,8 @@
 
 import java.io.IOException;
 import java.io.InputStream;
+import java.util.Arrays;
+import java.util.List;
 import java.util.Properties;
 import javax.annotation.Nullable;
 import org.slf4j.Logger;
@@ -11,29 +13,55 @@ public class EmbeddedGitInfoBuilder implements GitInfoBuilder {
 
   private static final Logger log = LoggerFactory.getLogger(EmbeddedGitInfoBuilder.class);
 
-  private static final String EMBEDDED_GIT_PROPERTIES_FILE_NAME = "git.properties";
+  // Spring boot fat jars and wars should have the git.properties file in a specific path,
+  // guaranteeing that it's not coming from a dependency
+  private static final String SPRING_BOOT_JAR_EMBEDDED_GIT_PROPERTIES_FILE_NAME =
+      "BOOT-INF/classes/git.properties";
+  private static final String SPRING_BOOT_JAR_EMBEDDED_DATADOG_GIT_PROPERTIES_FILE_NAME =
+      "BOOT-INF/classes/datadog_git.properties";
+  private static final String WAR_EMBEDDED_GIT_PROPERTIES_FILE_NAME =
+      "WEB-INF/classes/git.properties";
+  private static final String WAR_EMBEDDED_DATADOG_GIT_PROPERTIES_FILE_NAME =
+      "WEB-INF/classes/datadog_git.properties";
+  // If we can't find the files above, probably because we're not in a spring context, we can look
+  // at the root of the classpath.
+  // Since it could be tainted by dependencies, we're looking for a specific datadog_git.properties
+  // file.
+  private static final String EMBEDDED_DATADOOG_GIT_PROPERTIES_FILE_NAME = "datadog_git.properties";
 
-  private final String resourceName;
+  private final List<String> resourceNames;
 
   public EmbeddedGitInfoBuilder() {
-    this(EMBEDDED_GIT_PROPERTIES_FILE_NAME);
+    // Order is important here, from the most reliable sources to the least reliable ones
+    this(
+        Arrays.asList(
+            SPRING_BOOT_JAR_EMBEDDED_GIT_PROPERTIES_FILE_NAME,
+            SPRING_BOOT_JAR_EMBEDDED_DATADOG_GIT_PROPERTIES_FILE_NAME,
+            WAR_EMBEDDED_GIT_PROPERTIES_FILE_NAME,
+            WAR_EMBEDDED_DATADOG_GIT_PROPERTIES_FILE_NAME,
+            EMBEDDED_DATADOOG_GIT_PROPERTIES_FILE_NAME));
   }
 
-  EmbeddedGitInfoBuilder(String resourceName) {
-    this.resourceName = resourceName;
+  EmbeddedGitInfoBuilder(List<String> resourceNames) {
+    this.resourceNames = resourceNames;
   }
 
   @Override
   public GitInfo build(@Nullable String repositoryPath) {
     Properties gitProperties = new Properties();
-    try (InputStream is = ClassLoader.getSystemResourceAsStream(resourceName)) {
-      if (is != null) {
-        gitProperties.load(is);
-      } else {
-        log.debug("Could not find embedded Git properties resource: {}", resourceName);
+
+    for (String resourceName : resourceNames) {
+      try (InputStream is = ClassLoader.getSystemResourceAsStream(resourceName)) {
+        if (is != null) {
+          gitProperties.load(is);
+          // stop at the first git properties file found
+          break;
+        } else {
+          log.debug("Could not find embedded Git properties resource: {}", resourceName);
+        }
+      } catch (IOException e) {
+        log.error("Error reading embedded Git properties from {}", resourceName, e);
       }
-    } catch (IOException e) {
-      log.error("Error reading embedded Git properties from {}", resourceName, e);
     }
 
     String commitSha = gitProperties.getProperty("git.commit.id");