chore: format mutate rows impl to be side by side (#2137)

Change-Id: I24a416366207f0154befec267bc74a722a0e4944

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:
- [ ] Make sure to open an issue as a [bug/issue](https://togithub.com/googleapis/java-bigtable/issues/new/choose) before writing your code!  That way we can discuss the change, evaluate designs, and agree on the general idea
- [ ] Ensure the tests and linter pass
- [ ] Code coverage does not decrease (if any source code was changed)
- [ ] Appropriate docs were updated (if necessary)
- [ ] Rollback plan is reviewed and LGTMed
- [ ] All new data plane features have a completed end to end testing plan

Fixes #<issue_number_goes_here> ☕️

If you write sample code, please follow the [samples format](
https://togithub.com/GoogleCloudPlatform/java-docs-samples/blob/main/SAMPLE_FORMAT.md).

Author: igorbernstein2

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

@@ -664,6 +664,72 @@ public Map<String, String> extract(MutateRowRequest mutateRowRequest) {
         methodName, new MutateRowCallable(retrying, requestContext));
   }
 
+  /**
+   * Internal helper to create the base MutateRows callable chain. The chain is responsible for
+   * retrying individual entry in case of error.
+   *
+   * <p>NOTE: the caller is responsible for adding tracing & metrics.
+   *
+   * @see MutateRowsRetryingCallable for more details
+   */
+  private UnaryCallable<MutateRowsRequest, Void> createMutateRowsBaseCallable() {
+    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> base =
+        GrpcRawCallableFactory.createServerStreamingCallable(
+            GrpcCallSettings.<MutateRowsRequest, MutateRowsResponse>newBuilder()
+                .setMethodDescriptor(BigtableGrpc.getMutateRowsMethod())
+                .setParamsExtractor(
+                    new RequestParamsExtractor<MutateRowsRequest>() {
+                      @Override
+                      public Map<String, String> extract(MutateRowsRequest mutateRowsRequest) {
+                        return ImmutableMap.of(
+                            "table_name", mutateRowsRequest.getTableName(),
+                            "app_profile_id", mutateRowsRequest.getAppProfileId());
+                      }
+                    })
+                .build(),
+            settings.bulkMutateRowsSettings().getRetryableCodes());
+
+    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> callable =
+        new StatsHeadersServerStreamingCallable<>(base);
+
+    if (settings.bulkMutateRowsSettings().isServerInitiatedFlowControlEnabled()) {
+      callable = new RateLimitingServerStreamingCallable(callable);
+    }
+
+    // Sometimes MutateRows connections are disconnected via an RST frame. This error is transient
+    // and
+    // should be treated similar to UNAVAILABLE. However, this exception has an INTERNAL error code
+    // which by default is not retryable. Convert the exception so it can be retried in the client.
+    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> convertException =
+        new ConvertExceptionCallable<>(callable);
+
+    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> withBigtableTracer =
+        new BigtableTracerStreamingCallable<>(convertException);
+
+    BasicResultRetryAlgorithm<Void> resultRetryAlgorithm;
+    if (settings.getEnableRetryInfo()) {
+      resultRetryAlgorithm = new RetryInfoRetryAlgorithm<>();
+    } else {
+      resultRetryAlgorithm = new ApiResultRetryAlgorithm<>();
+    }
+
+    RetryAlgorithm<Void> retryAlgorithm =
+        new RetryAlgorithm<>(
+            resultRetryAlgorithm,
+            new ExponentialRetryAlgorithm(
+                settings.bulkMutateRowsSettings().getRetrySettings(), clientContext.getClock()));
+
+    RetryingExecutorWithContext<Void> retryingExecutor =
+        new ScheduledRetryingExecutor<>(retryAlgorithm, clientContext.getExecutor());
+
+    return new MutateRowsRetryingCallable(
+        clientContext.getDefaultCallContext(),
+        withBigtableTracer,
+        retryingExecutor,
+        settings.bulkMutateRowsSettings().getRetryableCodes(),
+        retryAlgorithm);
+  }
+
   /**
    * Creates a callable chain to handle MutatesRows RPCs. This is meant to be used for manual
    * batching. The chain will:
@@ -773,72 +839,6 @@ public Batcher<ByteString, Row> newBulkReadRowsBatcher(
         MoreObjects.firstNonNull(ctx, clientContext.getDefaultCallContext()));
   }
 
-  /**
-   * Internal helper to create the base MutateRows callable chain. The chain is responsible for
-   * retrying individual entry in case of error.
-   *
-   * <p>NOTE: the caller is responsible for adding tracing & metrics.
-   *
-   * @see MutateRowsRetryingCallable for more details
-   */
-  private UnaryCallable<MutateRowsRequest, Void> createMutateRowsBaseCallable() {
-    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> base =
-        GrpcRawCallableFactory.createServerStreamingCallable(
-            GrpcCallSettings.<MutateRowsRequest, MutateRowsResponse>newBuilder()
-                .setMethodDescriptor(BigtableGrpc.getMutateRowsMethod())
-                .setParamsExtractor(
-                    new RequestParamsExtractor<MutateRowsRequest>() {
-                      @Override
-                      public Map<String, String> extract(MutateRowsRequest mutateRowsRequest) {
-                        return ImmutableMap.of(
-                            "table_name", mutateRowsRequest.getTableName(),
-                            "app_profile_id", mutateRowsRequest.getAppProfileId());
-                      }
-                    })
-                .build(),
-            settings.bulkMutateRowsSettings().getRetryableCodes());
-
-    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> callable =
-        new StatsHeadersServerStreamingCallable<>(base);
-
-    if (settings.bulkMutateRowsSettings().isServerInitiatedFlowControlEnabled()) {
-      callable = new RateLimitingServerStreamingCallable(callable);
-    }
-
-    // Sometimes MutateRows connections are disconnected via an RST frame. This error is transient
-    // and
-    // should be treated similar to UNAVAILABLE. However, this exception has an INTERNAL error code
-    // which by default is not retryable. Convert the exception so it can be retried in the client.
-    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> convertException =
-        new ConvertExceptionCallable<>(callable);
-
-    ServerStreamingCallable<MutateRowsRequest, MutateRowsResponse> withBigtableTracer =
-        new BigtableTracerStreamingCallable<>(convertException);
-
-    BasicResultRetryAlgorithm<Void> resultRetryAlgorithm;
-    if (settings.getEnableRetryInfo()) {
-      resultRetryAlgorithm = new RetryInfoRetryAlgorithm<>();
-    } else {
-      resultRetryAlgorithm = new ApiResultRetryAlgorithm<>();
-    }
-
-    RetryAlgorithm<Void> retryAlgorithm =
-        new RetryAlgorithm<>(
-            resultRetryAlgorithm,
-            new ExponentialRetryAlgorithm(
-                settings.bulkMutateRowsSettings().getRetrySettings(), clientContext.getClock()));
-
-    RetryingExecutorWithContext<Void> retryingExecutor =
-        new ScheduledRetryingExecutor<>(retryAlgorithm, clientContext.getExecutor());
-
-    return new MutateRowsRetryingCallable(
-        clientContext.getDefaultCallContext(),
-        withBigtableTracer,
-        retryingExecutor,
-        settings.bulkMutateRowsSettings().getRetryableCodes(),
-        retryAlgorithm);
-  }
-
   /**
    * Creates a callable chain to handle CheckAndMutateRow RPCs. THe chain will:
    *