Skip to content

Commit 0bb0d0e

Browse files
authored
feat(query): add metrics session_acquired_queries_total (#19087)
1 parent d377ab0 commit 0bb0d0e

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

src/common/metrics/src/metrics/session.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,8 @@ pub static SESSION_QUEUE_ACQUIRE_DURATION_MS: LazyLock<Histogram> =
4141

4242
pub static SESSION_RUNNING_ACQUIRED_QUERIES: LazyLock<Gauge> =
4343
LazyLock::new(|| register_gauge("session_running_acquired_queries"));
44+
pub static SESSION_ACQUIRED_QUERIES_TOTAL: LazyLock<Counter> =
45+
LazyLock::new(|| register_counter("session_acquired_queries_total"));
4446

4547
pub fn incr_session_connect_numbers() {
4648
SESSION_CONNECT_NUMBERS.inc();
@@ -81,3 +83,7 @@ pub fn inc_session_running_acquired_queries() {
8183
pub fn dec_session_running_acquired_queries() {
8284
SESSION_RUNNING_ACQUIRED_QUERIES.dec();
8385
}
86+
87+
pub fn inc_session_acquired_queries_total() {
88+
SESSION_ACQUIRED_QUERIES_TOTAL.inc();
89+
}

src/query/service/src/sessions/queue_mgr.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ use databend_common_meta_semaphore::errors::AcquireError;
4949
use databend_common_meta_store::MetaStore;
5050
use databend_common_meta_store::MetaStoreProvider;
5151
use databend_common_metrics::session::dec_session_running_acquired_queries;
52+
use databend_common_metrics::session::inc_session_acquired_queries_total;
5253
use databend_common_metrics::session::inc_session_running_acquired_queries;
5354
use databend_common_metrics::session::incr_session_queue_abort_count;
5455
use databend_common_metrics::session::incr_session_queue_acquire_error_count;
@@ -288,6 +289,7 @@ impl<Data: QueueData> QueueManager<Data> {
288289
guards.extend(self.acquire_warehouse_queue(data, timeout, instant).await?);
289290

290291
inc_session_running_acquired_queries();
292+
inc_session_acquired_queries_total();
291293
record_session_queue_acquire_duration_ms(start_time.elapsed().unwrap_or_default());
292294

293295
Ok(AcquireQueueGuard::create(guards))

0 commit comments

Comments
 (0)