polars_core/query_result.rs
1use crate::frame::DataFrame;
2
3pub enum QueryResult {
4 Single(DataFrame),
5 /// Collected to multiple in-memory sinks
6 Multiple(Vec<DataFrame>),
7}
8
9impl QueryResult {
10 pub fn unwrap_single(self) -> DataFrame {
11 use QueryResult::*;
12 match self {
13 Single(df) => df,
14 Multiple(_) => panic!(),
15 }
16 }
17
18 pub fn unwrap_multiple(self) -> Vec<DataFrame> {
19 use QueryResult::*;
20 match self {
21 Single(_) => panic!(),
22 Multiple(dfs) => dfs,
23 }
24 }
25}