(&self)
| 182 | } |
| 183 | |
| 184 | fn reader(&self) -> Result<Box<dyn RecordBatchReader>> { |
| 185 | let file = File::open(&self.location)?; |
| 186 | let schema = Arc::clone(&self.schema); |
| 187 | match &self.encoding { |
| 188 | StreamEncoding::Csv => { |
| 189 | let reader = arrow::csv::ReaderBuilder::new(schema) |
| 190 | .with_header(self.header) |
| 191 | .with_batch_size(self.batch_size) |
| 192 | .build(file)?; |
| 193 | |
| 194 | Ok(Box::new(reader)) |
| 195 | } |
| 196 | StreamEncoding::Json => { |
| 197 | let reader = arrow::json::ReaderBuilder::new(schema) |
| 198 | .with_batch_size(self.batch_size) |
| 199 | .build(BufReader::new(file))?; |
| 200 | |
| 201 | Ok(Box::new(reader)) |
| 202 | } |
| 203 | } |
| 204 | } |
| 205 | |
| 206 | fn writer(&self) -> Result<Box<dyn RecordBatchWriter>> { |
| 207 | match &self.encoding { |
no test coverage detected