rdf_fusion_functions/scalar/functional_form/
sparql_if.rs1use 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}