控制器规范：Openkruise 生命周期钩子之 inPlaceUpdate Hook (二)

2023-10-18 04:06:32

在上一篇文章中，我们已经介绍了 inPlaceUpdate Lifecycle Hook 的基本概念和实现原理。在本文中，我们将进一步探讨如何将 inPlaceUpdate Lifecycle Hook 的运行逻辑固化到自制的 Controller 中。

自制 Controller 的工作流程

自制 Controller 的工作流程一般分为以下几个步骤：

创建 Controller 对象。
注册监听器，以便当 Kubernetes 集群中的资源发生变化时能够收到通知。
当收到资源变化的通知时，Controller 会获取资源的当前状态和期望状态，并根据这两个状态来确定是否需要执行任何操作。
如果需要执行操作，Controller 会向 Kubernetes 集群发送命令，以执行这些操作。

如何实现 inPlaceUpdate Lifecycle Hook

要将 inPlaceUpdate Lifecycle Hook 的运行逻辑固化到自制的 Controller 中，我们需要在 Controller 中实现以下功能：

监听 Deployment 资源的变化。
当 Deployment 资源发生变化时，获取 Deployment 的当前状态和期望状态，并检查是否满足 inPlaceUpdate Lifecycle Hook 的触发条件。
如果满足 inPlaceUpdate Lifecycle Hook 的触发条件，则执行 inPlaceUpdate Lifecycle Hook 的操作。

以下是一个示例代码，展示了如何实现 inPlaceUpdate Lifecycle Hook：

// InPlaceUpdateHookController 是一个自制 Controller，用于实现 inPlaceUpdate Lifecycle Hook。
type InPlaceUpdateHookController struct {
    // clientset 是一个 Kubernetes 客户端，用于与 Kubernetes 集群交互。
    clientset *kubernetes.Clientset
}

// NewInPlaceUpdateHookController 创建一个新的 InPlaceUpdateHookController 对象。
func NewInPlaceUpdateHookController(clientset *kubernetes.Clientset) *InPlaceUpdateHookController {
    return &InPlaceUpdateHookController{
        clientset: clientset,
    }
}

// Run 启动 InPlaceUpdateHookController。
func (c *InPlaceUpdateHookController) Run(stopCh <-chan struct{}) error {
    // 创建一个 Deployment 资源监听器。
    deploymentInformerFactory := informers.NewSharedInformerFactoryWithOptions(c.clientset, 0, informers.WithTweakListOptions(func(options *metav1.ListOptions) {
        options.FieldSelector = "metadata.name=my-deployment"
    }))
    deploymentInformer := deploymentInformerFactory.Apps().V1().Deployments().Informer()

    // 当 Deployment 资源发生变化时，执行 inPlaceUpdate Lifecycle Hook 的操作。
    deploymentInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
        UpdateFunc: func(oldObj, newObj interface{}) {
            oldDeployment := oldObj.(*appsv1.Deployment)
            newDeployment := newObj.(*appsv1.Deployment)

            // 检查是否满足 inPlaceUpdate Lifecycle Hook 的触发条件。
            if oldDeployment.Spec.Template.Labels["app"] != newDeployment.Spec.Template.Labels["app"] {
                // 执行 inPlaceUpdate Lifecycle Hook 的操作。
                c.inplaceUpdateDeployment(newDeployment)
            }
        },
    })

    // 启动 Deployment 资源监听器。
    deploymentInformerFactory.Start(stopCh)

    // 阻塞直到收到 stopCh 信号。
    <-stopCh

    return nil
}

// inplaceUpdateDeployment 执行 inPlaceUpdate Lifecycle Hook 的操作。
func (c *InPlaceUpdateHookController) inplaceUpdateDeployment(deployment *appsv1.Deployment) error {
    // ...

    return nil
}