Former-commit-id: 5a32ee75fb9d0974b3a6f464cdcf6634a444c049
This commit is contained in:
yu yunfeng 2019-06-28 16:20:17 +08:00
parent b9ad92692a
commit 9f82b5e351
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();