minor changes

DataDog · Jun 21, 2024 · 60d33b8 · 60d33b8
1 parent 4380072
commit 60d33b8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/dd-java-agent/appsec/src/main/java/com/datadog/appsec/gateway/GatewayBridge.java b/dd-java-agent/appsec/src/main/java/com/datadog/appsec/gateway/GatewayBridge.java
@@ -172,8 +172,8 @@ public void init() {
                 }
               }
             } else if (hasUserTrackingEvent(traceSeg)) {
-                // Report all collected request headers on user tracking event
-                writeRequestHeaders(traceSeg, REQUEST_HEADERS_ALLOW_LIST, ctx.getRequestHeaders());
+              // Report all collected request headers on user tracking event
+              writeRequestHeaders(traceSeg, REQUEST_HEADERS_ALLOW_LIST, ctx.getRequestHeaders());
             } else {
               // Report minimum set of collected request headers
               writeRequestHeaders(

diff --git a/dd-java-agent/appsec/src/main/java/com/datadog/appsec/powerwaf/PowerWAFModule.java b/dd-java-agent/appsec/src/main/java/com/datadog/appsec/powerwaf/PowerWAFModule.java
@@ -457,7 +457,7 @@ public void onDataAvailable(
         Collection<AppSecEvent> events = buildEvents(resultWithData);
 
         if (!events.isEmpty() && !reqCtx.isThrottled(rateLimiter)) {
-          AgentSpan activeSpan = AgentTracer.activeSpan();
+          AgentSpan activeSpan = AgentTracer.get().activeSpan();
           if (activeSpan != null) {
             if (log.isDebugEnabled()) {
               log.debug("Setting force-keep tag on the current span");

diff --git a/...agent/appsec/src/test/groovy/com/datadog/appsec/gateway/GatewayBridgeSpecification.groovy b/...agent/appsec/src/test/groovy/com/datadog/appsec/gateway/GatewayBridgeSpecification.groovy
@@ -171,7 +171,6 @@ class GatewayBridgeSpecification extends DDSpecification {
     1 * mockAppSecCtx.isThrottled(_) >> true
     0 * spanInfo.getTags()
     0 * traceSegment.setDataTop("appsec", _)
-
   }
 
   void 'actor ip calculated from headers'() {