Support search topk metric

Signed-off-by: chasingegg <[email protected]>
zilliztech · Aug 16, 2023 · bb92652 · bb92652
1 parent aa4a42f
commit bb92652
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/include/knowhere/prometheus_client.h b/include/knowhere/prometheus_client.h
@@ -77,5 +77,6 @@ extern const std::unique_ptr<PrometheusClient> prometheusClient;
 DECLARE_PROMETHEUS_COUNTER(knowhere_build_count);
 DECLARE_PROMETHEUS_COUNTER(knowhere_search_count);
 DECLARE_PROMETHEUS_COUNTER(knowhere_range_search_count);
+DECLARE_PROMETHEUS_HISTOGRAM(knowhere_search_topk);
 
 }  // namespace knowhere
diff --git a/src/common/index.cc b/src/common/index.cc
@@ -78,6 +78,7 @@ Index<T>::Search(const DataSet& dataset, const Json& json, const BitsetView& bit
 
 #ifdef NOT_COMPILE_FOR_SWIG
     knowhere_search_count.Increment();
+    knowhere_search_topk.Observe(cfg->k.value());
 #endif
     return this->node->Search(dataset, *cfg, bitset);
 }

diff --git a/src/common/prometheus_client.cc b/src/common/prometheus_client.cc
@@ -28,5 +28,6 @@ const std::unique_ptr<PrometheusClient> prometheusClient = std::make_unique<Prom
 DEFINE_PROMETHEUS_COUNTER(knowhere_build_count, "knowhere index build count")
 DEFINE_PROMETHEUS_COUNTER(knowhere_search_count, "knowhere search count")
 DEFINE_PROMETHEUS_COUNTER(knowhere_range_search_count, "knowhere range search count")
+DEFINE_PROMETHEUS_HISTOGRAM(knowhere_search_topk, "knowhere search topk")
 
 }  // namespace knowhere