milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

line cov 

Former-commit-id: 98c5cb5ea36ee17f16966a9b6b3da1a084999299
pull/191/head
yu yunfeng 2019-06-28 16:20:17 +08:00 committed by jinhai
parent 0ca4160551
commit 46b26f075d
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cpp/src/metrics/PrometheusMetrics.h
View File

@ -46,7 +46,7 @@ class PrometheusMetrics: public MetricsBase {
std::shared_ptr<prometheus::Registry> registry_ = std::make_shared<prometheus::Registry>(); std::shared_ptr<prometheus::Registry> registry_ = std::make_shared<prometheus::Registry>();
bool startup_ = false; bool startup_ = false;
public: public:
void SetStartup(bool startup) {startup_ = startup;};
void AddVectorsSuccessTotalIncrement(double value = 1.0) override { if(startup_) add_vectors_success_total_.Increment(value);}; 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 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);}; void AddVectorsDurationHistogramOberve(double value) override { if(startup_) add_vectors_duration_histogram_.Observe(value);};

1
cpp/unittest/metrics/prometheus_test.cpp
View File

@ -14,6 +14,7 @@ using namespace zilliz::milvus;
TEST(PrometheusTest, Prometheus_Test){ TEST(PrometheusTest, Prometheus_Test){
server::PrometheusMetrics instance = server::PrometheusMetrics::GetInstance(); server::PrometheusMetrics instance = server::PrometheusMetrics::GetInstance();
instance.Init(); instance.Init();
instance.SetStartup(true);
server::SystemInfo::GetInstance().Init(); server::SystemInfo::GetInstance().Init();
instance.AddVectorsSuccessTotalIncrement(); instance.AddVectorsSuccessTotalIncrement();
instance.AddVectorsFailTotalIncrement(); instance.AddVectorsFailTotalIncrement();