mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-05-20 15:00:40 +00:00
180bc64cb0
* feat: change metrics to prometheus * chore: fix code advise * chore: resolve merge conflict * chore: fix code advise
61 lines
2.0 KiB
Rust
61 lines
2.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.
|
|
|
|
pub mod logging;
|
|
mod macros;
|
|
pub mod metric;
|
|
mod panic_hook;
|
|
|
|
use std::collections::hash_map::DefaultHasher;
|
|
use std::hash::{Hash, Hasher};
|
|
|
|
pub use logging::{init_default_ut_logging, init_global_logging, trace_id, TRACE_ID};
|
|
pub use metric::dump_metrics;
|
|
use once_cell::sync::OnceCell;
|
|
pub use panic_hook::set_panic_hook;
|
|
use parking_lot::Mutex;
|
|
use rand::random;
|
|
use snowflake::SnowflakeIdBucket;
|
|
pub use {common_error, tracing, tracing_appender, tracing_futures, tracing_subscriber};
|
|
|
|
static NODE_ID: OnceCell<u64> = OnceCell::new();
|
|
static TRACE_BUCKET: OnceCell<Mutex<SnowflakeIdBucket>> = OnceCell::new();
|
|
|
|
pub fn gen_trace_id() -> u64 {
|
|
let mut bucket = TRACE_BUCKET
|
|
.get_or_init(|| {
|
|
// if node_id is not initialized, how about random one?
|
|
let node_id = NODE_ID.get_or_init(|| 0);
|
|
info!("initializing bucket with node_id: {}", node_id);
|
|
let bucket = SnowflakeIdBucket::new(1, (*node_id) as i32);
|
|
Mutex::new(bucket)
|
|
})
|
|
.lock();
|
|
(*bucket).get_id() as u64
|
|
}
|
|
|
|
pub fn init_node_id(node_id: Option<String>) {
|
|
let node_id = node_id.map(|id| calculate_hash(&id)).unwrap_or(random());
|
|
match NODE_ID.set(node_id) {
|
|
Ok(_) => {}
|
|
Err(_) => warn!("node_id is already initialized"),
|
|
}
|
|
}
|
|
|
|
fn calculate_hash<T: Hash>(t: &T) -> u64 {
|
|
let mut s = DefaultHasher::new();
|
|
t.hash(&mut s);
|
|
s.finish()
|
|
}
|