package main import ( "context" "fmt" "log" "github.com/go-kratos/kratos/v2/middleware/metrics" prom "github.com/go-kratos/prometheus/metrics" "github.com/prometheus/client_golang/prometheus/promhttp" "github.com/go-kratos/kratos/examples/helloworld/helloworld" "github.com/go-kratos/kratos/v2" "github.com/go-kratos/kratos/v2/transport/grpc" "github.com/go-kratos/kratos/v2/transport/http" "github.com/prometheus/client_golang/prometheus" ) // go build -ldflags "-X main.Version=x.y.z" var ( // Name is the name of the compiled software. Name = "metrics" // Version is the version of the compiled software. Version = "v1.0.0" _metricSeconds = prometheus.NewHistogramVec(prometheus.HistogramOpts{ Namespace: "server", Subsystem: "requests", Name: "duration_ms", Help: "server requests duration(ms).", Buckets: []float64{5, 10, 25, 50, 100, 250, 500, 1000}, }, []string{"kind", "operation"}) _metricRequests = prometheus.NewCounterVec(prometheus.CounterOpts{ Namespace: "client", Subsystem: "requests", Name: "code_total", Help: "The total number of processed requests", }, []string{"kind", "operation", "code", "reason"}) ) // server is used to implement helloworld.GreeterServer. type server struct { helloworld.UnimplementedGreeterServer } // SayHello implements helloworld.GreeterServer func (s *server) SayHello(ctx context.Context, in *helloworld.HelloRequest) (*helloworld.HelloReply, error) { return &helloworld.HelloReply{Message: fmt.Sprintf("Hello %+v", in.Name)}, nil } func init() { prometheus.MustRegister(_metricSeconds, _metricRequests) } func main() { grpcSrv := grpc.NewServer( grpc.Address(":9000"), grpc.Middleware( metrics.Server( metrics.WithSeconds(prom.NewHistogram(_metricSeconds)), metrics.WithRequests(prom.NewCounter(_metricRequests)), ), ), ) httpSrv := http.NewServer( http.Address(":8000"), http.Middleware( metrics.Server( metrics.WithSeconds(prom.NewHistogram(_metricSeconds)), metrics.WithRequests(prom.NewCounter(_metricRequests)), ), ), ) httpSrv.Handle("/metrics", promhttp.Handler()) s := &server{} helloworld.RegisterGreeterServer(grpcSrv, s) helloworld.RegisterGreeterHTTPServer(httpSrv, s) app := kratos.New( kratos.Name(Name), kratos.Server( httpSrv, grpcSrv, ), ) if err := app.Run(); err != nil { log.Fatal(err) } }