Attempt to catch dump via Gradle

dd-java-agent/instrumentation/java-concurrent/java-concurrent-21/build.gradle

@@ -1,3 +1,6 @@
+import java.util.concurrent.Executors
+import java.util.concurrent.TimeUnit
+
 ext {
   minJavaVersionForTests = JavaVersion.VERSION_21
   // Structured concurrency is a preview feature in Java 21. Methods (e.g. ShutdownOnFailure) used in this instrumentation test are no longer available in Java 25, so we set the max version to 24.
@@ -58,3 +61,65 @@ compileJava.configure {
   sourceCompatibility = JavaVersion.VERSION_1_8
   targetCompatibility = JavaVersion.VERSION_1_8
 }
+
+tasks.withType(Test).configureEach {
+  doFirst {
+    // This manipulation is requred to move files into folder
+    // that will be collected by `collect_reports.sh` script.
+    String buildDir = layout.buildDirectory.asFile.get().absolutePath.replace("dd-trace-java/dd-java-agent",
+      "dd-trace-java/workspace/dd-java-agent")
+
+    def dupmsDir = new File(buildDir, "dumps")
+    dupmsDir.mkdirs()
+
+    // Scheduler to execute logic that will dump threads and heap
+    def scheduler = Executors.newSingleThreadScheduledExecutor({ r ->
+      Thread t = new Thread(r, "dump-scheduler")
+      t.setDaemon(true)
+      return t
+    })
+
+    // Schedule the dump job:
+    def future = scheduler.schedule({
+      try {
+        def threadDumpFile = new File(dupmsDir, "${System.currentTimeMillis()}-thread-dump.log")
+
+        new ProcessBuilder("jcmd", "0", "Thread.print", "-l")
+          .redirectErrorStream(true)
+          .redirectOutput(threadDumpFile)
+          .start().waitFor()
+
+        // Collect PIDs of all Java processes:
+        def jvmProcesses = "jcmd -l".execute().text.readLines()
+        jvmProcesses.each { p ->
+          // Process only test executors.
+          if (!p.contains("Gradle Test Executor")) {
+            return
+          }
+
+          def pid = p.substring(0, p.indexOf(' '))
+          def heapDumpFile = "${dupmsDir.absolutePath}/${System.currentTimeMillis()}-${pid}.hprof"
+          def cmd = "jcmd ${pid} GC.heap_dump ${heapDumpFile}"
+          cmd.execute().waitFor()
+        }
+      } catch (Throwable t) {
+        logger.warn("Dumping failed: ${t.message}", t)
+      } finally {
+        scheduler.shutdown()
+      }
+    }, 60, TimeUnit.SECONDS) // You can adjust this line with required delay.
+
+    // Store handles for cancellation in doLast:
+    ext.dumpFuture = future
+    ext.dumpScheduler = scheduler
+  }
+
+  doLast {
+    // Cancel if task completed faster than specified timeout.
+    try {
+      ext.dumpFuture?.cancel(false)
+    } finally {
+      ext.dumpScheduler?.shutdownNow()
+    }
+  }
+}

dd-java-agent/instrumentation/java-concurrent/java-concurrent-21/src/previewTest/groovy/StructuredConcurrencyTest.groovy

@@ -9,12 +9,6 @@ import static datadog.trace.agent.test.utils.TraceUtils.runnableUnderTrace
 import static java.time.Instant.now
 
 class StructuredConcurrencyTest extends AgentTestRunner {
-  @Override
-  boolean useStrictTraceWrites() {
-    // TODO: Monitor in CI to validate fix effectiveness against freezes.
-    return false
-  }
-
   /**
    * Tests the structured task scope with a single task.
    */

dd-java-agent/instrumentation/lettuce-4/build.gradle

@@ -1,3 +1,5 @@
+import java.util.concurrent.Executors
+import java.util.concurrent.TimeUnit
 
 muzzle {
   pass {
@@ -20,3 +22,65 @@ dependencies {
 
   latestDepTestImplementation group: 'biz.paluch.redis', name: 'lettuce', version: '4.+'
 }
+
+tasks.withType(Test).configureEach {
+  doFirst {
+    // This manipulation is requred to move files into folder
+    // that will be collected by `collect_reports.sh` script.
+    String buildDir = layout.buildDirectory.asFile.get().absolutePath.replace("dd-trace-java/dd-java-agent",
+      "dd-trace-java/workspace/dd-java-agent")
+
+    def dupmsDir = new File(buildDir, "dumps")
+    dupmsDir.mkdirs()
+
+    // Scheduler to execute logic that will dump threads and heap
+    def scheduler = Executors.newSingleThreadScheduledExecutor({ r ->
+      Thread t = new Thread(r, "dump-scheduler")
+      t.setDaemon(true)
+      return t
+    })
+
+    // Schedule the dump job:
+    def future = scheduler.schedule({
+      try {
+        def threadDumpFile = new File(dupmsDir, "${System.currentTimeMillis()}-thread-dump.log")
+
+        new ProcessBuilder("jcmd", "0", "Thread.print", "-l")
+          .redirectErrorStream(true)
+          .redirectOutput(threadDumpFile)
+          .start().waitFor()
+
+        // Collect PIDs of all Java processes:
+        def jvmProcesses = "jcmd -l".execute().text.readLines()
+        jvmProcesses.each { p ->
+          // Process only test executors.
+          if (!p.contains("Gradle Test Executor")) {
+            return
+          }
+
+          def pid = p.substring(0, p.indexOf(' '))
+          def heapDumpFile = "${dupmsDir.absolutePath}/${System.currentTimeMillis()}-${pid}.hprof"
+          def cmd = "jcmd ${pid} GC.heap_dump ${heapDumpFile}"
+          cmd.execute().waitFor()
+        }
+      } catch (Throwable t) {
+        logger.warn("Dumping failed: ${t.message}", t)
+      } finally {
+        scheduler.shutdown()
+      }
+    }, 60, TimeUnit.SECONDS) // You can adjust this line with required delay.
+
+    // Store handles for cancellation in doLast:
+    ext.dumpFuture = future
+    ext.dumpScheduler = scheduler
+  }
+
+  doLast {
+    // Cancel if task completed faster than specified timeout.
+    try {
+      ext.dumpFuture?.cancel(false)
+    } finally {
+      ext.dumpScheduler?.shutdownNow()
+    }
+  }
+}

dd-java-agent/instrumentation/lettuce-4/src/test/groovy/Lettuce4ClientTestBase.groovy

@@ -83,10 +83,4 @@ abstract class Lettuce4ClientTestBase extends VersionedNamingTestBase {
     redisClient.shutdown()
     redisServer.stop()
   }
-
-  @Override
-  boolean useStrictTraceWrites() {
-    // TODO: Monitor in CI to validate fix effectiveness against freezes.
-    return false
-  }
 }