Skip to content

Commit e8e6bde

Browse files
committed
Fix StreamingChunkDownloadTask to catch RuntimeException (parity with #1302)
PR #1302 fixed ChunkLinkDownloadService (legacy RemoteChunkProvider path) to catch DatabricksError (extends RuntimeException) when the server returns HTTP 404 for expired results. StreamingChunkDownloadTask had the same gap — only caught IOException | SQLException, letting DatabricksError escape uncaught and leaving the chunk in a bad state. Widen catch to IOException | SQLException | RuntimeException so expired result errors are retried properly. Co-authored-by: Isaac Signed-off-by: Gopal Lal <gopal.lal@databricks.com>
1 parent b8d30b7 commit e8e6bde

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/main/java/com/databricks/jdbc/api/impl/arrow/StreamingChunkDownloadTask.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ public Void call() throws DatabricksSQLException {
6464

6565
LOGGER.debug("Successfully downloaded chunk {}", chunk.getChunkIndex());
6666

67-
} catch (IOException | SQLException e) {
67+
} catch (IOException | SQLException | RuntimeException e) {
6868
retries++;
6969
if (retries >= MAX_RETRIES) {
7070
LOGGER.error(

0 commit comments

Comments
 (0)