[#11210] Update DispatcherHandler.handleResult() of spring webflux 6

pinpoint-apm · Jul 3, 2024 · 887ec53 · 887ec53
1 parent ae66fc2
commit 887ec53
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...bflux/src/main/java/com/navercorp/pinpoint/plugin/spring/webflux/SpringWebFluxPlugin.java b/...bflux/src/main/java/com/navercorp/pinpoint/plugin/spring/webflux/SpringWebFluxPlugin.java
@@ -123,11 +123,21 @@ public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, Strin
             if (invokerHandlerMethod != null) {
                 invokerHandlerMethod.addInterceptor(DispatchHandlerInvokeHandlerMethodInterceptor.class, va(this.uriStatEnable, Boolean.valueOf(false)));
             }
+            // 6.x
+            final InstrumentMethod handleRequestWithMethod = target.getDeclaredMethod("handleRequestWith", "org.springframework.web.server.ServerWebExchange", "java.lang.Object");
+            if (handleRequestWithMethod != null) {
+                handleRequestWithMethod.addInterceptor(DispatchHandlerInvokeHandlerMethodInterceptor.class, va(this.uriStatEnable, Boolean.valueOf(false)));
+            }
             // Result
             final InstrumentMethod handleResultMethod = target.getDeclaredMethod("handleResult", "org.springframework.web.server.ServerWebExchange", "org.springframework.web.reactive.HandlerResult");
             if (handleResultMethod != null) {
                 handleResultMethod.addInterceptor(DispatchHandlerInvokeHandlerMethodInterceptor.class, va(this.uriStatEnable, this.uriStatUseUserInput));
             }
+            // 6.x
+            final InstrumentMethod handleResultMethod2 = target.getDeclaredMethod("handleResult", "org.springframework.web.server.ServerWebExchange", "org.springframework.web.reactive.HandlerResult", "java.lang.String");
+            if (handleResultMethod2 != null) {
+                handleResultMethod2.addInterceptor(DispatchHandlerInvokeHandlerMethodInterceptor.class, va(this.uriStatEnable, this.uriStatUseUserInput));
+            }
 
             return target.toBytecode();
         }