WIP

Author: mcculls

dd-java-agent/instrumentation/crac/build.gradle

@@ -0,0 +1,13 @@
+muzzle {
+  pass {
+    group = "org.crac"
+    module = "crac"
+    versions = "[1.3,)"
+  }
+}
+
+apply from: "$rootDir/gradle/java.gradle"
+
+dependencies {
+  compileOnly group: 'org.crac', name: 'crac', version: '1.3.0'
+}

dd-java-agent/instrumentation/crac/src/main/java/trace/instrumentation/crac/CracRuntimeInstrumentation.java

@@ -0,0 +1,41 @@
+package trace.instrumentation.crac;
+
+import static net.bytebuddy.matcher.ElementMatchers.isTypeInitializer;
+
+import com.google.auto.service.AutoService;
+import datadog.trace.agent.tooling.Instrumenter;
+import datadog.trace.agent.tooling.InstrumenterModule;
+import net.bytebuddy.asm.Advice;
+import org.crac.Core;
+
+@AutoService(InstrumenterModule.class)
+public final class CracRuntimeInstrumentation extends InstrumenterModule.Tracing
+    implements Instrumenter.ForSingleType {
+
+  public CracRuntimeInstrumentation() {
+    super("crac");
+  }
+
+  @Override
+  public String instrumentedType() {
+    return "org.crac.Crac";
+  }
+
+  @Override
+  public String[] helperClassNames() {
+    return new String[] {packageName + ".TracerResource"};
+  }
+
+  @Override
+  public void methodAdvice(Instrumenter.MethodTransformer transformer) {
+    transformer.applyAdvice(
+        isTypeInitializer(), CracRuntimeInstrumentation.class.getName() + "$TracerResourceAdvice");
+  }
+
+  public static class TracerResourceAdvice {
+    @Advice.OnMethodExit(suppress = Throwable.class)
+    public static void onExit() {
+      Core.getGlobalContext().register(new TracerResource());
+    }
+  }
+}

dd-java-agent/instrumentation/crac/src/main/java/trace/instrumentation/crac/TracerResource.java

@@ -0,0 +1,16 @@
+package trace.instrumentation.crac;
+
+import org.crac.Context;
+import org.crac.Resource;
+
+public final class TracerResource implements Resource {
+  @Override
+  public void beforeCheckpoint(Context<? extends Resource> context) throws Exception {
+    System.err.println("BEFORE CHECKPOINT");
+  }
+
+  @Override
+  public void afterRestore(Context<? extends Resource> context) throws Exception {
+    System.err.println("AFTER RESTORE");
+  }
+}

settings.gradle

@@ -218,6 +218,7 @@ include ':dd-java-agent:instrumentation:couchbase:couchbase-2.0'
 include ':dd-java-agent:instrumentation:couchbase:couchbase-2.6'
 include ':dd-java-agent:instrumentation:couchbase:couchbase-3.1'
 include ':dd-java-agent:instrumentation:couchbase:couchbase-3.2'
+include ':dd-java-agent:instrumentation:crac'
 include ':dd-java-agent:instrumentation:cucumber'
 include ':dd-java-agent:instrumentation:cxf-2.1'
 include ':dd-java-agent:instrumentation:datanucleus-4'