mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-01-05 12:52:57 +00:00
89 lines
3.0 KiB
Rust
89 lines
3.0 KiB
Rust
// Copyright 2023 Greptime Team
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#![feature(error_iter)]
|
|
|
|
pub mod ext;
|
|
pub mod mock;
|
|
pub mod status_code;
|
|
|
|
use http::{HeaderMap, HeaderValue};
|
|
pub use snafu;
|
|
|
|
use crate::status_code::StatusCode;
|
|
|
|
// HACK - these headers are here for shared in gRPC services. For common HTTP headers,
|
|
// please define in `src/servers/src/http/header.rs`.
|
|
pub const GREPTIME_DB_HEADER_ERROR_CODE: &str = "x-greptime-err-code";
|
|
pub const GREPTIME_DB_HEADER_ERROR_MSG: &str = "x-greptime-err-msg";
|
|
|
|
/// Create a http header map from error code and message.
|
|
/// using `GREPTIME_DB_HEADER_ERROR_CODE` and `GREPTIME_DB_HEADER_ERROR_MSG` as keys.
|
|
pub fn from_err_code_msg_to_header(code: u32, msg: &str) -> HeaderMap {
|
|
let mut header = HeaderMap::new();
|
|
|
|
let msg = HeaderValue::from_str(msg).unwrap_or_else(|_| {
|
|
HeaderValue::from_bytes(
|
|
&msg.as_bytes()
|
|
.iter()
|
|
.flat_map(|b| std::ascii::escape_default(*b))
|
|
.collect::<Vec<u8>>(),
|
|
)
|
|
.expect("Already escaped string should be valid ascii")
|
|
});
|
|
|
|
header.insert(GREPTIME_DB_HEADER_ERROR_CODE, code.into());
|
|
header.insert(GREPTIME_DB_HEADER_ERROR_MSG, msg);
|
|
header
|
|
}
|
|
|
|
/// Extract [StatusCode] and error message from [HeaderMap], if any.
|
|
///
|
|
/// Note that if the [StatusCode] is illegal, for example, a random number that is not pre-defined
|
|
/// as a [StatusCode], the result is still `None`.
|
|
pub fn from_header_to_err_code_msg(headers: &HeaderMap) -> Option<(StatusCode, &str)> {
|
|
let code = headers
|
|
.get(GREPTIME_DB_HEADER_ERROR_CODE)
|
|
.and_then(|value| {
|
|
value
|
|
.to_str()
|
|
.ok()
|
|
.and_then(|x| x.parse::<u32>().ok())
|
|
.and_then(StatusCode::from_u32)
|
|
});
|
|
let msg = headers
|
|
.get(GREPTIME_DB_HEADER_ERROR_MSG)
|
|
.and_then(|x| x.to_str().ok());
|
|
match (code, msg) {
|
|
(Some(code), Some(msg)) => Some((code, msg)),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
/// Returns the external root cause of the source error (exclude the current error).
|
|
pub fn root_source(err: &dyn std::error::Error) -> Option<&dyn std::error::Error> {
|
|
// There are some divergence about the behavior of the `sources()` API
|
|
// in https://github.com/rust-lang/rust/issues/58520
|
|
// So this function iterates the sources manually.
|
|
let mut root = err.source();
|
|
while let Some(r) = root {
|
|
if let Some(s) = r.source() {
|
|
root = Some(s);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
root
|
|
}
|