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

View File

@ -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);};

View File

@ -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();