refact: extend registerControllerUpstreamMessageService to reg hdrs
Extend registerControllerUpstreamMessageService to register upstream message handlers
This commit is contained in:
1 parent
9f01e1b3c1
commit
f9b45ca5b7
1 file changed
+10
@@ -16,6 +16,10 @@ import (
|
||||
|
||||
var controllerGrpcServiceRegistrationFunctions []func(context.Context, *Controller, *grpc.Server) error
|
||||
|
||||
// a stack of funcs; each func will use handlers.RegisterUpstreamMessageHandler
|
||||
// to register a handlers.UpstreamMessageHandler
|
||||
var controllerRegisterUpstreamMessageHandlerFunctions []func(context.Context, *Controller) error
|
||||
|
||||
func init() {
|
||||
controllerGrpcServiceRegistrationFunctions = append(controllerGrpcServiceRegistrationFunctions,
|
||||
registerControllerServerCoordinationService,
|
||||
@@ -115,5 +119,11 @@ func registerControllerUpstreamMessageService(ctx context.Context, c *Controller
|
||||
return fmt.Errorf("%s: error creating upstream message service handler: %w", op, err)
|
||||
}
|
||||
pbs.RegisterUpstreamMessageServiceServer(server, upstreamMsgService)
|
||||
|
||||
for _, registerHandlerFn := range controllerRegisterUpstreamMessageHandlerFunctions {
|
||||
if err := registerHandlerFn(ctx, c); err != nil {
|
||||
return fmt.Errorf("%s: error registering upstream message handler: %w", op, err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in new issue
Block a user