fix: fix translate exception in compat layer (#2879)

Change-Id: I23d2c4103d4459f4d5cdceab22cae8d5ba5cd64d

Author: mutianf

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/internal/compat/ops/DivertingUnaryCallable.java

@@ -34,6 +34,8 @@
 import io.grpc.StatusRuntimeException;
 import java.time.Duration;
 import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.CompletionException;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ThreadLocalRandom;
 
 /** A callable to fork traffic between classic and session based operations. */
@@ -110,15 +112,25 @@ private boolean useExperimental(ReqT req) {
   }
 
   ApiException translateException(Throwable e) {
+    Throwable cause = e;
+    while (cause instanceof CompletionException || cause instanceof ExecutionException) {
+      if (cause.getCause() != null) {
+        cause = cause.getCause();
+      } else {
+        break;
+      }
+    }
+
     Status.Code code = Status.Code.UNKNOWN;
 
-    if (e instanceof StatusRuntimeException) {
-      code = ((StatusRuntimeException) e).getStatus().getCode();
+    if (cause instanceof StatusRuntimeException) {
+      code = ((StatusRuntimeException) cause).getStatus().getCode();
     }
-    if (e instanceof StatusException) {
-      code = ((StatusException) e).getStatus().getCode();
+    if (cause instanceof StatusException) {
+      code = ((StatusException) cause).getStatus().getCode();
     }
 
-    return ApiExceptionFactory.createException(e.getMessage(), e, GrpcStatusCode.of(code), false);
+    return ApiExceptionFactory.createException(
+        cause.getMessage(), e, GrpcStatusCode.of(code), false);
   }
 }