Skip to content

Commit 027e79d

Browse files
committed
draft: using arrow-avro
1 parent e27e1f8 commit 027e79d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

datafusion/datasource-avro/src/source.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
use std::any::Any;
2121
use std::sync::Arc;
2222

23-
use arrow::datatypes::{Schema, SchemaRef};
23+
use arrow::datatypes::SchemaRef;
2424
use arrow_avro::reader::{Reader, ReaderBuilder};
2525
use arrow_avro::schema::AvroSchema;
2626
use datafusion_common::error::Result;
@@ -52,8 +52,7 @@ impl AvroSource {
5252
}
5353

5454
fn open<R: std::io::Read>(&self, reader: R) -> Result<Reader<R>> {
55-
let schema: &Schema = self.schema.as_ref().expect("Schema must set before open");
56-
let avro_schema = AvroSchema::try_from(schema)?;
55+
let avro_schema = AvroSchema::try_from(self.schema.expect("Schema must set before open").as_ref())?;
5756
ReaderBuilder::new()
5857
.with_reader_schema(avro_schema)
5958
.with_batch_size(self.batch_size.expect("Batch size must set before open"))

0 commit comments

Comments
 (0)