server: register DistSQL, Tracing service with DRPC server

Enable `DistSQL` and `Tracing` services on the DRPC server in addition
to gRPC. This is controlled by `rpc.experimental_drpc.enabled` (off by
default).

This change is part of a series and is similar to: #146926

Note: This only registers the service; the client is not updated to use
the DRPC client, so this service will not have any functional effect.

Epic: CRDB-48925
Release note: None

Author: cthumuluru-crdb

pkg/server/BUILD.bazel

@@ -375,6 +375,7 @@ go_library(
         "@com_github_prometheus_client_model//go",
         "@com_github_prometheus_common//expfmt",
         "@in_gopkg_yaml_v2//:yaml_v2",
+        "@io_storj_drpc//:drpc",
         "@io_storj_drpc//drpcerr",
         "@io_storj_drpc//drpcmigrate",
         "@org_golang_google_grpc//:grpc",

pkg/server/server.go

@@ -1152,6 +1152,7 @@ func NewServer(cfg Config, stopper *stop.Stopper) (serverctl.ServerStartupInterf
 			nodeLiveness:             optionalnodeliveness.MakeContainer(nodeLiveness),
 			gossip:                   gossip.MakeOptionalGossip(g),
 			grpcServer:               grpcServer.Server,
+			drpcMux:                  drpcServer.DRPCServer,
 			nodeIDContainer:          idContainer,
 			externalStorage:          externalStorage,
 			externalStorageFromURI:   externalStorageFromURI,

pkg/server/server_sql.go

@@ -140,6 +140,7 @@ import (
 	"github.com/marusama/semaphore"
 	"github.com/nightlyone/lockfile"
 	"google.golang.org/grpc"
+	"storj.io/drpc"
 )
 
 // SQLServer encapsulates the part of a CRDB server that is dedicated to SQL
@@ -232,6 +233,7 @@ type sqlServerOptionalKVArgs struct {
 	gossip gossip.OptionalGossip
 	// To register blob and DistSQL servers.
 	grpcServer *grpc.Server
+	drpcMux    drpc.Mux
 	// For the temporaryObjectCleaner.
 	isMeta1Leaseholder func(context.Context, hlc.ClockTimestamp) (bool, error)
 	// DistSQL, lease management, and others want to know the node they're on.
@@ -555,6 +557,9 @@ func newSQLServer(ctx context.Context, cfg sqlServerArgs) (*SQLServer, error) {
 	// Create trace service for inter-node sharing of inflight trace spans.
 	tracingService := service.New(cfg.Tracer)
 	tracingservicepb.RegisterTracingServer(cfg.grpcServer, tracingService)
+	if err := tracingservicepb.DRPCRegisterTracing(cfg.drpcMux, tracingService); err != nil {
+		return nil, err
+	}
 
 	// If the node id is already populated, we only need to create a placeholder
 	// instance provider without initializing the instance, since this is not a
@@ -873,6 +878,9 @@ func newSQLServer(ctx context.Context, cfg sqlServerArgs) (*SQLServer, error) {
 
 	distSQLServer := distsql.NewServer(ctx, distSQLCfg, cfg.remoteFlowRunner)
 	execinfrapb.RegisterDistSQLServer(cfg.grpcServer, distSQLServer)
+	if err := execinfrapb.DRPCRegisterDistSQL(cfg.drpcMux, distSQLServer.AsDRPCServer()); err != nil {
+		return nil, err
+	}
 
 	// Set up Executor
 

pkg/server/tenant.go

@@ -1335,6 +1335,7 @@ func makeTenantSQLServerArgs(
 			nodeLiveness:      optionalnodeliveness.MakeContainer(nil),
 			gossip:            gossip.MakeOptionalGossip(nil),
 			grpcServer:        grpcServer.Server,
+			drpcMux:           drpcServer.DRPCServer,
 			isMeta1Leaseholder: func(_ context.Context, _ hlc.ClockTimestamp) (bool, error) {
 				return false, errors.New("isMeta1Leaseholder is not available to secondary tenants")
 			},