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())
}