Skip to content

Commit 2f549e2

Browse files
committed
refactor(gax): make RetryingContext overload primary and simplify streaming exception wrap (#13995)
1 parent 6ec0bc9 commit 2f549e2

2 files changed

Lines changed: 10 additions & 17 deletions

File tree

sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/rpc/ApiResultRetryAlgorithm.java

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,15 @@ class ApiResultRetryAlgorithm<ResponseT> extends BasicResultRetryAlgorithm<Respo
4545
@Nullable Throwable previousThrowable,
4646
@Nullable ResponseT previousResponse,
4747
TimedAttemptSettings previousSettings) {
48+
return createNextAttempt(null, previousThrowable, previousResponse, previousSettings);
49+
}
50+
51+
@Override
52+
public @Nullable TimedAttemptSettings createNextAttempt(
53+
@Nullable RetryingContext context,
54+
@Nullable Throwable previousThrowable,
55+
@Nullable ResponseT previousResponse,
56+
TimedAttemptSettings previousSettings) {
4857
if (previousThrowable instanceof UnauthenticatedException
4958
&& ((UnauthenticatedException) previousThrowable).isRetryable()
5059
&& previousSettings.getOverallAttemptCount() == previousSettings.getAttemptCount()) {
@@ -64,15 +73,6 @@ class ApiResultRetryAlgorithm<ResponseT> extends BasicResultRetryAlgorithm<Respo
6473
return null;
6574
}
6675

67-
@Override
68-
public @Nullable TimedAttemptSettings createNextAttempt(
69-
@Nullable RetryingContext context,
70-
@Nullable Throwable previousThrowable,
71-
@Nullable ResponseT previousResponse,
72-
TimedAttemptSettings previousSettings) {
73-
return createNextAttempt(previousThrowable, previousResponse, previousSettings);
74-
}
75-
7676
@Override
7777
public boolean shouldRetry(
7878
@Nullable Throwable previousThrowable, @Nullable ResponseT previousResponse) {

sdk-platform-java/gax-java/gax/src/main/java/com/google/api/gax/rpc/ServerStreamingAttemptCallable.java

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -274,14 +274,7 @@ public void onErrorImpl(Throwable t) {
274274
for (Throwable suppressed : unauthenticatedException.getSuppressed()) {
275275
newEx.addSuppressed(suppressed);
276276
}
277-
if (t instanceof ServerStreamingAttemptException) {
278-
ServerStreamingAttemptException attemptEx = (ServerStreamingAttemptException) t;
279-
t =
280-
new ServerStreamingAttemptException(
281-
newEx, attemptEx.canResume(), attemptEx.hasSeenResponses());
282-
} else {
283-
t = newEx;
284-
}
277+
t = newEx;
285278
}
286279
}
287280
}

0 commit comments

Comments
 (0)