server: register Login and Logout services with DRPC server

Enable these service 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: shubhamdhama

pkg/server/authserver/BUILD.bazel

@@ -44,6 +44,7 @@ go_library(
         "@com_github_cockroachdb_logtags//:logtags",
         "@com_github_cockroachdb_redact//:redact",
         "@com_github_grpc_ecosystem_grpc_gateway//runtime:go_default_library",
+        "@io_storj_drpc//:drpc",
         "@org_golang_google_grpc//:grpc",
         "@org_golang_google_grpc//codes",
         "@org_golang_google_grpc//metadata",

pkg/server/authserver/api.go

@@ -18,12 +18,15 @@ import (
 	"github.com/cockroachdb/cockroach/pkg/sql/pgwire"
 	"github.com/grpc-ecosystem/grpc-gateway/runtime"
 	"google.golang.org/grpc"
+	"storj.io/drpc"
 )
 
 type Server interface {
 	RegisterService(*grpc.Server)
 	RegisterGateway(ctx context.Context, mux *runtime.ServeMux, conn *grpc.ClientConn) error
 
+	RegisterDRPCService(drpc.Mux) error
+
 	// UserLogin verifies an incoming request by a user to create an web
 	// authentication session. It checks the provided credentials against
 	// the system.users table, and if successful creates a new

pkg/server/authserver/authentication.go

@@ -47,6 +47,7 @@ import (
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/status"
+	"storj.io/drpc"
 )
 
 const (
@@ -132,6 +133,15 @@ type authenticationServer struct {
 func (s *authenticationServer) RegisterService(g *grpc.Server) {
 	serverpb.RegisterLogInServer(g, s)
 	serverpb.RegisterLogOutServer(g, s)
+
+}
+
+// RegisterService registers the LogIn and LogOut services with DRPC.
+func (s *authenticationServer) RegisterDRPCService(d drpc.Mux) error {
+	if err := serverpb.DRPCRegisterLogIn(d, s); err != nil {
+		return err
+	}
+	return serverpb.DRPCRegisterLogOut(d, s)
 }
 
 // RegisterGateway starts the gateway (i.e. reverse proxy) that proxies HTTP requests

pkg/server/server.go

@@ -1253,6 +1253,10 @@ func NewServer(cfg Config, stopper *stop.Stopper) (serverctl.ServerStartupInterf
 		gw.RegisterService(grpcServer.Server)
 	}
 
+	if err := sAuth.RegisterDRPCService(drpcServer); err != nil {
+		return nil, err
+	}
+
 	// Tell the node event logger (join, restart) how to populate SQL entries
 	// into system.eventlog.
 	node.InitLogger(sqlServer.execCfg)

pkg/server/tenant.go

@@ -480,6 +480,10 @@ func newTenantServer(
 		gw.RegisterService(args.grpc.Server)
 	}
 
+	if err := sAuth.RegisterDRPCService(args.drpc); err != nil {
+		return nil, err
+	}
+
 	// Tell the status/admin servers how to access SQL structures.
 	sStatus.setStmtDiagnosticsRequester(sqlServer.execCfg.StmtDiagnosticsRecorder)
 	serverIterator.sqlServer = sqlServer