Route detection Support for netty reactor

newrelic · Jul 3, 2024 · 3458dd1 · 3458dd1
1 parent 76aeff3
commit 3458dd1
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/...r-0.7.0/src/main/java/reactor/ipc/netty/http/server/HttpRouteHandler_Instrumentation.java b/...r-0.7.0/src/main/java/reactor/ipc/netty/http/server/HttpRouteHandler_Instrumentation.java
@@ -1,5 +1,9 @@
 package reactor.ipc.netty.http.server;
 
+import com.newrelic.api.agent.security.NewRelicSecurity;
+import com.newrelic.api.agent.security.instrumentation.helpers.URLMappingsHelper;
+import com.newrelic.api.agent.security.schema.Framework;
+import com.newrelic.api.agent.security.schema.HttpRequest;
 import com.newrelic.api.agent.weaver.Weave;
 import com.newrelic.api.agent.weaver.Weaver;
 import org.reactivestreams.Publisher;
@@ -9,8 +13,17 @@
 @Weave(originalName = "reactor.ipc.netty.http.server.DefaultHttpServerRoutes$HttpRouteHandler")
 final class HttpRouteHandler_Instrumentation {
     final Predicate<? super HttpServerRequest> condition = Weaver.callOriginal();
+
     public Publisher<Void> apply(HttpServerRequest request, HttpServerResponse response) {
-        // TODO: Calculate route for endpoints
+        if (NewRelicSecurity.isHookProcessingActive()){
+            HttpRequest securityRequest = NewRelicSecurity.getAgent().getSecurityMetaData().getRequest();
+            if (condition instanceof HttpPredicate){
+                securityRequest.setRoute(((HttpPredicate) condition).uri);
+            } else {
+                securityRequest.setRoute(URLMappingsHelper.WILDCARD);
+            }
+            NewRelicSecurity.getAgent().getSecurityMetaData().getMetaData().setFramework(Framework.NETTY_REACTOR);
+        }
         return Weaver.callOriginal();
     }
 }
diff --git a/...actor-0.8.0/src/main/java/reactor/netty/http/server/HttpRouteHandler_Instrumentation.java b/...actor-0.8.0/src/main/java/reactor/netty/http/server/HttpRouteHandler_Instrumentation.java
@@ -1,5 +1,9 @@
 package reactor.netty.http.server;
 
+import com.newrelic.api.agent.security.NewRelicSecurity;
+import com.newrelic.api.agent.security.instrumentation.helpers.URLMappingsHelper;
+import com.newrelic.api.agent.security.schema.Framework;
+import com.newrelic.api.agent.security.schema.HttpRequest;
 import com.newrelic.api.agent.weaver.Weave;
 import com.newrelic.api.agent.weaver.Weaver;
 import org.reactivestreams.Publisher;
@@ -9,8 +13,17 @@
 @Weave(originalName = "reactor.netty.http.server.DefaultHttpServerRoutes$HttpRouteHandler")
 final class HttpRouteHandler_Instrumentation {
     final Predicate<? super HttpServerRequest> condition = Weaver.callOriginal();
+
     public Publisher<Void> apply(HttpServerRequest request, HttpServerResponse response) {
-        // TODO: Calculate route for endpoints
+        if (NewRelicSecurity.isHookProcessingActive()){
+            HttpRequest securityRequest = NewRelicSecurity.getAgent().getSecurityMetaData().getRequest();
+            if (condition instanceof HttpPredicate){
+                securityRequest.setRoute(((HttpPredicate) condition).uri);
+            } else {
+                securityRequest.setRoute(URLMappingsHelper.WILDCARD);
+            }
+            NewRelicSecurity.getAgent().getSecurityMetaData().getMetaData().setFramework(Framework.NETTY_REACTOR);
+        }
         return Weaver.callOriginal();
     }
 }