返回顶部

9.【go-kit教程】go-kit集成Prometheus

  • 在 Go kit 中集成 Prometheus 进行 API 监控可以帮助开发人员更好地了解系统的性能和行为,提高系统的可观察性和可靠性。下面是一个简单的示例,演示如何在 Go kit 中集成 Prometheus 进行 API 监控:

    package main
    
    import (
    	"net/http"
    
    	"github.com/go-kit/kit/log"
    	"github.com/go-kit/kit/metrics/prometheus"
    	"github.com/go-kit/kit/metrics/provider"
    	"github.com/go-kit/kit/transport"
    	httptransport "github.com/go-kit/kit/transport/http"
    	"github.com/prometheus/client_golang/prometheus/promhttp"
    )
    
    func main() {
    	// 创建 Prometheus 监控器
    	fieldKeys := []string{"method", "error"}
    	requestCount := prometheus.NewCounterFrom(provider.NewProvider(), prometheus.CounterOpts{
    		Namespace: "my_api",
    		Name:      "request_count",
    		Help:      "Number of requests received.",
    	}, fieldKeys)
    	requestLatency := prometheus.NewSummaryFrom(provider.NewProvider(), prometheus.SummaryOpts{
    		Namespace: "my_api",
    		Name:      "request_latency",
    		Help:      "Total duration of requests in microseconds.",
    	}, fieldKeys)
    
    	// 创建 endpoint
    	helloEndpoint := func() string {
    		return "Hello, World!"
    	}
    	helloEndpoint = transport.NewServer(
    		helloEndpoint,
    		func(ctx context.Context, r *http.Request) (interface{}, error) {
    			return nil, nil
    		},
    		func(ctx context.Context, w http.ResponseWriter, response interface{}) error {
    			w.Header().Set("Content-Type", "text/plain; charset=utf-8")
    			fmt.Fprint(w, response.(string))
    			return nil
    		},
    		transport.ServerErrorLogger(log.NewNopLogger()),
    		transport.ServerErrorEncoder(errorEncoder),
    		transport.ServerBefore(prometheus.HTTPToContext()),
    	).Endpoint()
    
    	// 创建 HTTP server
    	r := mux.NewRouter()
    	r.Handle("/hello", httptransport.NewServer(
    		helloEndpoint,
    		httpDecodeRequest,
    		httpEncodeResponse,
    		httptransport.ServerBefore(prometheus.HTTPToContext()),
    		httptransport.ServerErrorLogger(log.NewNopLogger()),
    		httptransport.ServerErrorEncoder(errorEncoder),
    		httptransport.ServerBefore(prometheus.HTTPToContext()),
    	)).Methods("GET")
    	r.Handle("/metrics", promhttp.Handler())
    
    	fmt.Println("listening on :8080...")
    	http.ListenAndServe(":8080", r)
    }
    
    func httpDecodeRequest(_ context.Context, r *http.Request) (interface{}, error) {
    	return nil, nil
    }
    
    func httpEncodeResponse(_ context.Context, w http.ResponseWriter, response interface{}) error {
    	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
    	fmt.Fprint(w, response.(string))
    	return nil
    }
    
    func errorEncoder(_ context.Context, err error, w http.ResponseWriter) {
    	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
    	w.WriteHeader(http.StatusBadRequest)
    	fmt.Fprint(w, err.Error())
    }
    
    
posted @ 2023-02-25 22:03  高薪程序员  阅读(90)  评论(0编辑  收藏  举报