From f9b45ca5b7a7308afc963108805dd1dd40149374 Mon Sep 17 00:00:00 2001
From: Jim <jlambert@hashicorp.com>
Date: Mon, 3 Apr 2023 13:55:53 -0400
Subject: [PATCH] refact: extend registerControllerUpstreamMessageService to
 reg hdrs

Extend registerControllerUpstreamMessageService to register upstream
message handlers
---
 internal/daemon/controller/rpc_registration.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/internal/daemon/controller/rpc_registration.go b/internal/daemon/controller/rpc_registration.go
index bcfb43312..e42686673 100644
--- a/internal/daemon/controller/rpc_registration.go
+++ b/internal/daemon/controller/rpc_registration.go
@@ -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
 }