poc: Add ClientMetricsTracer to CompositeTracerFactory

google-cloud-bigtable/pom.xml

@@ -85,10 +85,12 @@
     <dependency>
       <groupId>com.google.api</groupId>
       <artifactId>gax-grpc</artifactId>
+      <version>2.30.1-SNAPSHOT</version>
     </dependency>
     <dependency>
       <groupId>com.google.api</groupId>
       <artifactId>gax</artifactId>
+      <version>2.30.1-SNAPSHOT</version>
     </dependency>
     <dependency>
       <groupId>com.google.api.grpc</groupId>

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/metrics/BuiltinMetricsTracer.java

@@ -132,6 +132,16 @@ public void attemptSucceeded() {
     recordAttemptCompletion(null);
   }
 
+  @Override
+  public void attemptSucceeded(Object response) {
+    attemptSucceeded();
+  }
+
+  @Override
+  public void operationSucceeded(Object response) {
+    operationSucceeded();
+  }
+
   @Override
   public void attemptCancelled() {
     recordAttemptCompletion(new CancellationException());

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/metrics/CompositeTracer.java

@@ -69,6 +69,20 @@ public void operationSucceeded() {
     }
   }
 
+  @Override
+  public void operationSucceeded(Object response) {
+    for (ApiTracer child : children) {
+      child.operationSucceeded(response);
+    }
+  }
+
+  @Override
+  public void retryCount(int count) {
+    for (ApiTracer child : children) {
+      child.retryCount(count);
+    }
+  }
+
   @Override
   public void operationCancelled() {
     for (ApiTracer child : children) {
@@ -110,6 +124,13 @@ public void attemptSucceeded() {
     }
   }
 
+  @Override
+  public void attemptSucceeded(Object response) {
+    for (ApiTracer child : children) {
+      child.attemptSucceeded(response);
+    }
+  }
+
   @Override
   public void attemptCancelled() {
     for (ApiTracer child : children) {
@@ -225,4 +246,22 @@ public void grpcChannelQueuedLatencies(long queuedTimeMs) {
       tracer.grpcChannelQueuedLatencies(queuedTimeMs);
     }
   }
+
+  @Override
+  public void grpcTargetResolutionDelay(long elapsed) {
+    for (ApiTracer child : children) {
+      child.grpcTargetResolutionDelay(elapsed);
+    }  }
+
+  @Override
+  public void grpcChannelReadinessDelay(long elapsed) {
+    for (ApiTracer child : children) {
+      child.grpcChannelReadinessDelay(elapsed);
+    }  }
+
+  @Override
+  public void grpcCallSendDelay(long elapsed) {
+    for (ApiTracer child : children) {
+      child.grpcCallSendDelay(elapsed);
+    }  }
 }

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/metrics/CompositeTracerFactory.java

@@ -19,6 +19,7 @@
 import com.google.api.gax.tracing.ApiTracer;
 import com.google.api.gax.tracing.ApiTracerFactory;
 import com.google.api.gax.tracing.BaseApiTracerFactory;
+import com.google.api.gax.tracing.ClientMetricsTracer;
 import com.google.api.gax.tracing.SpanName;
 import com.google.common.collect.ImmutableList;
 import java.util.ArrayList;
@@ -42,4 +43,15 @@ public ApiTracer newTracer(ApiTracer parent, SpanName spanName, OperationType op
     }
     return new CompositeTracer(children);
   }
+
+  @Override
+  public ClientMetricsTracer newClientMetricsTracer() {
+    for (ApiTracerFactory factory : apiTracerFactories) {
+      ClientMetricsTracer clientMetricsTracer = factory.newClientMetricsTracer();
+      if (clientMetricsTracer != null) {
+        return clientMetricsTracer;
+      }
+    }
+    return null;
+  }
 }

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/metrics/MetricsTracer.java

@@ -139,6 +139,16 @@ public void attemptSucceeded() {
     recordAttemptCompletion(null);
   }
 
+  @Override
+  public void attemptSucceeded(Object response) {
+    attemptSucceeded();
+  }
+
+  @Override
+  public void operationSucceeded(Object response) {
+    operationSucceeded();
+  }
+
   @Override
   public void attemptCancelled() {
     recordAttemptCompletion(new CancellationException());