[metrics] add copy_keys_qps and copy_time_us to DataStorageMetricsCollector

Author: fengfeisha

kv_cache_manager/metrics/metrics_collector.cc

@@ -196,6 +196,8 @@ DEFINE_METRICS_NAME_FOR_DATA_STORAGE(create_counter);
 DEFINE_METRICS_NAME_FOR_DATA_STORAGE(create_keys_qps);
 DEFINE_METRICS_NAME_FOR_DATA_STORAGE(create_keys_counter);
 DEFINE_METRICS_NAME_FOR_DATA_STORAGE(create_time_us);
+DEFINE_METRICS_NAME_FOR_DATA_STORAGE(copy_keys_qps);
+DEFINE_METRICS_NAME_FOR_DATA_STORAGE(copy_time_us);
 
 DataStorageMetricsCollector::DataStorageMetricsCollector(std::shared_ptr<MetricsRegistry> metrics_registry) noexcept
     : MetricsCollector(std::move(metrics_registry)) {}
@@ -213,6 +215,8 @@ bool DataStorageMetricsCollector::Init() {
     REGISTER_GAUGE_METRICS_FOR_DATA_STORAGE(create_keys_qps);
     REGISTER_COUNTER_METRICS_FOR_DATA_STORAGE(create_keys_counter);
     REGISTER_GAUGE_METRICS_FOR_DATA_STORAGE(create_time_us);
+    REGISTER_GAUGE_METRICS_FOR_DATA_STORAGE(copy_keys_qps);
+    REGISTER_GAUGE_METRICS_FOR_DATA_STORAGE(copy_time_us);
 
     return true;
 }

kv_cache_manager/metrics/metrics_collector.h

@@ -335,6 +335,9 @@ class DataStorageMetricsCollector final : public MetricsCollector {
 
     KVCM_CHRONO_METRICS(data_storage, create_time_us, DataStorageCreate)
 
+    KVCM_GAUGE_METRICS(data_storage, copy_keys_qps)
+    KVCM_CHRONO_METRICS(data_storage, copy_time_us, DataStorageCopy)
+
 public:
     DataStorageMetricsCollector() = delete;
     explicit DataStorageMetricsCollector(std::shared_ptr<MetricsRegistry> metrics_registry) noexcept;

kv_cache_manager/metrics/test/local_metrics_reporter_test.cc

@@ -110,16 +110,17 @@ TEST_F(LocalMetricsReporterTest, TestReportPerQuery00) {
 }
 
 TEST_F(LocalMetricsReporterTest, TestReportPerQuery01) {
+    constexpr int kDataStorageMetricsCount = 9;
     EXPECT_EQ(3, metrics_registry_->GetSize());
 
     DataStorageMetricsCollector collector(metrics_registry_);
     collector.Init();
 
-    EXPECT_EQ(7, metrics_registry_->GetSize());
+    EXPECT_EQ(kDataStorageMetricsCount, metrics_registry_->GetSize());
 
     {
         reporter_->ReportPerQuery(&collector);
-        EXPECT_EQ(7, metrics_registry_->GetSize());
+        EXPECT_EQ(kDataStorageMetricsCount, metrics_registry_->GetSize());
 
         std::uint64_t v;
         GET_METRICS_(&collector, data_storage, create_counter, v);
@@ -135,7 +136,7 @@ TEST_F(LocalMetricsReporterTest, TestReportPerQuery01) {
         g = 6.0;
 
         reporter_->ReportPerQuery(&collector);
-        EXPECT_EQ(7, metrics_registry_->GetSize());
+        EXPECT_EQ(kDataStorageMetricsCount, metrics_registry_->GetSize());
 
         std::uint64_t v;
         GET_METRICS_(&collector, data_storage, create_counter, v);
@@ -151,7 +152,7 @@ TEST_F(LocalMetricsReporterTest, TestReportPerQuery01) {
         g = 2.0;
 
         reporter_->ReportPerQuery(&collector);
-        EXPECT_EQ(7, metrics_registry_->GetSize());
+        EXPECT_EQ(kDataStorageMetricsCount, metrics_registry_->GetSize());
 
         std::uint64_t v;
         GET_METRICS_(&collector, data_storage, create_counter, v);