polars-plan 0.54.3

Lazy query engine for the Polars DataFrame library
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

use super::*;

#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "dsl-schema", derive(schemars::JsonSchema))]
#[derive(Clone, PartialEq, Debug, Hash)]
pub enum ExtensionFunction {
    To(DataTypeExpr),
    Storage,
}

impl Display for ExtensionFunction {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        use ExtensionFunction::*;
        match self {
            To(dtype) => write!(f, "ext.to({dtype:?})"),
            Storage => write!(f, "ext.storage()"),
        }
    }
}

impl From<ExtensionFunction> for FunctionExpr {
    fn from(func: ExtensionFunction) -> Self {
        FunctionExpr::Extension(func)
    }
}