common_error/
status_code.rs1use std::fmt;
16
17use strum::{AsRefStr, EnumIter, EnumString, FromRepr};
18use tonic::Code;
19
20#[derive(Debug, Clone, Copy, PartialEq, Eq, EnumString, AsRefStr, EnumIter, FromRepr)]
22pub enum StatusCode {
23 Success = 0,
26
27 Unknown = 1000,
29 Unsupported = 1001,
31 Unexpected = 1002,
33 Internal = 1003,
35 InvalidArguments = 1004,
37 Cancelled = 1005,
39 IllegalState = 1006,
41 External = 1007,
43 DeadlineExceeded = 1008,
45 InvalidSyntax = 2000,
50 PlanQuery = 3000,
55 EngineExecuteQuery = 3001,
57 TableAlreadyExists = 4000,
62 TableNotFound = 4001,
64 TableColumnNotFound = 4002,
66 TableColumnExists = 4003,
68 DatabaseNotFound = 4004,
70 RegionNotFound = 4005,
72 RegionAlreadyExists = 4006,
74 RegionReadonly = 4007,
76 RegionNotReady = 4008,
78 RegionBusy = 4009,
80 TableUnavailable = 4010,
82 DatabaseAlreadyExists = 4011,
84 StorageUnavailable = 5000,
89 RequestOutdated = 5001,
91 RuntimeResourcesExhausted = 6000,
96
97 RateLimited = 6001,
99 UserNotFound = 7000,
104 UnsupportedPasswordType = 7001,
106 UserPasswordMismatch = 7002,
108 AuthHeaderNotFound = 7003,
110 InvalidAuthHeader = 7004,
112 AccessDenied = 7005,
114 PermissionDenied = 7006,
116 FlowAlreadyExists = 8000,
120 FlowNotFound = 8001,
121 TriggerAlreadyExists = 9000,
125 TriggerNotFound = 9001,
126 }
128
129impl StatusCode {
130 pub fn is_success(code: u32) -> bool {
132 Self::Success as u32 == code
133 }
134
135 pub fn is_retryable(&self) -> bool {
137 match self {
138 StatusCode::StorageUnavailable
139 | StatusCode::RuntimeResourcesExhausted
140 | StatusCode::Internal
141 | StatusCode::RegionNotReady
142 | StatusCode::TableUnavailable
143 | StatusCode::RegionBusy => true,
144
145 StatusCode::Success
146 | StatusCode::Unknown
147 | StatusCode::Unsupported
148 | StatusCode::IllegalState
149 | StatusCode::Unexpected
150 | StatusCode::InvalidArguments
151 | StatusCode::Cancelled
152 | StatusCode::DeadlineExceeded
153 | StatusCode::InvalidSyntax
154 | StatusCode::DatabaseAlreadyExists
155 | StatusCode::PlanQuery
156 | StatusCode::EngineExecuteQuery
157 | StatusCode::TableAlreadyExists
158 | StatusCode::TableNotFound
159 | StatusCode::RegionAlreadyExists
160 | StatusCode::RegionNotFound
161 | StatusCode::FlowAlreadyExists
162 | StatusCode::FlowNotFound
163 | StatusCode::TriggerAlreadyExists
164 | StatusCode::TriggerNotFound
165 | StatusCode::RegionReadonly
166 | StatusCode::TableColumnNotFound
167 | StatusCode::TableColumnExists
168 | StatusCode::DatabaseNotFound
169 | StatusCode::RateLimited
170 | StatusCode::UserNotFound
171 | StatusCode::UnsupportedPasswordType
172 | StatusCode::UserPasswordMismatch
173 | StatusCode::AuthHeaderNotFound
174 | StatusCode::InvalidAuthHeader
175 | StatusCode::AccessDenied
176 | StatusCode::PermissionDenied
177 | StatusCode::RequestOutdated
178 | StatusCode::External => false,
179 }
180 }
181
182 pub fn should_log_error(&self) -> bool {
185 match self {
186 StatusCode::Unknown
187 | StatusCode::Unexpected
188 | StatusCode::Internal
189 | StatusCode::Cancelled
190 | StatusCode::DeadlineExceeded
191 | StatusCode::IllegalState
192 | StatusCode::EngineExecuteQuery
193 | StatusCode::StorageUnavailable
194 | StatusCode::RuntimeResourcesExhausted
195 | StatusCode::External => true,
196
197 StatusCode::Success
198 | StatusCode::Unsupported
199 | StatusCode::InvalidArguments
200 | StatusCode::InvalidSyntax
201 | StatusCode::TableAlreadyExists
202 | StatusCode::TableNotFound
203 | StatusCode::RegionAlreadyExists
204 | StatusCode::RegionNotFound
205 | StatusCode::PlanQuery
206 | StatusCode::FlowAlreadyExists
207 | StatusCode::FlowNotFound
208 | StatusCode::TriggerAlreadyExists
209 | StatusCode::TriggerNotFound
210 | StatusCode::RegionNotReady
211 | StatusCode::RegionBusy
212 | StatusCode::RegionReadonly
213 | StatusCode::TableColumnNotFound
214 | StatusCode::TableColumnExists
215 | StatusCode::DatabaseNotFound
216 | StatusCode::RateLimited
217 | StatusCode::UserNotFound
218 | StatusCode::TableUnavailable
219 | StatusCode::DatabaseAlreadyExists
220 | StatusCode::UnsupportedPasswordType
221 | StatusCode::UserPasswordMismatch
222 | StatusCode::AuthHeaderNotFound
223 | StatusCode::InvalidAuthHeader
224 | StatusCode::AccessDenied
225 | StatusCode::PermissionDenied
226 | StatusCode::RequestOutdated => false,
227 }
228 }
229
230 pub fn from_u32(value: u32) -> Option<Self> {
231 StatusCode::from_repr(value as usize)
232 }
233}
234
235impl fmt::Display for StatusCode {
236 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
237 write!(f, "{self:?}")
239 }
240}
241
242#[macro_export]
243macro_rules! define_into_tonic_status {
244 ($Error: ty) => {
245 impl From<$Error> for tonic::Status {
246 fn from(err: $Error) -> Self {
247 use tonic::codegen::http::{HeaderMap, HeaderValue};
248 use tonic::metadata::MetadataMap;
249 use $crate::GREPTIME_DB_HEADER_ERROR_CODE;
250
251 let mut headers = HeaderMap::<HeaderValue>::with_capacity(2);
252
253 let status_code = err.status_code();
256 headers.insert(
257 GREPTIME_DB_HEADER_ERROR_CODE,
258 HeaderValue::from(status_code as u32),
259 );
260 let root_error = err.output_msg();
261
262 let metadata = MetadataMap::from_headers(headers);
263 tonic::Status::with_metadata(
264 $crate::status_code::status_to_tonic_code(status_code),
265 root_error,
266 metadata,
267 )
268 }
269 }
270 };
271}
272
273pub fn status_to_tonic_code(status_code: StatusCode) -> Code {
275 match status_code {
276 StatusCode::Success => Code::Ok,
277 StatusCode::Unknown | StatusCode::External => Code::Unknown,
278 StatusCode::Unsupported => Code::Unimplemented,
279 StatusCode::Unexpected
280 | StatusCode::IllegalState
281 | StatusCode::Internal
282 | StatusCode::PlanQuery
283 | StatusCode::EngineExecuteQuery => Code::Internal,
284 StatusCode::InvalidArguments | StatusCode::InvalidSyntax | StatusCode::RequestOutdated => {
285 Code::InvalidArgument
286 }
287 StatusCode::Cancelled => Code::Cancelled,
288 StatusCode::DeadlineExceeded => Code::DeadlineExceeded,
289 StatusCode::TableAlreadyExists
290 | StatusCode::TableColumnExists
291 | StatusCode::RegionAlreadyExists
292 | StatusCode::DatabaseAlreadyExists
293 | StatusCode::TriggerAlreadyExists
294 | StatusCode::FlowAlreadyExists => Code::AlreadyExists,
295 StatusCode::TableNotFound
296 | StatusCode::RegionNotFound
297 | StatusCode::TableColumnNotFound
298 | StatusCode::DatabaseNotFound
299 | StatusCode::UserNotFound
300 | StatusCode::TriggerNotFound
301 | StatusCode::FlowNotFound => Code::NotFound,
302 StatusCode::TableUnavailable
303 | StatusCode::StorageUnavailable
304 | StatusCode::RegionNotReady => Code::Unavailable,
305 StatusCode::RuntimeResourcesExhausted
306 | StatusCode::RateLimited
307 | StatusCode::RegionBusy => Code::ResourceExhausted,
308 StatusCode::UnsupportedPasswordType
309 | StatusCode::UserPasswordMismatch
310 | StatusCode::AuthHeaderNotFound
311 | StatusCode::InvalidAuthHeader => Code::Unauthenticated,
312 StatusCode::AccessDenied | StatusCode::PermissionDenied | StatusCode::RegionReadonly => {
313 Code::PermissionDenied
314 }
315 }
316}
317
318pub fn convert_tonic_code_to_status_code(code: Code) -> StatusCode {
320 match code {
321 Code::Cancelled => StatusCode::Cancelled,
322 Code::DeadlineExceeded => StatusCode::DeadlineExceeded,
323 _ => StatusCode::Internal,
324 }
325}
326
327#[cfg(test)]
328mod tests {
329 use strum::IntoEnumIterator;
330
331 use super::*;
332
333 fn assert_status_code_display(code: StatusCode, msg: &str) {
334 let code_msg = format!("{code}");
335 assert_eq!(msg, code_msg);
336 }
337
338 #[test]
339 fn test_display_status_code() {
340 assert_status_code_display(StatusCode::Unknown, "Unknown");
341 assert_status_code_display(StatusCode::TableAlreadyExists, "TableAlreadyExists");
342 }
343
344 #[test]
345 fn test_from_u32() {
346 for code in StatusCode::iter() {
347 let num = code as u32;
348 assert_eq!(StatusCode::from_u32(num), Some(code));
349 }
350
351 assert_eq!(StatusCode::from_u32(10000), None);
352 }
353
354 #[test]
355 fn test_is_success() {
356 assert!(StatusCode::is_success(0));
357 assert!(!StatusCode::is_success(1));
358 assert!(!StatusCode::is_success(2));
359 assert!(!StatusCode::is_success(3));
360 }
361}