From d2ab5206275c398ec26fe39a2326dc78382a1297 Mon Sep 17 00:00:00 2001 From: John Cai Date: Fri, 14 Feb 2020 11:41:21 -0800 Subject: Add method type metric middleware in praefect --- changelogs/unreleased/jc-add-method-type-metrics.yml | 5 +++++ internal/praefect/server.go | 2 ++ 2 files changed, 7 insertions(+) create mode 100644 changelogs/unreleased/jc-add-method-type-metrics.yml diff --git a/changelogs/unreleased/jc-add-method-type-metrics.yml b/changelogs/unreleased/jc-add-method-type-metrics.yml new file mode 100644 index 000000000..630ac6dd5 --- /dev/null +++ b/changelogs/unreleased/jc-add-method-type-metrics.yml @@ -0,0 +1,5 @@ +--- +title: Add method type metrics +merge_request: 1878 +author: +type: added diff --git a/internal/praefect/server.go b/internal/praefect/server.go index 76473adf5..4d95ce0bf 100644 --- a/internal/praefect/server.go +++ b/internal/praefect/server.go @@ -77,6 +77,7 @@ func NewServer(director proxy.StreamDirector, l *logrus.Entry, r *protoregistry. cancelhandler.Stream, // Should be below LogHandler grpctracing.StreamServerTracingInterceptor(), auth.StreamServerInterceptor(conf.Auth), + middleware.MethodTypeStreamInterceptor(r), // Panic handler should remain last so that application panics will be // converted to errors and logged panichandler.StreamPanicHandler, @@ -92,6 +93,7 @@ func NewServer(director proxy.StreamDirector, l *logrus.Entry, r *protoregistry. cancelhandler.Unary, // Should be below LogHandler grpctracing.UnaryServerTracingInterceptor(), auth.UnaryServerInterceptor(conf.Auth), + middleware.MethodTypeUnaryInterceptor(r), // Panic handler should remain last so that application panics will be // converted to errors and logged panichandler.UnaryPanicHandler, -- cgit v1.2.3