From b70e9ff90d2c4c6333f891e33c4a8a632913a0d4 Mon Sep 17 00:00:00 2001 From: bryanqiu Date: Wed, 21 Dec 2022 16:01:53 +0800 Subject: [PATCH] add instancee --- api.go | 14 +++++++++++++ context.go | 11 +---------- prometheus.go | 27 +++++++++++++++++-------- util.go | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 89 insertions(+), 18 deletions(-) diff --git a/api.go b/api.go index b55ccf0..109f4dd 100644 --- a/api.go +++ b/api.go @@ -111,6 +111,9 @@ func middlewareMyApiEngine() gin.HandlerFunc { // Step 3. do request c.Next() + + // Step 4. metrics record + cc.recordMetric() } } @@ -143,6 +146,17 @@ func ApiHandler(c *gin.Context) Handler { return newc.Keys["handler"].(Handler) } +func RecordMetrics(api, errcode, appid string, costms float64) { + var ( + counter = metricApiCounter.WithLabelValues(api, errcode, appid) + summary = metricApiSummary.WithLabelValues(api, errcode, appid) + ) + // Metric 1. api request counter + counter.Inc() + // Metric 2. api request cost/latency + summary.Observe(costms) +} + func DumpHandler(c *Context) error { errmsg := "api not implemented" errcode := errCodeUnimplementApi diff --git a/context.go b/context.go index ba46f20..bf0bc10 100644 --- a/context.go +++ b/context.go @@ -149,9 +149,6 @@ func (c *Context) RESULT(output interface{}) error { cost := time.Now().Sub(c.starttime).Milliseconds() noticeStr := fmt.Sprintf("cost: %dms, output: '%s'", cost, string(b)) log.NoticefWithDepth(calldepth, noticeStr) - - //// metric //////////////////////////////////////// - c.recordMetric() }(&output) if _, ok := t.FieldByName("ErrCode"); !ok { @@ -227,12 +224,6 @@ func (c *Context) recordMetric() { costus = time.Now().Sub(c.starttime).Microseconds() costms = float64(costus) / 1000.0 errcode = strconv.Itoa(c.errcode) - counter = metricApiCounter.WithLabelValues(api, errcode, appid) - summary = metricApiSummary.WithLabelValues(api, errcode, appid) ) - - // Metric 1. api request counter - counter.Inc() - // Metric 2. api request cost/latency - summary.Observe(costms) + RecordMetrics(api, errcode, appid, costms) } diff --git a/prometheus.go b/prometheus.go index 59ffc98..de25bd9 100644 --- a/prometheus.go +++ b/prometheus.go @@ -1,16 +1,18 @@ package api import ( + "errors" "fmt" "net/http" "github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus/promhttp" "github.com/prometheus/client_golang/prometheus/push" + "qoobing.com/gomod/log" ) var ( - MetricExporterPusher = func() {} + MetricExporterPusher = func() error { return nil } MetricExporterHandler Handler = nil metricRegistry *prometheus.Registry = nil metricApiCounter *prometheus.CounterVec = nil @@ -33,19 +35,27 @@ func (m *metricExporterHandler) HandlerFunc() HandlerFunc { } } +func dumpMetricExporterPusher() error { + return errors.New("not initilized, forgot call 'SetupMetricsExporterPusher'?") +} + func InitMetrics() { + instance := getInstanceIpAddress() + log.Infof("self instance ip: [%s]", instance) metricApiCounter = prometheus.NewCounterVec( prometheus.CounterOpts{ - Name: "api_requests_total", - Help: "How many HTTP requests processed", + Name: "api_requests_total", + Help: "How many HTTP requests processed", + ConstLabels: prometheus.Labels{"instance": instance}, }, []string{"api", "errcode", "appid"}, ) metricApiSummary = prometheus.NewSummaryVec( prometheus.SummaryOpts{ - Name: "api_requests_summary", - Help: "The api request summary of cost.", - Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001}, + Name: "api_requests_summary", + Help: "The api request summary of cost.", + Objectives: map[float64]float64{0.5: 0.05, 0.9: 0.01, 0.99: 0.001}, + ConstLabels: prometheus.Labels{"instance": instance}, }, []string{"api", "errcode", "appid"}, ) @@ -72,7 +82,8 @@ func SetupMetricsExporterPusher(pushgateway string, jobname string) { } var pusher = push.New(pushgateway, jobname).Gatherer(metricRegistry) - MetricExporterPusher = func() { - pusher.Push() + MetricExporterPusher = func() error { + RecordMetrics("push_metrics_to_prometheus", "0", "selfmonitoring", 1.0) + return pusher.Push() } } diff --git a/util.go b/util.go index 9f55402..ad522c0 100644 --- a/util.go +++ b/util.go @@ -1,8 +1,11 @@ package api import ( + "fmt" + "net" "os" "path/filepath" + "regexp" ) // getExeFilename @@ -13,3 +16,55 @@ func getExeFilename() string { } return logfilename } + +// preferencesMatch +func preferencesMatch(s string, preferences ...string) bool { + for _, pstr := range preferences { + reg := regexp.MustCompile(pstr) + if reg.Match([]byte(s)) { + return true + } + } + return false +} + +// getInstanceIpAddress +func getInstanceIpAddress(preferences ...string) string { + addrs, err := net.InterfaceAddrs() + if err != nil { + panic(fmt.Sprintf("getInstanceIpAddress failed: [%s]", err)) + } + + prefer := func() func(string) bool { + if len(preferences) <= 0 { + return nil + } + return func(ip string) bool { + return preferencesMatch(ip, preferences...) + } + }() + + var ipnets = []*net.IPNet{} + for _, address := range addrs { + if ipnet, ok := address.(*net.IPNet); !ok { + continue + } else if ipnet.IP.IsLoopback() { + continue + } else if ipv4 := ipnet.IP.To4(); ipv4 == nil { + continue + } else if prefer == nil { + var n = *ipnet + ipnets = append(ipnets, &n) + continue + } else if prefer != nil && prefer(ipv4.String()) { + return ipv4.String() + } + } + + // TODO: sort & return first + if len(ipnets) > 0 { + return ipnets[0].IP.To4().String() + } + panic(fmt.Sprintf("getInstanceIpAddreass failed,"+ + "preferences=[%v], ipnets=[%v]", preferences, ipnets)) +}