fix: Add missing AfterControlPlaneUpgradeLifecycleHandler interface (#…

…113)
nutanix-cloud-native · Aug 22, 2023 · 7297882 · 7297882
1 parent ba830d7
commit 7297882
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/internal/runtimehooks/webhooks/server.go b/internal/runtimehooks/webhooks/server.go
@@ -116,6 +116,17 @@ func (s *Server) Start(ctx context.Context) error {
 			}
 		}
 
+		if t, ok := h.(handlers.AfterControlPlaneUpgradeLifecycleHandler); ok {
+			if err := webhookServer.AddExtensionHandler(server.ExtensionHandler{
+				Hook:        runtimehooksv1.AfterControlPlaneUpgrade,
+				Name:        h.Name(),
+				HandlerFunc: t.AfterControlPlaneUpgrade,
+			}); err != nil {
+				setupLog.Error(err, "error adding handler")
+				return err
+			}
+		}
+
 		if t, ok := h.(handlers.BeforeClusterDeleteLifecycleHandler); ok {
 			if err := webhookServer.AddExtensionHandler(server.ExtensionHandler{
 				Hook:        runtimehooksv1.BeforeClusterDelete,

diff --git a/pkg/handlers/interfaces.go b/pkg/handlers/interfaces.go
@@ -32,6 +32,13 @@ type BeforeClusterUpgradeLifecycleHandler interface {
 		*runtimehooksv1.BeforeClusterUpgradeResponse,
 	)
 }
+type AfterControlPlaneUpgradeLifecycleHandler interface {
+	AfterControlPlaneUpgrade(
+		context.Context,
+		*runtimehooksv1.AfterControlPlaneUpgradeRequest,
+		*runtimehooksv1.AfterControlPlaneUpgradeResponse,
+	)
+}
 type BeforeClusterDeleteLifecycleHandler interface {
 	BeforeClusterDelete(
 		context.Context,