generated from kubernetes/kubernetes-template-project
-
Notifications
You must be signed in to change notification settings - Fork 62
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
356 changed files
with
73,435 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,84 @@ | ||
| package metrics | ||
|
|
||
| import ( | ||
| "net" | ||
| "net/http" | ||
| "strings" | ||
|
|
||
| grpcprom "github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus" | ||
| "github.com/prometheus/client_golang/prometheus" | ||
| "google.golang.org/grpc" | ||
| "k8s.io/component-base/metrics/legacyregistry" | ||
| "k8s.io/klog/v2" | ||
| ) | ||
|
|
||
| var ( | ||
| // metricsList is a list of all raw metrics that should be registered always, regardless of any feature gate's value. | ||
| metricsList []prometheus.Collector | ||
| grpcServerMetrics *grpcprom.ServerMetrics | ||
| ) | ||
|
|
||
| // Register registers a list of metrics. | ||
| func Register() { | ||
| grpcServerMetrics = grpcprom.NewServerMetrics( | ||
| grpcprom.WithServerHandlingTimeHistogram( | ||
| grpcprom.WithHistogramBuckets([]float64{0.001, 0.01, 0.1, 0.3, 0.6, 1, 3, 6, 9, 20, 30, 60, 90, 120}), | ||
| ), | ||
| ) | ||
|
|
||
| metricsList = []prometheus.Collector{ | ||
| grpcServerMetrics, | ||
| } | ||
|
|
||
| for _, metric := range metricsList { | ||
| legacyregistry.RawMustRegister(metric) | ||
| } | ||
| } | ||
|
|
||
| // SetupMetricsServer creates an HTTP server to expose the gRPC server metrics | ||
| // at the`endpoint at `metricsAddress` | ||
| func SetupMetricsServer(metricsAddress string) error { | ||
| if metricsAddress == "" { | ||
| return nil | ||
| } | ||
| l, err := net.Listen("tcp", metricsAddress) | ||
| if err != nil { | ||
| klog.Warningf("failed to get listener for metrics endpoint: %v", err) | ||
| return err | ||
| } | ||
| path := l.Addr().String() | ||
| klog.V(2).Infof("set up prometheus server on %v", path) | ||
| go func() { | ||
| defer l.Close() | ||
|
|
||
| m := http.NewServeMux() | ||
| m.Handle("/metrics", legacyregistry.Handler()) | ||
| if err := trapClosedConnErr(http.Serve(l, m)); err != nil { | ||
| klog.Fatalf("serve failure(%v), address(%v)", err, path) | ||
| } | ||
| }() | ||
| return nil | ||
| } | ||
|
|
||
| func trapClosedConnErr(err error) error { | ||
| if err == nil { | ||
| return nil | ||
| } | ||
| if strings.Contains(err.Error(), "use of closed network connection") { | ||
| return nil | ||
| } | ||
| return err | ||
| } | ||
|
|
||
| // GRPCServerMetricsOptions returns the ServerOption applying on gRPC server | ||
| // to collect server metrics | ||
| func GRPCServerMetricsOptions() []grpc.ServerOption { | ||
| return []grpc.ServerOption{ | ||
| grpc.ChainUnaryInterceptor( | ||
| grpcServerMetrics.UnaryServerInterceptor(), | ||
| ), | ||
| grpc.ChainStreamInterceptor( | ||
| grpcServerMetrics.StreamServerInterceptor(), | ||
| ), | ||
| } | ||
| } |
Oops, something went wrong.