use datafusion::arrow::{array::RecordBatch, datatypes::Schema};
use std::sync::Arc;
/// Record batches + schema
pub struct Answer {
pub schema: Arc<Schema>,
pub record_batches: Vec<RecordBatch>,
}
impl Answer {
pub fn new(schema: Arc<Schema>, record_batches: Vec<RecordBatch>) -> Self {
Self {
schema,
record_batches,
}
}
pub fn from_record_batch(rc: RecordBatch) -> Self {
Self {
schema: rc.schema(),
record_batches: vec![rc],
}
}
}