Merge pull request apache#39 from rafael-telles/fix-exception-messages

Fix query exception message
Bit-Quill · May 9, 2022 · e383cef · e383cef
2 parents 7f1eeec + 792a29c
commit e383cef
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...light-jdbc-driver/src/main/java/org/apache/arrow/driver/jdbc/utils/FlightStreamQueue.java b/...light-jdbc-driver/src/main/java/org/apache/arrow/driver/jdbc/utils/FlightStreamQueue.java
@@ -93,7 +93,7 @@ public boolean isClosed() {
    */
   @FunctionalInterface
   interface FlightStreamSupplier {
-    Future<FlightStream> get() throws SQLTimeoutException;
+    Future<FlightStream> get() throws SQLException;
   }
 
   private FlightStream next(final FlightStreamSupplier flightStreamSupplier) throws SQLException {
@@ -107,7 +107,7 @@ private FlightStream next(final FlightStreamSupplier flightStreamSupplier) throw
           return stream;
         }
       } catch (final ExecutionException | InterruptedException | CancellationException e) {
-        throw AvaticaConnection.HELPER.wrap("Query canceled", e);
+        throw AvaticaConnection.HELPER.wrap(e.getMessage(), e);
       }
     }
     return null;
@@ -147,7 +147,7 @@ public FlightStream next() throws SQLException {
       try {
         return completionService.take();
       } catch (final InterruptedException e) {
-        throw AvaticaConnection.HELPER.wrap("Query canceled", e);
+        throw AvaticaConnection.HELPER.wrap(e.getMessage(), e);
       }
     });
   }