rdf_fusion_functions/scalar/functional_form/
sparql_if.rs

1use crate::scalar::dispatch::dispatch_ternary_typed_value;
2use crate::scalar::sparql_op_impl::{
3    ScalarSparqlOpImpl, create_typed_value_sparql_op_impl,
4};
5use crate::scalar::{ScalarSparqlOp, ScalarSparqlOpSignature, SparqlOpArity};
6use rdf_fusion_encoding::typed_value::TypedValueEncoding;
7use rdf_fusion_extensions::functions::BuiltinName;
8use rdf_fusion_extensions::functions::FunctionName;
9use rdf_fusion_model::Boolean;
10
11#[derive(Debug, Hash, PartialEq, Eq)]
12pub struct IfSparqlOp;
13
14impl Default for IfSparqlOp {
15    fn default() -> Self {
16        Self::new()
17    }
18}
19
20impl IfSparqlOp {
21    const NAME: FunctionName = FunctionName::Builtin(BuiltinName::If);
22
23    pub fn new() -> Self {
24        Self {}
25    }
26}
27
28impl ScalarSparqlOp for IfSparqlOp {
29    fn name(&self) -> &FunctionName {
30        &Self::NAME
31    }
32
33    fn signature(&self) -> ScalarSparqlOpSignature {
34        ScalarSparqlOpSignature::default_with_arity(SparqlOpArity::Fixed(3))
35    }
36
37    fn typed_value_encoding_op(
38        &self,
39    ) -> Option<Box<dyn ScalarSparqlOpImpl<TypedValueEncoding>>> {
40        Some(create_typed_value_sparql_op_impl(|args| {
41            dispatch_ternary_typed_value(
42                &args.args[0],
43                &args.args[1],
44                &args.args[2],
45                |arg0, arg1, arg2| {
46                    let test = Boolean::try_from(arg0)?;
47                    if test.as_bool() { Ok(arg1) } else { Ok(arg2) }
48                },
49                |arg0, arg1, arg2| {
50                    let test = Boolean::try_from(arg0?)?;
51                    if test.as_bool() { arg1 } else { arg2 }
52                },
53            )
54        }))
55    }
56}