risingwave_common/util/tracing/
layer.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::sync::OnceLock;
16
17static TOGGLE_OTEL_LAYER: OnceLock<Box<dyn Fn(bool) + Sync + Send>> = OnceLock::new();
18
19/// Set the function to toggle the opentelemetry tracing layer. Panics if called twice.
20pub fn set_toggle_otel_layer_fn(f: impl Fn(bool) + Sync + Send + 'static) {
21    TOGGLE_OTEL_LAYER
22        .set(Box::new(f))
23        .ok()
24        .expect("toggle otel layer fn set twice");
25}
26
27/// Toggle the opentelemetry tracing layer.
28pub fn toggle_otel_layer(enabled: bool) {
29    if let Some(f) = TOGGLE_OTEL_LAYER.get() {
30        f(enabled);
31    }
32}
risingwave_common/util/tracing/layer.rs

risingwave_common/util/tracing/
layer.rs
