From 9f82b5e351b2b469f0f4e1da3d328e18bde57f41 Mon Sep 17 00:00:00 2001
From: yu yunfeng <yunfeng.yu@zilliz.com>
Date: Fri, 28 Jun 2019 16:20:17 +0800
Subject: [PATCH] line cov

Former-commit-id: 5a32ee75fb9d0974b3a6f464cdcf6634a444c049
---
 cpp/src/metrics/PrometheusMetrics.h      | 2 +-
 cpp/unittest/metrics/prometheus_test.cpp | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/cpp/src/metrics/PrometheusMetrics.h b/cpp/src/metrics/PrometheusMetrics.h
index b7e991e1c2..5b651ec14f 100644
--- a/cpp/src/metrics/PrometheusMetrics.h
+++ b/cpp/src/metrics/PrometheusMetrics.h
@@ -46,7 +46,7 @@ class PrometheusMetrics: public MetricsBase {
     std::shared_ptr<prometheus::Registry> registry_ = std::make_shared<prometheus::Registry>();
     bool startup_ = false;
  public:
-
+    void SetStartup(bool startup) {startup_ = startup;};
     void AddVectorsSuccessTotalIncrement(double value = 1.0) override { if(startup_) add_vectors_success_total_.Increment(value);};
     void AddVectorsFailTotalIncrement(double value = 1.0) override { if(startup_) add_vectors_fail_total_.Increment(value);};
     void AddVectorsDurationHistogramOberve(double value) override { if(startup_) add_vectors_duration_histogram_.Observe(value);};
diff --git a/cpp/unittest/metrics/prometheus_test.cpp b/cpp/unittest/metrics/prometheus_test.cpp
index f1e239d73b..885abed566 100644
--- a/cpp/unittest/metrics/prometheus_test.cpp
+++ b/cpp/unittest/metrics/prometheus_test.cpp
@@ -14,6 +14,7 @@ using namespace zilliz::milvus;
 TEST(PrometheusTest, Prometheus_Test){
     server::PrometheusMetrics instance = server::PrometheusMetrics::GetInstance();
     instance.Init();
+    instance.SetStartup(true);
     server::SystemInfo::GetInstance().Init();
     instance.AddVectorsSuccessTotalIncrement();
     instance.AddVectorsFailTotalIncrement();