linkrs-query 0.1.0

Query engine, parser, and executor for Linkrs
Documentation 
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

use crate::executor::expression::ExpressionError;
use linkrs_core::value::geography::{Geography, GeographyValue};
use linkrs_core::value::NullType;
use linkrs_core::Value;

pub fn execute_st_point(args: &[Value]) -> Result<Value, ExpressionError> {
    let (lon, lat) = match (&args[0], &args[1]) {
        (Value::Float(lon), Value::Float(lat)) => (*lon as f64, *lat as f64),
        (Value::Double(lon), Value::Double(lat)) => (*lon, *lat),
        (Value::SmallInt(lon), Value::SmallInt(lat)) => (*lon as f64, *lat as f64),
        (Value::Int(lon), Value::Int(lat)) => (*lon as f64, *lat as f64),
        (Value::BigInt(lon), Value::BigInt(lat)) => (*lon as f64, *lat as f64),
        (Value::Float(lon), Value::Double(lat)) => (*lon as f64, *lat),
        (Value::Double(lon), Value::Float(lat)) => (*lon, *lat as f64),
        (Value::Null(_), _) | (_, Value::Null(_)) => return Ok(Value::Null(NullType::Null)),
        _ => {
            return Err(ExpressionError::type_error(
                "The st_point function takes numeric arguments",
            ))
        }
    };

    let geo = Geography::Point(GeographyValue::new(lat, lon));
    Ok(Value::Geography(geo))
}

pub fn execute_st_geogfromtext(args: &[Value]) -> Result<Value, ExpressionError> {
    match &args[0] {
        Value::String(wkt) => match Geography::from_wkt(wkt) {
            Ok(geo) => Ok(Value::Geography(geo)),
            Err(e) => Err(ExpressionError::type_error(format!(
                "Failed to parse WKT: {}",
                e
            ))),
        },
        Value::Null(_) => Ok(Value::Null(NullType::Null)),
        _ => Err(ExpressionError::type_error(
            "The st_geogfromtext function takes string arguments",
        )),
    }
}

pub fn execute_st_geomfromgeojson(args: &[Value]) -> Result<Value, ExpressionError> {
    match &args[0] {
        Value::String(json_str) => match Geography::from_geojson_string(json_str) {
            Ok(geo) => Ok(Value::Geography(geo)),
            Err(e) => Err(ExpressionError::type_error(format!(
                "Invalid GeoJSON: {}",
                e
            ))),
        },
        Value::Null(_) => Ok(Value::Null(NullType::Null)),
        _ => Err(ExpressionError::type_error(
            "The st_geomfromgeojson function requires string argument",
        )),
    }
}