diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerRetryHelper.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerRetryHelper.java index c464e37cf0..e22a0f7485 100644 --- a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerRetryHelper.java +++ b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/SpannerRetryHelper.java @@ -63,7 +63,7 @@ static T runTxWithRetriesOnAborted(Callable callable) { static T runTxWithRetriesOnAborted(Callable callable, RetrySettings retrySettings) { try { return RetryHelper.runWithRetries( - callable, txRetrySettings, new TxRetryAlgorithm<>(), NanoClock.getDefaultClock()); + callable, retrySettings, new TxRetryAlgorithm<>(), NanoClock.getDefaultClock()); } catch (RetryHelperException e) { if (e.getCause() != null) { Throwables.throwIfUnchecked(e.getCause()); diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/SpannerRetryHelperTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/SpannerRetryHelperTest.java index ffa5651c74..8991691bff 100644 --- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/SpannerRetryHelperTest.java +++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/SpannerRetryHelperTest.java @@ -188,13 +188,18 @@ public Integer call() throws Exception { @Test public void testExceptionWithRetryInfo() { - final int RETRY_DELAY_NANOS = 100_000_000; + final int RETRY_DELAY_MILLIS = 100; Metadata.Key key = ProtoUtils.keyForProto(RetryInfo.getDefaultInstance()); Status status = Status.fromCodeValue(Status.Code.ABORTED.value()); Metadata trailers = new Metadata(); RetryInfo retryInfo = RetryInfo.newBuilder() - .setRetryDelay(Duration.newBuilder().setNanos(RETRY_DELAY_NANOS).build()) + .setRetryDelay( + Duration.newBuilder() + .setNanos( + (int) + TimeUnit.NANOSECONDS.convert(RETRY_DELAY_MILLIS, TimeUnit.MILLISECONDS)) + .build()) .build(); trailers.put(key, retryInfo); final SpannerException e = @@ -214,8 +219,8 @@ public Integer call() throws Exception { // retry info of the exception. Stopwatch watch = Stopwatch.createStarted(); assertThat(SpannerRetryHelper.runTxWithRetriesOnAborted(callable)).isEqualTo(2); - long elapsed = watch.elapsed(TimeUnit.NANOSECONDS); - assertThat(elapsed >= RETRY_DELAY_NANOS).isTrue(); + long elapsed = watch.elapsed(TimeUnit.MILLISECONDS); + assertThat(elapsed >= RETRY_DELAY_MILLIS).isTrue(); } private SpannerException abortedWithRetryInfo(int nanos) {