Flush pending traces synchronously before terminating child process

nikita-tkachenko-datadog · nikita-tkachenko-datadog · commit cab30685b760 · 2024-04-02T13:44:58.000+02:00
diff --git a/dd-java-agent/agent-ci-visibility/src/main/java/datadog/trace/civisibility/domain/buildsystem/ProxyTestSession.java b/dd-java-agent/agent-ci-visibility/src/main/java/datadog/trace/civisibility/domain/buildsystem/ProxyTestSession.java
@@ -4,6 +4,7 @@
 import datadog.trace.api.civisibility.config.ModuleExecutionSettings;
 import datadog.trace.api.civisibility.coverage.CoverageDataSupplier;
 import datadog.trace.api.civisibility.telemetry.CiVisibilityMetricCollector;
+import datadog.trace.bootstrap.instrumentation.api.AgentTracer;
 import datadog.trace.civisibility.codeowners.Codeowners;
 import datadog.trace.civisibility.coverage.CoverageProbeStoreFactory;
 import datadog.trace.civisibility.decorator.TestDecorator;
@@ -63,7 +64,10 @@ public ProxyTestSession(
 
   @Override
   public void end(Long startTime) {
-    // no op
+    // flushing written traces synchronously:
+    // as soon as all tests have been executed,
+    // the process can be killed by the build system
+    AgentTracer.get().flush();
   }
 
   @Override
