Change semantics when it comes to retries

Retry only when certain conditions happen, instead of aborting when they not happen
trinodb · Aug 13, 2024 · 5f191af · 5f191af
1 parent 8e92e09
commit 5f191af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/ReadSessionCreator.java b/plugin/trino-bigquery/src/main/java/io/trino/plugin/bigquery/ReadSessionCreator.java
@@ -113,7 +113,7 @@ public ReadSession create(ConnectorSession session, TableId remoteTable, List<St
                             .withMaxRetries(maxCreateReadSessionRetries)
                             .withBackoff(10, 500, MILLIS)
                             .onRetry(event -> log.debug("Request failed, retrying: %s", event.getLastException()))
-                            .abortOn(failure -> !BigQueryUtil.isRetryable(failure))
+                            .handleIf(BigQueryUtil::isRetryable)
                             .build())
                     .get(() -> {
                         try {