Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
pola-rs
GitHub Repository: pola-rs/polars
Path: blob/main/crates/polars-plan/src/dsl/expr/named_serde.rs
6940 views
1
use std::sync::{Arc, LazyLock, RwLock};
2
3
use super::AnonymousColumnsUdf;
4
5
// Can be used to have named anonymous functions.
6
// The receiver must have implemented this registry and map the names to the proper UDFs.
7
pub trait ExprRegistry: Sync + Send {
8
fn get_function(&self, name: &str, payload: &[u8]) -> Option<Arc<dyn AnonymousColumnsUdf>>;
9
}
10
11
pub(super) static NAMED_SERDE_REGISTRY_EXPR: LazyLock<RwLock<Option<Arc<dyn ExprRegistry>>>> =
12
LazyLock::new(Default::default);
13
14
pub fn set_named_serde_registry(reg: Arc<dyn ExprRegistry>) {
15
let mut lock = NAMED_SERDE_REGISTRY_EXPR.write().unwrap();
16
*lock = Some(reg);
17
}
18
19