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
103
104
105
106
107
108
109
110
111

use std::ffi::CStr;
use std::fmt;

use zyxdb_sys as sys;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ErrorCode {
    InvalidArgument,
    NotFound,
    OpenFailed,
    ParseError,
    ExecutionError,
    TransactionError,
    ReadOnlyViolation,
    TypeMismatch,
    OutOfRange,
    IoError,
    OutOfMemory,
    InternalError,
    Unknown(i32),
}

impl ErrorCode {
    pub(crate) fn from_status(status: sys::zyx_driver_status_t) -> Self {
        match status {
            sys::ZYX_DRIVER_INVALID_ARGUMENT => Self::InvalidArgument,
            sys::ZYX_DRIVER_NOT_FOUND => Self::NotFound,
            sys::ZYX_DRIVER_OPEN_FAILED => Self::OpenFailed,
            sys::ZYX_DRIVER_PARSE_ERROR => Self::ParseError,
            sys::ZYX_DRIVER_EXECUTION_ERROR => Self::ExecutionError,
            sys::ZYX_DRIVER_TRANSACTION_ERROR => Self::TransactionError,
            sys::ZYX_DRIVER_READ_ONLY_VIOLATION => Self::ReadOnlyViolation,
            sys::ZYX_DRIVER_TYPE_MISMATCH => Self::TypeMismatch,
            sys::ZYX_DRIVER_OUT_OF_RANGE => Self::OutOfRange,
            sys::ZYX_DRIVER_IO_ERROR => Self::IoError,
            sys::ZYX_DRIVER_OUT_OF_MEMORY => Self::OutOfMemory,
            sys::ZYX_DRIVER_INTERNAL_ERROR => Self::InternalError,
            other => Self::Unknown(other),
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Error {
    code: ErrorCode,
    message: String,
}

impl Error {
    pub(crate) fn new(code: ErrorCode, message: impl Into<String>) -> Self {
        Self {
            code,
            message: message.into(),
        }
    }

    pub(crate) unsafe fn from_abi(
        status: sys::zyx_driver_status_t,
        error: *mut sys::zyx_driver_error_t,
    ) -> Self {
        if error.is_null() {
            return Self::new(
                ErrorCode::from_status(status),
                format!("ZYX driver error status {status}"),
            );
        }

        let code = ErrorCode::from_status(sys::zyx_driver_error_code(error));
        let message_ptr = sys::zyx_driver_error_message(error);
        let message = if message_ptr.is_null() {
            format!("ZYX driver error status {status}")
        } else {
            CStr::from_ptr(message_ptr).to_string_lossy().into_owned()
        };
        sys::zyx_driver_error_free(error);
        Self::new(code, message)
    }

    pub fn code(&self) -> ErrorCode {
        self.code
    }

    pub fn message(&self) -> &str {
        &self.message
    }
}

impl fmt::Display for Error {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{:?}: {}", self.code, self.message)
    }
}

impl std::error::Error for Error {}

impl From<std::ffi::NulError> for Error {
    fn from(error: std::ffi::NulError) -> Self {
        Self::new(ErrorCode::InvalidArgument, error.to_string())
    }
}

pub(crate) fn status_to_result(
    status: sys::zyx_driver_status_t,
    error: *mut sys::zyx_driver_error_t,
) -> crate::Result<()> {
    if status == sys::ZYX_DRIVER_OK {
        Ok(())
    } else {
        Err(unsafe { Error::from_abi(status, error) })
    }
}