This actually ensures close() is called

apache · Sep 18, 2023 · e28ed60 · e28ed60
1 parent 5cd8579
commit e28ed60
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.../flight-sql/src/main/java/org/apache/arrow/adbc/driver/flightsql/FlightSqlConnection.java b/.../flight-sql/src/main/java/org/apache/arrow/adbc/driver/flightsql/FlightSqlConnection.java
@@ -110,8 +110,9 @@ public AdbcStatement bulkIngest(String targetTableName, BulkIngestMode mode)
   @Override
   public ArrowReader getInfo(int[] infoCodes) throws AdbcException {
     try (InfoMetadataBuilder builder = new InfoMetadataBuilder(allocator, client, infoCodes)) {
-      final VectorSchemaRoot root = builder.build();
-      return RootArrowReader.fromRoot(allocator, root);
+      try (final VectorSchemaRoot root = builder.build()) {
+        return RootArrowReader.fromRoot(allocator, root);
+      }
     } catch (Exception e) {
       throw AdbcException.invalidState("[Flight SQL] Failed to get info");
     }