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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
use gitql_core::signature::Function;
use gitql_core::signature::Signature;
use gitql_core::types::DataType;
use gitql_core::value::Value;

use std::collections::HashMap;

#[inline(always)]
pub fn register_std_range_functions(map: &mut HashMap<&'static str, Function>) {
    map.insert("int4range", int4range);
    map.insert("daterange", daterange);
    map.insert("tsrange", tsrange);
}

#[inline(always)]
pub fn register_std_range_function_signatures(map: &mut HashMap<&'static str, Signature>) {
    map.insert(
        "int4range",
        Signature {
            parameters: vec![DataType::Integer, DataType::Integer],
            return_type: DataType::Range(Box::new(DataType::Integer)),
        },
    );
    map.insert(
        "daterange",
        Signature {
            parameters: vec![DataType::Date, DataType::Date],
            return_type: DataType::Range(Box::new(DataType::Date)),
        },
    );
    map.insert(
        "tsrange",
        Signature {
            parameters: vec![DataType::DateTime, DataType::DateTime],
            return_type: DataType::Range(Box::new(DataType::DateTime)),
        },
    );
}

pub fn int4range(inputs: &[Value]) -> Value {
    Value::Range(
        DataType::Integer,
        Box::new(inputs[0].clone()),
        Box::new(inputs[1].clone()),
    )
}

pub fn daterange(inputs: &[Value]) -> Value {
    Value::Range(
        DataType::Date,
        Box::new(inputs[0].clone()),
        Box::new(inputs[1].clone()),
    )
}

pub fn tsrange(inputs: &[Value]) -> Value {
    Value::Range(
        DataType::DateTime,
        Box::new(inputs[0].clone()),
        Box::new(inputs[1].clone()),
    )
}