micromegas_analytics/lakehouse/
answer.rs

1use datafusion::arrow::{array::RecordBatch, datatypes::Schema};
2use std::sync::Arc;
3
4/// Record batches + schema
5pub struct Answer {
6    /// The schema of the record batches.
7    pub schema: Arc<Schema>,
8    /// The record batches.
9    pub record_batches: Vec<RecordBatch>,
10}
11
12impl Answer {
13    pub fn new(schema: Arc<Schema>, record_batches: Vec<RecordBatch>) -> Self {
14        Self {
15            schema,
16            record_batches,
17        }
18    }
19
20    pub fn from_record_batch(rc: RecordBatch) -> Self {
21        Self {
22            schema: rc.schema(),
23            record_batches: vec![rc],
24        }
25    }
26}