[#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();
  }