From f5ecc40f5d0a3cbc634889540c5f1662996eea3b Mon Sep 17 00:00:00 2001 From: chasingegg Date: Wed, 16 Aug 2023 12:34:08 +0800 Subject: [PATCH] Support search topk metric Signed-off-by: chasingegg --- include/knowhere/prometheus_client.h | 1 + src/common/index.cc | 1 + src/common/prometheus_client.cc | 1 + 3 files changed, 3 insertions(+) diff --git a/include/knowhere/prometheus_client.h b/include/knowhere/prometheus_client.h index dcfa309b2..1a28a46df 100644 --- a/include/knowhere/prometheus_client.h +++ b/include/knowhere/prometheus_client.h @@ -78,4 +78,5 @@ DECLARE_PROMETHEUS_HISTOGRAM(knowhere_build_latency); DECLARE_PROMETHEUS_HISTOGRAM(knowhere_search_latency); DECLARE_PROMETHEUS_HISTOGRAM(knowhere_range_search_latency); DECLARE_PROMETHEUS_HISTOGRAM(knowhere_ann_iterator_init_latency); +DECLARE_PROMETHEUS_HISTOGRAM(knowhere_search_topk); } // namespace knowhere diff --git a/src/common/index.cc b/src/common/index.cc index c43279064..82b8be2cd 100644 --- a/src/common/index.cc +++ b/src/common/index.cc @@ -82,6 +82,7 @@ Index::Search(const DataSet& dataset, const Json& json, const BitsetView& bit auto span = rc.ElapseFromBegin("done"); span *= 0.001; // convert to ms knowhere_search_latency.Observe(span); + knowhere_search_topk.Observe(cfg->k.value()); #else auto res = this->node->Search(dataset, *cfg, bitset); #endif diff --git a/src/common/prometheus_client.cc b/src/common/prometheus_client.cc index 0475eae82..b2e0065dc 100644 --- a/src/common/prometheus_client.cc +++ b/src/common/prometheus_client.cc @@ -29,5 +29,6 @@ DEFINE_PROMETHEUS_HISTOGRAM(knowhere_build_latency, "index build latency in know DEFINE_PROMETHEUS_HISTOGRAM(knowhere_search_latency, "search latency in knowhere (ms)") DEFINE_PROMETHEUS_HISTOGRAM(knowhere_range_search_latency, "range search latency in knowhere (ms)") DEFINE_PROMETHEUS_HISTOGRAM(knowhere_ann_iterator_init_latency, "ann iterator init latency in knowhere (ms)") +DEFINE_PROMETHEUS_HISTOGRAM(knowhere_search_topk, "knowhere search topk") } // namespace knowhere