support for MVC

Author: dhilpipre

settings.gradle

@@ -1,2 +1,3 @@
 rootProject.name = 'java-instrumentation-template'
 include 'spring-cloud-gateway-reactive'
+include 'spring-cloud-gateway-mvc-4.1'

spring-cloud-gateway-mvc-4.1/build.gradle

@@ -0,0 +1,27 @@
+
+// Build.gradle generated for instrumentation module spring-cloud-gateway-mvc
+
+apply plugin: 'java'
+
+sourceCompatibility = JavaVersion.VERSION_17
+targetCompatibility = JavaVersion.VERSION_17
+
+dependencies {
+	implementation 'org.springframework.cloud:spring-cloud-gateway-server-mvc:4.1.0'
+    implementation 'org.springframework:spring-webmvc:5.3.16'
+
+   implementation fileTree(include: ['*.jar'], dir: '../libs')
+}
+
+jar {
+  manifest {
+    attributes 'Implementation-Title': 'com.newrelic.instrumentation.labs.spring-cloud-gateway-mvc-4.1'
+    attributes 'Implementation-Vendor': 'New Relic'
+    attributes 'Implementation-Vendor-Id': 'com.newrelic'
+    attributes 'Implementation-Version': 1.0
+  }
+}
+
+verifyInstrumentation {
+   passes 'org.springframework.cloud:spring-cloud-gateway-server-mvc:[4.1.0,)'
+}

spring-cloud-gateway-mvc-4.1/src/main/java/org/springframework/cloud/gateway/server/mvc/handler/HandlerFunctions.java

@@ -0,0 +1,45 @@
+package org.springframework.cloud.gateway.server.mvc.handler;
+
+import java.net.URI;
+import java.util.Map;
+import java.util.Set;
+
+import org.springframework.web.servlet.function.ServerRequest;
+import org.springframework.web.servlet.function.ServerResponse;
+
+import com.newrelic.api.agent.NewRelic;
+import com.newrelic.api.agent.Trace;
+import com.newrelic.api.agent.TransactionNamePriority;
+import com.newrelic.api.agent.weaver.Weave;
+import com.newrelic.api.agent.weaver.Weaver;
+
+@Weave
+public abstract class HandlerFunctions {
+
+	@Weave
+	static class LookupProxyExchangeHandlerFunction {
+		
+		private final URI uri = Weaver.callOriginal();
+		
+		@Trace
+		public ServerResponse handle(ServerRequest serverRequest) {
+			if(uri != null) {
+				String path = uri.getPath();
+				StringBuffer sb = new StringBuffer(path);
+				
+				Map<String, String> pathVars = serverRequest.pathVariables();
+				if(pathVars != null && !pathVars.isEmpty()) {
+					sb.append('?');
+					Set<String> keys = pathVars.keySet();
+					for(String key : keys) {
+						sb.append(key);
+						sb.append("=?");
+					}
+				}
+				NewRelic.getAgent().getTransaction().setTransactionName(TransactionNamePriority.FRAMEWORK_HIGH, false, "SoringGateway", "handle", sb.toString(),serverRequest.methodName());
+			}
+			
+			return Weaver.callOriginal();
+		}
+	}
+}

spring-cloud-gateway-mvc-4.1/src/main/java/org/springframework/cloud/gateway/server/mvc/handler/ProxyExchangeHandlerFunction.java

@@ -0,0 +1,25 @@
+package org.springframework.cloud.gateway.server.mvc.handler;
+
+import java.net.URI;
+
+import org.springframework.web.servlet.function.ServerRequest;
+import org.springframework.web.servlet.function.ServerResponse;
+
+import com.newrelic.api.agent.NewRelic;
+import com.newrelic.api.agent.Trace;
+import com.newrelic.api.agent.TracedMethod;
+import com.newrelic.api.agent.weaver.Weave;
+import com.newrelic.api.agent.weaver.Weaver;
+
+@Weave
+public class ProxyExchangeHandlerFunction {
+
+	@Trace
+	public ServerResponse handle(ServerRequest serverRequest) {
+		URI uri = serverRequest.uri();
+		TracedMethod traced = NewRelic.getAgent().getTracedMethod();
+		traced.addCustomAttribute("RequestURI", uri.toASCIIString());
+		traced.addCustomAttribute("Method", serverRequest.methodName());
+		return Weaver.callOriginal();
+	}
+}