GRPC Server: Add tracing interceptors (#56045)

Co-authored-by: Artur Wierzbicki <artur.wierzbicki@grafana.com>

pkg/services/grpcserver/service.go

@@ -7,11 +7,14 @@ import (
 
 	"github.com/grafana/grafana-plugin-sdk-go/backend"
 	"github.com/grafana/grafana/pkg/infra/log"
+	"github.com/grafana/grafana/pkg/infra/tracing"
 	"github.com/grafana/grafana/pkg/registry"
 	"github.com/grafana/grafana/pkg/services/apikey"
 	"github.com/grafana/grafana/pkg/services/featuremgmt"
+	"github.com/grafana/grafana/pkg/services/grpcserver/interceptors"
 	"github.com/grafana/grafana/pkg/services/user"
 	"github.com/grafana/grafana/pkg/setting"
+	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
 	grpcAuth "github.com/grpc-ecosystem/go-grpc-middleware/auth"
 
 	"google.golang.org/grpc"
@@ -31,7 +34,7 @@ type GPRCServerService struct {
 	address string
 }
 
-func ProvideService(cfg *setting.Cfg, apiKey apikey.Service, userService user.Service) (Provider, error) {
+func ProvideService(cfg *setting.Cfg, apiKey apikey.Service, userService user.Service, tracer tracing.Tracer) (Provider, error) {
 	s := &GPRCServerService{
 		cfg:    cfg,
 		logger: log.New("grpc-server"),
@@ -42,10 +45,21 @@ func ProvideService(cfg *setting.Cfg, apiKey apikey.Service, userService user.Se
 	// Default auth is admin token check, but this can be overridden by
 	// services which implement ServiceAuthFuncOverride interface.
 	// See https://github.com/grpc-ecosystem/go-grpc-middleware/blob/master/auth/auth.go#L30.
-	authenticator := newAuthenticator(apiKey, userService)
+	authenticator := interceptors.NewAuthenticator(apiKey, userService)
+
 	opts = append(opts, []grpc.ServerOption{
-		grpc.StreamInterceptor(grpcAuth.StreamServerInterceptor(authenticator.authenticate)),
-		grpc.UnaryInterceptor(grpcAuth.UnaryServerInterceptor(authenticator.authenticate)),
+		grpc.UnaryInterceptor(
+			grpc_middleware.ChainUnaryServer(
+				grpcAuth.UnaryServerInterceptor(authenticator.Authenticate),
+				interceptors.TracingUnaryInterceptor(tracer),
+			),
+		),
+		grpc.StreamInterceptor(
+			grpc_middleware.ChainStreamServer(
+				interceptors.TracingStreamInterceptor(tracer),
+				grpcAuth.StreamServerInterceptor(authenticator.Authenticate),
+			),
+		),
 	}...)
 
 	if s.cfg.GRPCServerTLSConfig != nil {