mirror of https://github.com/milvus-io/milvus.git
parent
0ca4160551
commit
46b26f075d
|
@ -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);};
|
||||
|
|
|
@ -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();
|
||||
|
|
Loading…
Reference in New Issue