config(wiring): pass Manager/Scheduler auth.jwt.key into gRPC server via auth registry

Signed-off-by: sabarixr <[email protected]>

Author: sabarixr

manager/rpcserver/rpcserver.go

@@ -26,6 +26,7 @@ import (
 	"d7y.io/dragonfly/v2/manager/database"
 	"d7y.io/dragonfly/v2/manager/models"
 	"d7y.io/dragonfly/v2/manager/searcher"
+	"d7y.io/dragonfly/v2/pkg/rpc/auth"
 	managerserver "d7y.io/dragonfly/v2/pkg/rpc/manager/server"
 )
 
@@ -59,6 +60,8 @@ func New(
 		searcher: searcher,
 	}
 
+	// Provide JWT key from config to manager server via auth package.
+	auth.SetServerKey("manager", cfg.Auth.JWT.Key)
 	return s, managerserver.New(
 		newManagerServerV1(s.config, database, s.cache, s.searcher),
 		newManagerServerV2(s.config, database, s.cache, s.searcher),

scheduler/rpcserver/rpcserver.go

@@ -19,6 +19,7 @@ package rpcserver
 import (
 	"google.golang.org/grpc"
 
+	"d7y.io/dragonfly/v2/pkg/rpc/auth"
 	"d7y.io/dragonfly/v2/pkg/rpc/scheduler/server"
 	"d7y.io/dragonfly/v2/scheduler/config"
 	"d7y.io/dragonfly/v2/scheduler/job"
@@ -39,6 +40,8 @@ func New(
 	dynconfig config.DynconfigInterface,
 	opts ...grpc.ServerOption,
 ) *grpc.Server {
+	// Provide JWT key from config to scheduler server via auth package.
+	auth.SetServerKey("scheduler", cfg.Auth.JWT.Key)
 	return server.New(
 		newSchedulerServerV1(cfg, resource, scheduling, dynconfig),
 		newSchedulerServerV2(cfg, resource, persistentResource, persistentCacheResource, scheduling, job, dynconfig),