risingwave_common_metrics/monitor/
rwlock.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 prometheus::HistogramVec;
16use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
17
18pub struct MonitoredRwLock<T> {
19    // labels: [lock_name, lock_type]
20    metrics: HistogramVec,
21    inner: RwLock<T>,
22    lock_name: &'static str,
23}
24
25impl<T> MonitoredRwLock<T> {
26    pub fn new(metrics: HistogramVec, val: T, lock_name: &'static str) -> Self {
27        Self {
28            metrics,
29            inner: RwLock::new(val),
30            lock_name,
31        }
32    }
33
34    pub async fn read(&self) -> RwLockReadGuard<'_, T> {
35        let _timer = self
36            .metrics
37            .with_label_values(&[self.lock_name, "read"])
38            .start_timer();
39        self.inner.read().await
40    }
41
42    pub async fn write(&self) -> RwLockWriteGuard<'_, T> {
43        let _timer = self
44            .metrics
45            .with_label_values(&[self.lock_name, "write"])
46            .start_timer();
47        self.inner.write().await
48    }
49}
risingwave_common_metrics/monitor/rwlock.rs

risingwave_common_metrics/monitor/
rwlock.rs
