Merge branch 'master' into alexeyk/fixed-flaky-pekko-test

Author: AlexeyKuznetsov-DD

benchmark/Dockerfile

@@ -1,14 +1,20 @@
 # Petclinic download and compilation stage
 FROM eclipse-temurin:17-jammy as petclinic
 
+ARG SPRING_PETCLINIC_COMMIT=cefaf55dd124d0635abfe857c3c99a3d3ea62017
+
 RUN apt-get update \
   && apt-get -y install git  \
   && apt-get -y clean \
   && rm -rf /var/lib/apt/lists/*
 
-RUN git clone --depth 1 --branch main --single-branch https://github.com/spring-projects/spring-petclinic.git \
-  && cd spring-petclinic \
-  && ./mvnw dependency:go-offline
+RUN set -eux;\
+  git init spring-petclinic;\
+  cd spring-petclinic;\
+  git remote add origin https://github.com/spring-projects/spring-petclinic.git;\
+  git fetch --depth 1 origin ${SPRING_PETCLINIC_COMMIT};\
+  git checkout ${SPRING_PETCLINIC_COMMIT};\
+  ./mvnw dependency:go-offline
 
 RUN cd spring-petclinic \
   && ./mvnw package -Dmaven.test.skip=true \

dd-java-agent/src/test/groovy/datadog/trace/agent/InstrumenterUnloadTest.groovy

@@ -26,16 +26,26 @@ class InstrumenterUnloadTest extends Specification {
       , ["DD_API_KEY": API_KEY]
       , new PrintStream(testOutput))
 
-    int unloadCount = 0
+    boolean canaryUnloaded = false
+    int unloadedInstrumentationCount = 0
     new ByteArrayInputStream((testOutput.toByteArray())).eachLine {
       System.out.println(it)
+      if (it =~ /(?i)unload.*Canary/) {
+        canaryUnloaded = true
+      }
       if (it =~ /(?i)unload.* datadog.trace.instrumentation./) {
-        unloadCount++
+        unloadedInstrumentationCount++
       }
     }
 
+    if (!canaryUnloaded) {
+      System.out.println("WARNING: Canary class was not unloaded!")
+    }
+
     then:
     returnCode == 0
-    unloadCount > 0
+    // skip check if we couldn't even unload our Canary class, as that
+    // indicates full GC didn't happen enough to trigger any unloading
+    !canaryUnloaded || unloadedInstrumentationCount > 0
   }
 }

dd-java-agent/src/test/java/jvmbootstraptest/UnloadingChecker.java

@@ -1,13 +1,32 @@
 package jvmbootstraptest;
 
+import static java.util.concurrent.TimeUnit.MINUTES;
+
 import datadog.trace.test.util.GCUtils;
+import java.lang.management.ClassLoadingMXBean;
+import java.lang.management.ManagementFactory;
 
 public class UnloadingChecker {
-  public static void main(final String[] args) {
-    try {
-      GCUtils.awaitGC();
-    } catch (InterruptedException e) {
-      e.printStackTrace();
+  static class Canary {}
+
+  public static void main(final String[] args) throws Exception {
+    ClassLoadingMXBean classLoadingMXBean = ManagementFactory.getClassLoadingMXBean();
+    long initialUnloadCount = classLoadingMXBean.getUnloadedClassCount();
+
+    // load an isolated class which we know can be unloaded after a full GC
+    new IsolatingClassLoader().loadClass("jvmbootstraptest.UnloadingChecker$Canary");
+
+    long waitNanos = MINUTES.toNanos(2);
+    long startNanos = System.nanoTime();
+
+    while (System.nanoTime() - startNanos < waitNanos) {
+      try {
+        GCUtils.awaitGC();
+      } catch (Throwable ignore) {
+      }
+      if (initialUnloadCount < classLoadingMXBean.getUnloadedClassCount()) {
+        break; // some class unloading has taken place, stop and check results
+      }
     }
   }
 }

dd-smoke-tests/jboss-modules/build.gradle

@@ -1,4 +1,9 @@
 
+ext {
+  // see https://datadoghq.atlassian.net/wiki/x/H4S2NQE
+  excludeJdk = ['IBM8']
+}
+
 apply from: "$rootDir/gradle/java.gradle"
 description = 'JBoss Modules Smoke Tests.'
 

utils/test-utils/src/main/java/datadog/trace/test/util/GCUtils.java

@@ -22,6 +22,8 @@ public static void awaitGC(final WeakReference<?> ref) throws InterruptedExcepti
 
   public static void awaitGC(final WeakReference<?> ref, final long duration, final TimeUnit unit)
       throws InterruptedException {
+    System.gc();
+    System.runFinalization();
     final long waitNanos = unit.toNanos(duration);
     final long start = System.nanoTime();
     while (System.nanoTime() - start < waitNanos) {