zyxdb 0.1.28

Rust bindings for the ZYX embedded graph database
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use std::ffi::CString;
use std::ptr;

use zyxdb_sys as sys;

use crate::error::status_to_result;

pub struct Params {
    pub(crate) raw: *mut sys::zyx_driver_params_t,
}

impl Params {
    pub fn try_new() -> crate::Result<Self> {
        let mut raw = ptr::null_mut();
        let mut error = ptr::null_mut();
        let status = unsafe { sys::zyx_driver_params_create(&mut raw, &mut error) };
        status_to_result(status, error)?;
        Ok(Self { raw })
    }

    pub fn set<V: IntoParam>(self, key: &str, value: V) -> crate::Result<Self> {
        value.set_param(&self, key)?;
        Ok(self)
    }
}

impl Drop for Params {
    fn drop(&mut self) {
        if !self.raw.is_null() {
            let mut error = ptr::null_mut();
            unsafe {
                sys::zyx_driver_params_free(self.raw, &mut error);
                if !error.is_null() {
                    sys::zyx_driver_error_free(error);
                }
            }
        }
    }
}

pub trait IntoParam {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()>;
}

fn key_cstring(key: &str) -> crate::Result<CString> {
    CString::new(key).map_err(Into::into)
}

impl IntoParam for i64 {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        let key = key_cstring(key)?;
        let mut error = ptr::null_mut();
        let status =
            unsafe { sys::zyx_driver_params_set_int64(params.raw, key.as_ptr(), self, &mut error) };
        status_to_result(status, error)
    }
}

impl IntoParam for i32 {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        i64::from(self).set_param(params, key)
    }
}

impl IntoParam for f64 {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        let key = key_cstring(key)?;
        let mut error = ptr::null_mut();
        let status = unsafe {
            sys::zyx_driver_params_set_double(params.raw, key.as_ptr(), self, &mut error)
        };
        status_to_result(status, error)
    }
}

impl IntoParam for bool {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        let key = key_cstring(key)?;
        let mut error = ptr::null_mut();
        let status =
            unsafe { sys::zyx_driver_params_set_bool(params.raw, key.as_ptr(), self, &mut error) };
        status_to_result(status, error)
    }
}

impl IntoParam for &str {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        let key = key_cstring(key)?;
        let value = CString::new(self)?;
        let mut error = ptr::null_mut();
        let status = unsafe {
            sys::zyx_driver_params_set_string(params.raw, key.as_ptr(), value.as_ptr(), &mut error)
        };
        status_to_result(status, error)
    }
}

impl IntoParam for String {
    fn set_param(self, params: &Params, key: &str) -> crate::Result<()> {
        self.as_str().set_param(params, key)
    }
}