Fixes opentracing-contrib#135 : add ContainerRequestContext to decora…

…teResponse arguments.
hypnoce · Jan 5, 2020 · ffd154d · ffd154d
1 parent 32ca3b8
commit ffd154d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...racing-jaxrs2/src/main/java/io/opentracing/contrib/jaxrs2/server/ServerSpanDecorator.java b/...racing-jaxrs2/src/main/java/io/opentracing/contrib/jaxrs2/server/ServerSpanDecorator.java
@@ -22,10 +22,11 @@ public interface ServerSpanDecorator {
     /**
      * Decorate spans by outgoing object.
      *
+     * @param requestContext
      * @param responseContext
      * @param span
      */
-    void decorateResponse(ContainerResponseContext responseContext, Span span);
+    void decorateResponse(ContainerRequestContext requestContext, ContainerResponseContext responseContext, Span span);
 
     /**
      * Adds standard tags: {@link io.opentracing.tag.Tags#SPAN_KIND},
@@ -45,7 +46,7 @@ public void decorateRequest(ContainerRequestContext requestContext, Span span) {
         }
 
         @Override
-        public void decorateResponse(ContainerResponseContext responseContext, Span span) {
+        public void decorateResponse(ContainerRequestContext requestContext, ContainerResponseContext responseContext, Span span) {
             Tags.HTTP_STATUS.set(span, responseContext.getStatus());
         }
     };

diff --git a/...racing-jaxrs2/src/main/java/io/opentracing/contrib/jaxrs2/server/ServerTracingFilter.java b/...racing-jaxrs2/src/main/java/io/opentracing/contrib/jaxrs2/server/ServerTracingFilter.java
@@ -121,7 +121,7 @@ public void filter(ContainerRequestContext requestContext,
 
         if (spanDecorators != null) {
             for (ServerSpanDecorator decorator: spanDecorators) {
-                decorator.decorateResponse(responseContext, spanWrapper.get());
+                decorator.decorateResponse(requestContext, responseContext, spanWrapper.get());
             }
         }
     }