apacheGH-483: Use ArrowReader instead of ArrowStreamReader in FlightS…

…qlClient.executeIngest method
eramitmittal · Jan 6, 2025 · d6df3c2 · d6df3c2
1 parent be627bf
commit d6df3c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flight/flight-sql/src/main/java/org/apache/arrow/flight/sql/FlightSqlClient.java b/flight/flight-sql/src/main/java/org/apache/arrow/flight/sql/FlightSqlClient.java
@@ -92,7 +92,7 @@
 import org.apache.arrow.util.AutoCloseables;
 import org.apache.arrow.util.Preconditions;
 import org.apache.arrow.vector.VectorSchemaRoot;
-import org.apache.arrow.vector.ipc.ArrowStreamReader;
+import org.apache.arrow.vector.ipc.ArrowReader;
 import org.apache.arrow.vector.ipc.ReadChannel;
 import org.apache.arrow.vector.ipc.message.MessageSerializer;
 import org.apache.arrow.vector.types.pojo.Schema;
@@ -236,7 +236,7 @@ public long executeIngest(
    * @return the number of rows affected.
    */
   public long executeIngest(
-      final ArrowStreamReader dataReader,
+      final ArrowReader dataReader,
       final ExecuteIngestOptions ingestOptions,
       final CallOption... options) {
     return executeIngest(dataReader, ingestOptions, /*transaction*/ null, options);
@@ -270,7 +270,7 @@ public long executeIngest(
    * @return the number of rows affected.
    */
   public long executeIngest(
-      final ArrowStreamReader dataReader,
+      final ArrowReader dataReader,
       final ExecuteIngestOptions ingestOptions,
       Transaction transaction,
       final CallOption... options) {