chore(spanner): set precommit token in CommitRequest

harshachinta · harshachinta · commit 3f9e5ec597c8 · 2024-10-14T11:37:47.000Z
diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/TransactionRunnerImpl.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/TransactionRunnerImpl.java
@@ -429,6 +429,10 @@ public void run() {
             }
             requestBuilder.setRequestOptions(requestOptionsBuilder.build());
           }
+          if (session.getIsMultiplexed() && getLatestPrecommitToken() != null) {
+            // Set the precommit token in the CommitRequest for multiplexed sessions.
+            requestBuilder.setPrecommitToken(getLatestPrecommitToken());
+          }
           final CommitRequest commitRequest = requestBuilder.build();
           span.addAnnotation("Starting Commit");
           final ApiFuture<com.google.spanner.v1.CommitResponse> commitFuture;

Original file line number	Diff line number	Diff line change
`@@ -429,6 +429,10 @@ public void run() {`
`429`	`429`	`}`
`430`	`430`	`requestBuilder.setRequestOptions(requestOptionsBuilder.build());`
`431`	`431`	`}`
	`432`	`+ if (session.getIsMultiplexed() && getLatestPrecommitToken() != null) {`
	`433`	`+ // Set the precommit token in the CommitRequest for multiplexed sessions.`
	`434`	`+ requestBuilder.setPrecommitToken(getLatestPrecommitToken());`
	`435`	`+ }`
`432`	`436`	`final CommitRequest commitRequest = requestBuilder.build();`
`433`	`437`	`span.addAnnotation("Starting Commit");`
`434`	`438`	`final ApiFuture<com.google.spanner.v1.CommitResponse> commitFuture;`