Set resource name using the route pattern from spring.

This provides a better static resorce name than trying to conjure one out of the original URL.

Author: tylerbenson

dd-java-agent/dd-java-agent.gradle

@@ -18,9 +18,13 @@ whitelistedInstructionClasses += whitelistedBranchClasses += [
 
 dependencies {
   compile project(':dd-trace')
+  compile project(':dd-java-agent:tooling')
   compile project(':dd-trace-annotations')
 
-  compile group: 'net.bytebuddy', name: 'byte-buddy', version: '1.7.6'
+
+  compile project(':dd-java-agent:integrations:spring-web')
+
+  compile deps.bytebuddy
   compile group: 'org.jboss.byteman', name: 'byteman', version: '3.0.10'
 
   compile group: 'com.google.auto.service', name: 'auto-service', version: '1.0-rc3'

dd-java-agent/integrations/spring-web/spring-web.gradle

@@ -0,0 +1,29 @@
+//apply plugin: 'version-scan'
+//
+//versionScan {
+//  group = 'org.springframework'
+//  module = 'spring-webmvc'
+//  legacyModule = "servlet-api"
+//  versions = "[3.0,)"
+//  verifyPresent = [
+//    "javax.servlet.AsyncEvent"   : null,
+//    "javax.servlet.AsyncListener": null,
+//  ]
+//}
+
+apply from: "${rootDir}/gradle/java.gradle"
+
+dependencies {
+  compileOnly group: 'org.springframework', name: 'spring-webmvc', version: '4.0.0.RELEASE'
+  compileOnly group: 'javax.servlet', name: 'javax.servlet-api', version: '3.1.0'
+//  compileOnly group: 'org.springframework', name: 'spring-webmvc', version: '2.5.6'
+//  compileOnly group: 'javax.servlet', name: 'servlet-api', version: '2.4'
+
+  compile project(':dd-trace')
+  compile project(':dd-java-agent:tooling')
+
+  compile deps.bytebuddy
+  compile deps.opentracing
+
+  compile group: 'com.google.auto.service', name: 'auto-service', version: '1.0-rc3'
+}

dd-java-agent/integrations/spring-web/src/main/java/dd/inst/springweb/SpringWebInstrumentation.java

@@ -0,0 +1,59 @@
+package dd.inst.springweb;
+
+import static net.bytebuddy.matcher.ElementMatchers.hasSuperType;
+import static net.bytebuddy.matcher.ElementMatchers.isInterface;
+import static net.bytebuddy.matcher.ElementMatchers.isMethod;
+import static net.bytebuddy.matcher.ElementMatchers.isPublic;
+import static net.bytebuddy.matcher.ElementMatchers.nameStartsWith;
+import static net.bytebuddy.matcher.ElementMatchers.named;
+import static net.bytebuddy.matcher.ElementMatchers.not;
+import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
+
+import com.datadoghq.trace.DDTags;
+import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
+import io.opentracing.ActiveSpan;
+import io.opentracing.util.GlobalTracer;
+import java.sql.PreparedStatement;
+import java.util.Map;
+import java.util.WeakHashMap;
+import javax.servlet.http.HttpServletRequest;
+import net.bytebuddy.agent.builder.AgentBuilder;
+import net.bytebuddy.asm.Advice;
+import org.springframework.web.servlet.HandlerMapping;
+
+@AutoService(Instrumenter.class)
+public final class SpringWebInstrumentation implements Instrumenter {
+  public static final Map<PreparedStatement, String> preparedStatements = new WeakHashMap<>();
+
+  @Override
+  public AgentBuilder instrument(final AgentBuilder agentBuilder) {
+    return agentBuilder
+        .type(
+            not(isInterface())
+                .and(hasSuperType(named("org.springframework.web.servlet.HandlerAdapter"))))
+        .transform(
+            new AgentBuilder.Transformer.ForAdvice()
+                .advice(
+                    isMethod()
+                        .and(isPublic())
+                        .and(nameStartsWith("handle"))
+                        .and(takesArgument(0, named("javax.servlet.http.HttpServletRequest"))),
+                    SpringWebAdvice.class.getName()));
+  }
+
+  public static class SpringWebAdvice {
+
+    @Advice.OnMethodEnter
+    public static void nameResource(@Advice.Argument(0) final HttpServletRequest request) {
+      final ActiveSpan span = GlobalTracer.get().activeSpan();
+      if (span != null) {
+        final String method = request.getMethod();
+        final String bestMatchingPattern =
+            request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE).toString();
+        final String resourceName = method + " " + bestMatchingPattern;
+        span.setTag(DDTags.RESOURCE_NAME, resourceName);
+      }
+    }
+  }
+}

dd-java-agent/src/main/java/com/datadoghq/agent/TracingAgent.java

@@ -23,7 +23,7 @@
 import static net.bytebuddy.matcher.ElementMatchers.nameContains;
 import static net.bytebuddy.matcher.ElementMatchers.nameStartsWith;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
+import dd.trace.Instrumenter;
 import java.lang.instrument.Instrumentation;
 import java.util.ServiceLoader;
 import lombok.extern.slf4j.Slf4j;
@@ -71,6 +71,9 @@ public static void addByteBuddy(final Instrumentation inst) {
             .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
             .with(new Listener())
             .ignore(nameStartsWith("com.datadoghq.agent.integration"))
+            .or(nameStartsWith("dd.trace"))
+            .or(nameStartsWith("dd.inst"))
+            .or(nameStartsWith("dd.deps"))
             .or(nameStartsWith("java."))
             .or(nameStartsWith("com.sun."))
             .or(nameStartsWith("sun."))

dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/annotation/TraceAnnotationInstrumentation.java

@@ -4,9 +4,9 @@
 import static net.bytebuddy.matcher.ElementMatchers.hasSuperType;
 import static net.bytebuddy.matcher.ElementMatchers.isAnnotatedWith;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
 import com.datadoghq.trace.Trace;
 import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
 import io.opentracing.ActiveSpan;
 import io.opentracing.tag.Tags;
 import io.opentracing.util.GlobalTracer;

dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/jdbc/ConnectionInstrumentation.java

@@ -8,8 +8,8 @@
 import static net.bytebuddy.matcher.ElementMatchers.returns;
 import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
 import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.util.Map;

dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/jdbc/DriverInstrumentation.java

@@ -6,8 +6,8 @@
 import static net.bytebuddy.matcher.ElementMatchers.not;
 import static net.bytebuddy.matcher.ElementMatchers.takesArguments;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
 import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
 import java.sql.Connection;
 import java.sql.Driver;
 import java.util.Map;

dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/jdbc/PreparedStatementInstrumentation.java

@@ -8,9 +8,9 @@
 import static net.bytebuddy.matcher.ElementMatchers.not;
 import static net.bytebuddy.matcher.ElementMatchers.takesArguments;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
 import com.datadoghq.trace.DDTags;
 import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
 import io.opentracing.ActiveSpan;
 import io.opentracing.NoopActiveSpanSource;
 import io.opentracing.tag.Tags;

dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/jdbc/StatementInstrumentation.java

@@ -8,9 +8,9 @@
 import static net.bytebuddy.matcher.ElementMatchers.not;
 import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
 
-import com.datadoghq.agent.instrumentation.Instrumenter;
 import com.datadoghq.trace.DDTags;
 import com.google.auto.service.AutoService;
+import dd.trace.Instrumenter;
 import io.opentracing.ActiveSpan;
 import io.opentracing.NoopActiveSpanSource;
 import io.opentracing.tag.Tags;

…/agent/instrumentation/Instrumenter.java …src/main/java/dd/trace/Instrumenter.javadd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/Instrumenter.java renamed to dd-java-agent/tooling/src/main/java/dd/trace/Instrumenter.java dd-java-agent/src/main/java/com/datadoghq/agent/instrumentation/Instrumenter.java renamed to dd-java-agent/tooling/src/main/java/dd/trace/Instrumenter.java

@@ -1,4 +1,4 @@
-package com.datadoghq.agent.instrumentation;
+package dd.trace;
 
 import net.bytebuddy.agent.builder.AgentBuilder;