diff options
Diffstat (limited to 'internal/praefect/server.go')
| -rw-r--r-- | internal/praefect/server.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/internal/praefect/server.go b/internal/praefect/server.go index 08f5aac4f..84a9ebfb7 100644 --- a/internal/praefect/server.go +++ b/internal/praefect/server.go @@ -42,6 +42,7 @@ func NewGRPCServer( nodeMgr nodes.Manager, txMgr *transactions.Manager, queue datastore.ReplicationEventQueue, + gs datastore.GenerationStore, grpcOpts ...grpc.ServerOption, ) *grpc.Server { ctxTagOpts := []grpc_ctxtags.Option{ @@ -85,7 +86,7 @@ func NewGRPCServer( warnDupeAddrs(logger, conf) srv := grpc.NewServer(grpcOpts...) - registerServices(srv, nodeMgr, txMgr, conf, queue) + registerServices(srv, nodeMgr, txMgr, conf, queue, gs) return srv } @@ -97,10 +98,10 @@ func proxyRequiredOpts(director proxy.StreamDirector) []grpc.ServerOption { } // registerServices registers services praefect needs to handle RPCs on its own. -func registerServices(srv *grpc.Server, nm nodes.Manager, tm *transactions.Manager, conf config.Config, queue datastore.ReplicationEventQueue) { +func registerServices(srv *grpc.Server, nm nodes.Manager, tm *transactions.Manager, conf config.Config, queue datastore.ReplicationEventQueue, gs datastore.GenerationStore) { // ServerServiceServer is necessary for the ServerInfo RPC gitalypb.RegisterServerServiceServer(srv, server.NewServer(conf, nm)) - gitalypb.RegisterPraefectInfoServiceServer(srv, info.NewServer(nm, conf, queue)) + gitalypb.RegisterPraefectInfoServiceServer(srv, info.NewServer(nm, conf, queue, gs)) gitalypb.RegisterRefTransactionServer(srv, transaction.NewServer(tm)) healthpb.RegisterHealthServer(srv, health.NewServer()) |