diff options
| author | John Cai <jcai@gitlab.com> | 2020-02-14 22:58:51 +0300 |
|---|---|---|
| committer | John Cai <jcai@gitlab.com> | 2020-02-17 22:50:02 +0300 |
| commit | 57fb842c30dd8b3518d77908f1d072964c35053d (patch) | |
| tree | 72b05b9cb60bd619ca42e0fcbbb9999575c9df1e | |
| parent | 2e2d2f6078ab3077abadd192de101b65703abbf4 (diff) | |
Add praefect client metric
| -rw-r--r-- | changelogs/unreleased/jc-add-praefect-client-metric.yml | 5 | ||||
| -rw-r--r-- | internal/praefect/nodes/manager.go | 3 |
2 files changed, 8 insertions, 0 deletions
diff --git a/changelogs/unreleased/jc-add-praefect-client-metric.yml b/changelogs/unreleased/jc-add-praefect-client-metric.yml new file mode 100644 index 000000000..74a920190 --- /dev/null +++ b/changelogs/unreleased/jc-add-praefect-client-metric.yml @@ -0,0 +1,5 @@ +--- +title: Add praefect client prometheus interceptor +merge_request: 1836 +author: +type: added diff --git a/internal/praefect/nodes/manager.go b/internal/praefect/nodes/manager.go index 3b090372d..d45affbfe 100644 --- a/internal/praefect/nodes/manager.go +++ b/internal/praefect/nodes/manager.go @@ -6,6 +6,7 @@ import ( "sync" "time" + grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus" "github.com/sirupsen/logrus" gitalyauth "gitlab.com/gitlab-org/gitaly/auth" "gitlab.com/gitlab-org/gitaly/client" @@ -87,6 +88,8 @@ func NewManager(log *logrus.Entry, c config.Config, dialOpts ...grpc.DialOption) []grpc.DialOption{ grpc.WithDefaultCallOptions(grpc.CallCustomCodec(proxy.Codec())), grpc.WithPerRPCCredentials(gitalyauth.RPCCredentials(node.Token)), + grpc.WithStreamInterceptor(grpc_prometheus.StreamClientInterceptor), + grpc.WithUnaryInterceptor(grpc_prometheus.UnaryClientInterceptor), }, dialOpts...), ) if err != nil { |