chore(spanner): lint format

Author: harshachinta

google-cloud-spanner/src/main/java/com/google/cloud/spanner/AbortedException.java

@@ -32,6 +32,7 @@ public class AbortedException extends SpannerException {
    * new transaction attempt) before a retry can succeed.
    */
   private static final boolean IS_RETRYABLE = false;
+
   private ByteString transactionID;
 
   /** Private constructor. Use {@link SpannerExceptionFactory} to create instances. */

google-cloud-spanner/src/main/java/com/google/cloud/spanner/AsyncTransactionManager.java

@@ -170,7 +170,16 @@ interface AsyncTransactionFunction<I, O> {
    */
   TransactionContextFuture beginAsync();
 
-  TransactionContextFuture beginAsync(AbortedException abortedException);
+  /**
+   * Initializes a new read-write transaction. This method must be called before performing any
+   * operations, and it can only be invoked once per transaction lifecycle.
+   *
+   * <p>This is especially useful in scenarios involving multiplexed sessions and when creating a
+   * new transaction for retry attempts. If {@link #resetForRetryAsync()} is not used, you can pass
+   * the {@link AbortedException} from a previous attempt here to preserve the transaction's
+   * priority.
+   */
+  TransactionContextFuture beginAsync(AbortedException exception);
 
   /**
    * Rolls back the currently active transaction. In most cases there should be no need to call this

google-cloud-spanner/src/main/java/com/google/cloud/spanner/AsyncTransactionManagerImpl.java

@@ -80,13 +80,15 @@ public TransactionContextFutureImpl beginAsync() {
   }
 
   @Override
-  public TransactionContextFutureImpl beginAsync(AbortedException abortedException) {
+  public TransactionContextFutureImpl beginAsync(AbortedException exception) {
     Preconditions.checkState(txn == null, "begin can only be called once");
-    ByteString abortedTransactionId = abortedException.getTransactionID() != null ? abortedException.getTransactionID() : ByteString.EMPTY;
+    ByteString abortedTransactionId =
+        exception.getTransactionID() != null ? exception.getTransactionID() : ByteString.EMPTY;
     return new TransactionContextFutureImpl(this, internalBeginAsync(true, abortedTransactionId));
   }
 
-  private ApiFuture<TransactionContext> internalBeginAsync(boolean firstAttempt, ByteString abortedTransactionID) {
+  private ApiFuture<TransactionContext> internalBeginAsync(
+      boolean firstAttempt, ByteString abortedTransactionID) {
     txnState = TransactionState.STARTED;
 
     // Determine the latest transactionId when using a multiplexed session.

google-cloud-spanner/src/main/java/com/google/cloud/spanner/DelayedAsyncTransactionManager.java

@@ -51,8 +51,8 @@ public TransactionContextFuture beginAsync() {
   }
 
   @Override
-  public TransactionContextFuture beginAsync(AbortedException abortedException) {
-    return getAsyncTransactionManager().beginAsync(abortedException);
+  public TransactionContextFuture beginAsync(AbortedException exception) {
+    return getAsyncTransactionManager().beginAsync(exception);
   }
 
   @Override

google-cloud-spanner/src/main/java/com/google/cloud/spanner/SessionPool.java

@@ -902,6 +902,8 @@ public TransactionContext begin() {
 
     @Override
     public TransactionContext begin(AbortedException exception) {
+      // For regular sessions, the input exception is ignored and the behavior is equivalent to
+      // calling {@link #begin()}.
       return begin();
     }
 

google-cloud-spanner/src/main/java/com/google/cloud/spanner/SessionPoolAsyncTransactionManager.java

@@ -164,7 +164,9 @@ public void onSuccess(TransactionContext result) {
   }
 
   @Override
-  public TransactionContextFuture beginAsync(AbortedException abortedException) {
+  public TransactionContextFuture beginAsync(AbortedException exception) {
+    // For regular sessions, the input exception is ignored and the behavior is equivalent to
+    // calling {@link #beginAsync()}.
     return beginAsync();
   }
 

google-cloud-spanner/src/main/java/com/google/cloud/spanner/TransactionManager.java

@@ -61,6 +61,14 @@ enum TransactionState {
    */
   TransactionContext begin();
 
+  /**
+   * Initializes a new read-write transaction. This method must be called before performing any
+   * operations, and it can only be invoked once per transaction lifecycle.
+   *
+   * <p>This is especially useful in scenarios involving multiplexed sessions and when creating a
+   * new transaction for retry attempts. If {@link #resetForRetry()} is not used, you can pass the
+   * {@link AbortedException} from a previous attempt here to preserve the transaction's priority.
+   */
   TransactionContext begin(AbortedException exception);
 
   /**

google-cloud-spanner/src/main/java/com/google/cloud/spanner/TransactionManagerImpl.java

@@ -59,7 +59,8 @@ public TransactionContext begin() {
   @Override
   public TransactionContext begin(AbortedException exception) {
     Preconditions.checkState(txn == null, "begin can only be called once");
-    ByteString previousAbortedTransactionID = exception.getTransactionID() != null ? exception.getTransactionID() : ByteString.EMPTY;
+    ByteString previousAbortedTransactionID =
+        exception.getTransactionID() != null ? exception.getTransactionID() : ByteString.EMPTY;
     return begin(previousAbortedTransactionID);
   }
 

google-cloud-spanner/src/test/java/com/google/cloud/spanner/MultiplexedSessionDatabaseClientMockServerTest.java

@@ -2172,7 +2172,8 @@ public void testBatchWriteAtLeastOnce() {
 
   @Test
   public void testReadWriteTransactionUsingTransactionManager_SetsTransactionID_DuringAborted() {
-    // Whenever an ABORTED exception occurs, the transaction ID that caused the ABORT should be set in the AbortedException class.
+    // Whenever an ABORTED exception occurs, the transaction ID that caused the ABORT should be set
+    // in the AbortedException class.
     DatabaseClientImpl client =
         (DatabaseClientImpl) spanner.getDatabaseClient(DatabaseId.of("p", "i", "d"));
     // Force the Commit RPC to return Aborted the first time it is called. The exception is cleared
@@ -2229,17 +2230,32 @@ public void testReadWriteTransactionUsingTransactionManager_SetsTransactionID_Du
         mockSpanner.getRequestsOfType(ExecuteSqlRequest.class);
     assertEquals(1, executeSqlRequests.size());
     assertTrue(mockSpanner.getSession(executeSqlRequests.get(0).getSession()).getMultiplexed());
-    assertNotNull(executeSqlRequests.get(0).getTransaction().getBegin().getReadWrite().getMultiplexedSessionPreviousTransactionId());
-    assertEquals(executeSqlRequests.get(0).getTransaction().getBegin().getReadWrite().getMultiplexedSessionPreviousTransactionId(), abortedTransactionID);
+    assertNotNull(
+        executeSqlRequests
+            .get(0)
+            .getTransaction()
+            .getBegin()
+            .getReadWrite()
+            .getMultiplexedSessionPreviousTransactionId());
+    assertEquals(
+        executeSqlRequests
+            .get(0)
+            .getTransaction()
+            .getBegin()
+            .getReadWrite()
+            .getMultiplexedSessionPreviousTransactionId(),
+        abortedTransactionID);
 
     assertNotNull(client.multiplexedSessionDatabaseClient);
     assertEquals(2L, client.multiplexedSessionDatabaseClient.getNumSessionsAcquired().get());
     assertEquals(2L, client.multiplexedSessionDatabaseClient.getNumSessionsReleased().get());
   }
 
   @Test
-  public void testReadWriteTransactionUsingTransactionManager_SetsTransactionID_DuringAbortedInExecuteSql() {
-    // Whenever an ABORTED exception occurs, the transaction ID that caused the ABORT should be set in the AbortedException class.
+  public void
+      testReadWriteTransactionUsingTransactionManager_SetsTransactionID_DuringAbortedInExecuteSql() {
+    // Whenever an ABORTED exception occurs, the transaction ID that caused the ABORT should be set
+    // in the AbortedException class.
     DatabaseClientImpl client =
         (DatabaseClientImpl) spanner.getDatabaseClient(DatabaseId.of("p", "i", "d"));
 
@@ -2303,9 +2319,21 @@ public void testReadWriteTransactionUsingTransactionManager_SetsTransactionID_Du
         mockSpanner.getRequestsOfType(ExecuteSqlRequest.class);
     assertEquals(1, executeSqlRequests.size());
     assertTrue(mockSpanner.getSession(executeSqlRequests.get(0).getSession()).getMultiplexed());
-    assertNotNull(executeSqlRequests.get(0).getTransaction().getBegin().getReadWrite().getMultiplexedSessionPreviousTransactionId());
-    assertEquals(executeSqlRequests.get(0).getTransaction().getBegin().getReadWrite().getMultiplexedSessionPreviousTransactionId(), abortedTransactionID);
-
+    assertNotNull(
+        executeSqlRequests
+            .get(0)
+            .getTransaction()
+            .getBegin()
+            .getReadWrite()
+            .getMultiplexedSessionPreviousTransactionId());
+    assertEquals(
+        executeSqlRequests
+            .get(0)
+            .getTransaction()
+            .getBegin()
+            .getReadWrite()
+            .getMultiplexedSessionPreviousTransactionId(),
+        abortedTransactionID);
 
     assertNotNull(client.multiplexedSessionDatabaseClient);
     assertEquals(2L, client.multiplexedSessionDatabaseClient.getNumSessionsAcquired().get());