api_response/error_code/
mod.rs

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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
mod errtype;
pub mod ety_grpc;
mod modpath;
mod segment;

use std::{
    fmt::Display,
    ops::{Add, BitOr},
};

pub use errtype::*;
pub use modpath::*;
pub use segment::*;

use crate::ApiError;

#[derive(Debug, Clone, Copy, Eq, PartialEq, Hash)]
#[non_exhaustive]
pub struct ErrCode {
    pub err_type: ErrType,
    pub mod_path: ModPath,
}

impl From<ErrCode> for i32 {
    fn from(value: ErrCode) -> Self {
        value.code()
    }
}
impl ErrCode {
    pub const fn new(err_type: ErrType, mod_path: ModPath) -> Self {
        Self { err_type, mod_path }
    }
    pub fn code(&self) -> i32 {
        self.err_type.err_segment()
            | self.mod_path.mod1_segment()
            | self.mod_path.mod2_segment()
            | self.mod_path.mod3_segment()
    }
    pub const fn err_type(&self) -> &ErrType {
        &self.err_type
    }
    pub const fn err_segment(&self) -> ErrSegment {
        self.err_type.err_segment()
    }
    pub const fn text(&self) -> &'static str {
        self.err_type.text()
    }
    pub const fn mod_path(&self) -> ModPath {
        self.mod_path
    }
    pub const fn mod1(&self) -> ModSection {
        self.mod_path.mod1()
    }
    pub const fn mod2(&self) -> ModSection {
        self.mod_path.mod2()
    }
    pub const fn mod3(&self) -> ModSection {
        self.mod_path.mod3()
    }
    pub const fn with_mod1(mut self, mod1: ModSection) -> Self {
        self.mod_path.set_mod1(mod1);
        self
    }
    pub const fn with_mod2(mut self, mod2: ModSection) -> Self {
        self.mod_path.set_mod2(mod2);
        self
    }
    pub const fn with_mod3(mut self, mod3: ModSection) -> Self {
        self.mod_path.set_mod3(mod3);
        self
    }
    pub fn to_api_error(self) -> ApiError {
        ApiError {
            code: self.code(),
            message: self.text().to_owned(),
            details: None,
            source: None,
        }
    }
}

impl Display for ErrCode {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{} ErrCode({}), {}", self.text(), self.code(), self.mod_path)
    }
}

impl BitOr<ModPath> for ErrType {
    type Output = ErrCode;

    #[inline]
    fn bitor(self, rhs: ModPath) -> Self::Output {
        self.new_err_code(rhs)
    }
}
impl Add<ModPath> for ErrType {
    type Output = ApiError;

    #[inline]
    fn add(self, rhs: ModPath) -> Self::Output {
        self.new_api_error(rhs)
    }
}
impl BitOr<&ModPath> for ErrType {
    type Output = ErrCode;

    #[inline]
    fn bitor(self, rhs: &ModPath) -> Self::Output {
        self.new_err_code(*rhs)
    }
}
impl Add<&ModPath> for ErrType {
    type Output = ApiError;

    #[inline]
    fn add(self, rhs: &ModPath) -> Self::Output {
        self.new_api_error(*rhs)
    }
}

#[cfg(test)]
mod tests {
    use std::cell::LazyCell;

    use super::{ErrCode, ErrSegment, ErrType, ModPath, ModSection, ModSegment};
    use crate::ApiError;

    #[test]
    fn display() {
        const ET: ErrType = ErrType::new(ErrSegment::E100, "The operation was cancelled.");
        const MS0: ModSection = ModSection::new(ModSegment::M00, "module 0");
        const MS1: ModSection = ModSection::new(ModSegment::M01, "module 01");
        const MS2: ModSection = ModSection::new(ModSegment::M02, "module 012");
        const MP: ModPath = ModPath::new(MS0, MS1, MS2);
        const EC: ErrCode = ErrCode::new(ET, MP);
        assert_eq!(
            "The operation was cancelled. ErrCode(100000102), M00(module 0)/M01(module 01)/M02(module 012)",
            EC.to_string()
        );

        let err_code: ErrCode = ET | MP;
        assert_eq!(EC, err_code);
        let api_error: ApiError = ET + MP;
        assert_eq!(EC.to_api_error().code(), api_error.code());
        let mp: LazyCell<ModPath> = LazyCell::new(|| MP);
        let err_code: ErrCode = ET | *mp;
        assert_eq!(EC, err_code);
        let api_error: ApiError = ET + *mp;
        assert_eq!(EC.to_api_error().code(), api_error.code());
    }
}