Skip to content

Commit 25b9995

Browse files
committed
Inject request id into partitionQuery
1 parent deb5424 commit 25b9995

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

+3-1
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,9 @@ private List<Partition> partitionQuery(
315315

316316
final PartitionQueryRequest request = builder.build();
317317
try {
318-
PartitionResponse response = rpc.partitionQuery(request, options);
318+
XGoogSpannerRequestId reqId =
319+
this.session.requestIdCreator.nextRequestId(1 /* channelId */, 0);
320+
PartitionResponse response = rpc.partitionQuery(request, reqId.withOptions(options));
319321
ImmutableList.Builder<Partition> partitions = ImmutableList.builder();
320322
for (com.google.spanner.v1.Partition p : response.getPartitionsList()) {
321323
Partition partition =

0 commit comments

Comments
 (0)