---

igorbernstein2 · web-flow · commit 02754a49be91 · 2019-01-11T08:56:01.000-05:00
yaml --- r: 17773 b: refs/heads/autosynth-trace c: 6ada222 h: refs/heads/master i: 17771: b1eb587
diff --git a/[refs] b/[refs]
@@ -144,7 +144,7 @@ refs/heads/autosynth-spanner: 9bff86d057df31e04c76d72865e8e073ac5794fb
 refs/heads/autosynth-speech: 75d6c62a9d07d3a3642980502a25d07fbde0f232
 refs/heads/autosynth-tasks: b0cdb991f3f75345151a3f68db1aab273dfc069b
 refs/heads/autosynth-texttospeech: 2dcc5dc22be0f456caa1b6a8a4bcdace2641239c
-refs/heads/autosynth-trace: d468e04b9750b8e624d8ffe672feca453026d079
+refs/heads/autosynth-trace: 6ada2226d58a3528d4d818d96e70b5edc83ce936
 refs/heads/autosynth-websecurityscanner: d4febbffb6c648b74faec62fe90e20adadc9a7d3
 refs/heads/bigquerystorage: 06db74d123d7f8a3ef48755c2fcabed09faf8e64
 refs/heads/elharo-patch-1: ce159ef828d3c545991ff78e7b6e0d912a9453e9
diff --git a/branches/autosynth-trace/google-cloud-clients/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/mutaterows/MutateRowsAttemptCallable.java b/branches/autosynth-trace/google-cloud-clients/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/data/v2/stub/mutaterows/MutateRowsAttemptCallable.java
@@ -101,14 +101,14 @@ public Object getTransportCode() {
       };
 
   // Everything needed to issue an RPC
-  private final UnaryCallable<MutateRowsRequest, List<MutateRowsResponse>> innerCallable;
-  private final ApiCallContext callContext;
-  private MutateRowsRequest currentRequest;
+  @Nonnull private final UnaryCallable<MutateRowsRequest, List<MutateRowsResponse>> innerCallable;
+  @Nonnull private final ApiCallContext callContext;
+  @Nonnull private MutateRowsRequest currentRequest;
 
   // Everything needed to build a retry request
-  private List<Integer> originalIndexes;
-  private final Set<StatusCode.Code> retryableCodes;
-  private final List<FailedMutation> permanentFailures;
+  @Nullable private List<Integer> originalIndexes;
+  @Nonnull private final Set<StatusCode.Code> retryableCodes;
+  @Nullable private final List<FailedMutation> permanentFailures;
 
   // Parent controller
   private RetryingFuture<Void> externalFuture;
@@ -135,12 +135,12 @@ public List<MutateRowsResponse> apply(Throwable throwable) {
   MutateRowsAttemptCallable(
       @Nonnull UnaryCallable<MutateRowsRequest, List<MutateRowsResponse>> innerCallable,
       @Nonnull MutateRowsRequest originalRequest,
-      @Nullable ApiCallContext callContext,
+      @Nonnull ApiCallContext callContext,
       @Nonnull Set<StatusCode.Code> retryableCodes) {
-    this.innerCallable = Preconditions.checkNotNull(innerCallable);
-    this.currentRequest = Preconditions.checkNotNull(originalRequest);
-    this.callContext = callContext;
-    this.retryableCodes = Preconditions.checkNotNull(retryableCodes);
+    this.innerCallable = Preconditions.checkNotNull(innerCallable, "innerCallable");
+    this.currentRequest = Preconditions.checkNotNull(originalRequest, "currentRequest");
+    this.callContext = Preconditions.checkNotNull(callContext, "callContext");
+    this.retryableCodes = Preconditions.checkNotNull(retryableCodes, "retryableCodes");
 
     permanentFailures = Lists.newArrayList();
   }
@@ -167,10 +167,10 @@ public Void call() {
           currentRequest.getEntriesCount() > 0, "Request doesn't have any mutations to send");
 
       // Configure the deadline
-      ApiCallContext currentCallContext = null;
-      if (callContext != null) {
+      ApiCallContext currentCallContext = callContext;
+      if (!externalFuture.getAttemptSettings().getRpcTimeout().isZero()) {
         currentCallContext =
-            callContext.withTimeout(externalFuture.getAttemptSettings().getRpcTimeout());
+            currentCallContext.withTimeout(externalFuture.getAttemptSettings().getRpcTimeout());
       }
 
       // Handle concurrent cancellation
diff --git a/branches/autosynth-trace/google-cloud-clients/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/mutaterows/MutateRowsAttemptCallableTest.java b/branches/autosynth-trace/google-cloud-clients/google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/data/v2/stub/mutaterows/MutateRowsAttemptCallableTest.java
@@ -92,6 +92,28 @@ public void singleEntrySuccessTest() throws Exception {
     assertThat(innerCallable.lastRequest).isEqualTo(request);
   }
 
+  @Test
+  public void testNoRpcTimeout() {
+    parentFuture.timedAttemptSettings =
+        parentFuture.timedAttemptSettings.toBuilder().setRpcTimeout(Duration.ZERO).build();
+
+    MutateRowsRequest request =
+        MutateRowsRequest.newBuilder().addEntries(Entry.getDefaultInstance()).build();
+
+    innerCallable.response.add(
+        MutateRowsResponse.newBuilder()
+            .addEntries(
+                MutateRowsResponse.Entry.newBuilder().setIndex(0).setStatus(OK_STATUS_PROTO))
+            .build());
+
+    MutateRowsAttemptCallable attemptCallable =
+        new MutateRowsAttemptCallable(innerCallable, request, callContext, retryCodes);
+    attemptCallable.setExternalFuture(parentFuture);
+    attemptCallable.call();
+
+    assertThat(innerCallable.lastContext.getTimeout()).isNull();
+  }
+
   @Test
   public void mixedTest() {
     // Setup the request & response
@@ -340,7 +362,7 @@ public ApiFuture<List<MutateRowsResponse>> futureCall(
   static class MockRetryingFuture extends AbstractApiFuture<Void> implements RetryingFuture<Void> {
     ApiFuture<Void> attemptFuture;
 
-    final TimedAttemptSettings timedAttemptSettings;
+    TimedAttemptSettings timedAttemptSettings;
 
     MockRetryingFuture() {
       this(Duration.ofSeconds(5));
