mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-01-08 14:22:58 +00:00
abf34b845c
* tests: more mismatch errors * feat: check sink table schema if exists&prompt nice err msg * chore: rm unused variant * chore: fmt * chore: cargo clippy * feat: check schema on create * feat: better err msg when mismatch * tests: fix a schema mismatch * todo: create sink table * feat: create sink table * fix: find time index * tests: auto created sink table * fix: remove empty keys * refactor: per review * chore: fmt * test: sqlness * chore: after rebase
48 lines
1.7 KiB
Rust
48 lines
1.7 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;
|
|
|
|
// 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
|
|
}
|