From 46bf344e560a5e32b1a59eaa09bd710b29e38027 Mon Sep 17 00:00:00 2001 From: Wangjunqi123 Date: Wed, 1 Nov 2023 10:12:40 +0800 Subject: [PATCH] define functions to connect to prometheus database --- server/dao/prometheus.go | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 server/dao/prometheus.go diff --git a/server/dao/prometheus.go b/server/dao/prometheus.go new file mode 100644 index 0000000..4f1bda1 --- /dev/null +++ b/server/dao/prometheus.go @@ -0,0 +1,33 @@ +package dao + +import ( + "github.com/prometheus/client_golang/api" + v1 "github.com/prometheus/client_golang/api/prometheus/v1" + "github.com/pkg/errors" +) + +var Prome *Prometheus + +type Prometheus struct { + addr string + Api v1.API +} + +func CreatePrometheus(url string) *Prometheus { + return &Prometheus{ + addr: url, + } +} + +func (p *Prometheus) CreateAPI() error { + client, err := api.NewClient(api.Config{Address: p.addr}) + if err != nil { + err = errors.Errorf("failed to create prometheus client: %s **2", err.Error()) + return err + } + + promapi := v1.NewAPI(client) + p.Api = promapi + + return nil +} -- Gitee