Use raw vec instead of Cursor

parquet/benches/arrow_writer.rs

@@ -17,9 +17,7 @@
 
 #[macro_use]
 extern crate criterion;
-
 use criterion::{Criterion, Throughput};
-use std::io::Cursor;
 
 extern crate arrow;
 extern crate parquet;
@@ -278,8 +276,8 @@ fn _create_nested_bench_batch(
 #[inline]
 fn write_batch(batch: &RecordBatch) -> Result<()> {
     // Write batch to an in-memory writer
-    let cursor = Cursor::new(vec![]);
-    let mut writer = ArrowWriter::try_new(cursor, batch.schema(), None)?;
+    let buffer = vec![];
+    let mut writer = ArrowWriter::try_new(buffer, batch.schema(), None)?;
 
     writer.write(batch)?;
     writer.close()?;

parquet/src/arrow/arrow_writer.rs

@@ -695,7 +695,6 @@ mod tests {
     use super::*;
 
     use std::fs::File;
-    use std::io::Cursor;
     use std::sync::Arc;
 
     use arrow::datatypes::ToByteSlice;
@@ -748,16 +747,14 @@ mod tests {
         let expected_batch =
             RecordBatch::try_new(schema.clone(), vec![Arc::new(a), Arc::new(b)]).unwrap();
 
-        let mut cursor = Cursor::new(vec![]);
+        let mut buffer = vec![];
 
         {
-            let mut writer = ArrowWriter::try_new(&mut cursor, schema, None).unwrap();
+            let mut writer = ArrowWriter::try_new(&mut buffer, schema, None).unwrap();
             writer.write(&expected_batch).unwrap();
             writer.close().unwrap();
         }
 
-        let buffer = cursor.into_inner();
-
         let cursor = crate::file::serialized_reader::SliceableCursor::new(buffer);
         let reader = SerializedFileReader::new(cursor).unwrap();
         let mut arrow_reader = ParquetFileArrowReader::new(Arc::new(reader));