adding Couchbase Java Client instrumentation

Author: dhilpipre

instrumentation/couchbase-client-2.4.0/build.gradle

@@ -0,0 +1,23 @@
+jar {
+  manifest { attributes 'Implementation-Title': 'com.newrelic.instrumentation.labs.Couchbase_Client-2.4.0'}
+}
+
+dependencies {
+    implementation(project(":agent-bridge"))
+    implementation(project(":agent-bridge-datastore"))
+    implementation(project(":newrelic-api"))
+    implementation(project(":newrelic-weaver-api"))
+    implementation 'com.couchbase.client:java-client:2.4.0'
+}
+
+verifyInstrumentation {
+  	passes 'com.couchbase.client:java-client:[2.4.0,3.0.0)'
+  	exclude 'com.couchbase.client:java-client:2.7.5'
+  	exclude 'com.couchbase.client:java-client:2.7.8'
+  	excludeRegex '.*SNAPSHOT'
+}
+
+site {
+	title 'Couchbase'
+	type 'Framework'
+}

instrumentation/couchbase-client-2.4.0/src/main/java/com/couchbase/client/core/ClusterFacade_Instrumentation.java

@@ -0,0 +1,23 @@
+package com.couchbase.client.core;
+
+import rx.Observable;
+
+import com.couchbase.client.core.message.CouchbaseRequest;
+import com.couchbase.client.core.message.CouchbaseResponse;
+import com.newrelic.api.agent.Trace;
+import com.newrelic.api.agent.weaver.MatchType;
+import com.newrelic.api.agent.weaver.Weave;
+import com.newrelic.api.agent.weaver.Weaver;
+
+@Weave(type=MatchType.Interface, originalName = "com.couchbase.client.core.ClusterFacade")
+public abstract class ClusterFacade_Instrumentation {
+
+    @Trace
+    public <R extends CouchbaseResponse> Observable<R> send(CouchbaseRequest paramCouchbaseRequest) {
+        
+        Observable<R> observable = Weaver.callOriginal();
+        
+        return observable;
+        
+    }
+}

instrumentation/couchbase-client-2.4.0/src/main/java/com/couchbase/client/java/Bucket_Instrumentation.java

@@ -0,0 +1,76 @@
+package com.couchbase.client.java;
+
+import java.util.List;
+import java.util.concurrent.TimeUnit;
+
+import com.couchbase.client.java.document.Document;
+import com.couchbase.client.java.transcoder.Transcoder;
+import com.newrelic.api.agent.DatastoreParameters;
+import com.newrelic.api.agent.NewRelic;
+import com.newrelic.api.agent.Trace;
+import com.newrelic.api.agent.weaver.MatchType;
+import com.newrelic.api.agent.weaver.Weave;
+import com.newrelic.api.agent.weaver.Weaver;
+
+@SuppressWarnings("rawtypes")
+@Weave(type=MatchType.Interface, originalName = "com.couchbase.client.java.Cluster")
+public abstract class Cluster_Instrumentation {
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket() {
+        Bucket_Instrumentation bucket = Weaver.callOriginal();
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(bucket.name()).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return bucket;
+    }
+    
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(long timeout, TimeUnit timeUnit) {
+        Bucket_Instrumentation bucket = Weaver.callOriginal();
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(bucket.name()).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return bucket;
+    }
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name, long timeout, TimeUnit timeUnit) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name, String password) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name, String password, long timeout, TimeUnit timeUnit) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+
+	@Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name, String password, List<Transcoder<? extends Document, ?>> transcoders) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+
+    @Trace(leaf=true)
+    public Bucket_Instrumentation openBucket(String name, String password, List<Transcoder<? extends Document, ?>> transcoders, long timeout, TimeUnit timeUnit) {
+        DatastoreParameters params = DatastoreParameters.product("Couchbase").collection(name).operation("openBucket").build();
+        NewRelic.getAgent().getTracedMethod().reportAsExternal(params);
+        return Weaver.callOriginal();
+    }
+}