Add option to not include couchbase internal spans

Author: brentm5

dd-java-agent/instrumentation/couchbase/couchbase-3.1/src/main/java/datadog/trace/instrumentation/couchbase_31/client/DatadogRequestTracer.java

@@ -5,6 +5,7 @@
 import com.couchbase.client.core.Core;
 import com.couchbase.client.core.cnc.RequestSpan;
 import com.couchbase.client.core.cnc.RequestTracer;
+import datadog.trace.api.Config;
 import datadog.trace.bootstrap.ContextStore;
 import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
 import datadog.trace.bootstrap.instrumentation.api.AgentTracer;
@@ -33,11 +34,15 @@ public RequestSpan requestSpan(String requestName, RequestSpan requestParent) {
     boolean measured = true;
     Object seedNodes = null;
 
+    final Config config = Config.get();
     AgentSpan parent = DatadogRequestSpan.unwrap(requestParent);
     if (null == parent) {
       parent = tracer.activeSpan();
     }
     if (null != parent && COUCHBASE_CLIENT.equals(parent.getTag(Tags.COMPONENT))) {
+      if (!config.isCouchbaseInternalEnabled()) {
+        return requestParent;
+      }
       spanName = COUCHBASE_INTERNAL;
       measured = false;
       seedNodes = parent.getTag(InstrumentationTags.COUCHBASE_SEED_NODES);

dd-java-agent/instrumentation/couchbase/couchbase-3.2/src/main/java/datadog/trace/instrumentation/couchbase_32/client/DatadogRequestTracer.java

@@ -6,6 +6,7 @@
 import com.couchbase.client.core.Core;
 import com.couchbase.client.core.cnc.RequestSpan;
 import com.couchbase.client.core.cnc.RequestTracer;
+import datadog.trace.api.Config;
 import datadog.trace.bootstrap.ContextStore;
 import datadog.trace.bootstrap.instrumentation.api.AgentSpan;
 import datadog.trace.bootstrap.instrumentation.api.AgentTracer;
@@ -35,11 +36,15 @@ public RequestSpan requestSpan(String requestName, RequestSpan requestParent) {
     boolean measured = true;
     Object seedNodes = null;
 
+    final Config config = Config.get();
     AgentSpan parent = DatadogRequestSpan.unwrap(requestParent);
     if (null == parent) {
       parent = tracer.activeSpan();
     }
     if (null != parent && COUCHBASE_CLIENT.equals(parent.getTag(Tags.COMPONENT))) {
+      if (!config.isCouchbaseInternalEnabled()) {
+        return requestParent;
+      }
       spanName = COUCHBASE_INTERNAL;
       measured = false;
       seedNodes = parent.getTag(InstrumentationTags.COUCHBASE_SEED_NODES);

dd-trace-api/src/main/java/datadog/trace/api/ConfigDefaults.java

@@ -214,6 +214,7 @@ public final class ConfigDefaults {
 
   static final float DEFAULT_TRACE_FLUSH_INTERVAL = 1;
 
+  static final boolean DEFAULT_COUCHBASE_INTERNAL_ENABLED = true;
   static final boolean DEFAULT_ELASTICSEARCH_BODY_ENABLED = false;
   static final boolean DEFAULT_ELASTICSEARCH_PARAMS_ENABLED = true;
   static final boolean DEFAULT_ELASTICSEARCH_BODY_AND_PARAMS_ENABLED = false;

dd-trace-api/src/main/java/datadog/trace/api/config/TraceInstrumentationConfig.java

@@ -125,6 +125,8 @@ public final class TraceInstrumentationConfig {
   public static final String RESOLVER_RESET_INTERVAL = "resolver.reset.interval";
   public static final String RESOLVER_NAMES_ARE_UNIQUE = "resolver.names.are.unique";
 
+  public static final String COUCHBASE_INTERNAL_ENABLED = "trace.couchbase.internal.enabled";
+
   public static final String ELASTICSEARCH_BODY_ENABLED = "trace.elasticsearch.body.enabled";
   public static final String ELASTICSEARCH_PARAMS_ENABLED = "trace.elasticsearch.params.enabled";
   public static final String ELASTICSEARCH_BODY_AND_PARAMS_ENABLED =

internal-api/src/main/java/datadog/trace/api/Config.java

@@ -30,6 +30,7 @@
 import static datadog.trace.api.ConfigDefaults.DEFAULT_CIVISIBILITY_SOURCE_DATA_ROOT_CHECK_ENABLED;
 import static datadog.trace.api.ConfigDefaults.DEFAULT_CLIENT_IP_ENABLED;
 import static datadog.trace.api.ConfigDefaults.DEFAULT_CLOCK_SYNC_PERIOD;
+import static datadog.trace.api.ConfigDefaults.DEFAULT_COUCHBASE_INTERNAL_ENABLED;
 import static datadog.trace.api.ConfigDefaults.DEFAULT_CWS_ENABLED;
 import static datadog.trace.api.ConfigDefaults.DEFAULT_CWS_TLS_REFRESH;
 import static datadog.trace.api.ConfigDefaults.DEFAULT_DATA_STREAMS_BUCKET_DURATION;
@@ -328,6 +329,7 @@
 import static datadog.trace.api.config.RemoteConfigConfig.REMOTE_CONFIG_TARGETS_KEY_ID;
 import static datadog.trace.api.config.RemoteConfigConfig.REMOTE_CONFIG_URL;
 import static datadog.trace.api.config.TraceInstrumentationConfig.AXIS_PROMOTE_RESOURCE_NAME;
+import static datadog.trace.api.config.TraceInstrumentationConfig.COUCHBASE_INTERNAL_ENABLED;
 import static datadog.trace.api.config.TraceInstrumentationConfig.DB_CLIENT_HOST_SPLIT_BY_HOST;
 import static datadog.trace.api.config.TraceInstrumentationConfig.DB_CLIENT_HOST_SPLIT_BY_INSTANCE;
 import static datadog.trace.api.config.TraceInstrumentationConfig.DB_CLIENT_HOST_SPLIT_BY_INSTANCE_TYPE_SUFFIX;
@@ -894,6 +896,7 @@ static class HostNameHolder {
   private final boolean longRunningTraceEnabled;
   private final long longRunningTraceInitialFlushInterval;
   private final long longRunningTraceFlushInterval;
+  private final boolean couchbaseInternalEnabled;
   private final boolean elasticsearchBodyEnabled;
   private final boolean elasticsearchParamsEnabled;
   private final boolean elasticsearchBodyAndParamsEnabled;
@@ -981,6 +984,8 @@ private Config(final ConfigProvider configProvider, final InstrumenterConfig ins
     } else {
       secureRandom = configProvider.getBoolean(SECURE_RANDOM, DEFAULT_SECURE_RANDOM);
     }
+    couchbaseInternalEnabled =
+        configProvider.getBoolean(COUCHBASE_INTERNAL_ENABLED, DEFAULT_COUCHBASE_INTERNAL_ENABLED);
     elasticsearchBodyEnabled =
         configProvider.getBoolean(ELASTICSEARCH_BODY_ENABLED, DEFAULT_ELASTICSEARCH_BODY_ENABLED);
     elasticsearchParamsEnabled =
@@ -3287,6 +3292,10 @@ public BitSet getGrpcClientErrorStatuses() {
     return grpcClientErrorStatuses;
   }
 
+  public boolean isCouchbaseInternalEnabled() {
+    return couchbaseInternalEnabled;
+  }
+
   public boolean isElasticsearchBodyEnabled() {
     return elasticsearchBodyEnabled;
   }
@@ -4361,6 +4370,8 @@ public String toString() {
         + longRunningTraceInitialFlushInterval
         + ", longRunningTraceFlushInterval="
         + longRunningTraceFlushInterval
+        + ", couchbaseInternalEnabled="
+        + couchbaseInternalEnabled
         + ", elasticsearchBodyEnabled="
         + elasticsearchBodyEnabled
         + ", elasticsearchParamsEnabled="