Implement RecordBatchReader for arrow_csv::reader::BufReader (#4195)

apache · May 10, 2023 · adca63a · adca63a
1 parent b314118
commit adca63a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/arrow-csv/src/reader/mod.rs b/arrow-csv/src/reader/mod.rs
@@ -490,6 +490,12 @@ impl<R: BufRead> Iterator for BufReader<R> {
     }
 }
 
+impl<R: BufRead> RecordBatchReader for BufReader<R> {
+    fn schema(&self) -> SchemaRef {
+        self.decoder.schema.clone()
+    }
+}
+
 /// A push-based interface for decoding CSV data from an arbitrary byte stream
 ///
 /// See [`Reader`] for a higher-level interface for interface with [`Read`]