add template patcher

Author: ColdsteelRail

xset/api/xset_controller_types.go

@@ -35,6 +35,7 @@ type XSetController interface {
 
 	GetXSetSpec(object XSetObject) *XSetSpec
 	GetXSetPatch(object metav1.Object) ([]byte, error)
+	GetXSetTemplatePatcher(object metav1.Object) func(client.Object) error
 	UpdateScaleStrategy(object XSetObject, scaleStrategy *ScaleStrategy) (err error)
 	GetXSetStatus(object XSetObject) *XSetStatus
 	SetXSetStatus(object XSetObject, status *XSetStatus)

xset/synccontrols/sync_control.go

@@ -473,7 +473,9 @@ func (r *RealSyncControl) Scale(ctx context.Context, xsetObject api.XSetObject,
 							r.xsetLabelMgr.Set(object.GetLabels(), api.EnumXSetTargetCompletingLabel, strconv.FormatInt(time.Now().UnixNano(), 10))
 						}
 						return nil
-					})
+					},
+					r.xsetController.GetXSetTemplatePatcher(xsetObject),
+				)
 				if err != nil {
 					return fmt.Errorf("fail to new Target from revision %s: %w", revision.GetName(), err)
 				}

xset/synccontrols/x_replace.go

@@ -131,7 +131,8 @@ func (r *RealSyncControl) replaceOriginTargets(
 		replaceRevision := r.getReplaceRevision(originTarget, syncContext)
 
 		// create target using update revision if replaced by update, otherwise using current revision
-		newTarget, err := NewTargetFrom(r.xsetController, r.xsetLabelMgr, instance, replaceRevision, originTargetId)
+		newTarget, err := NewTargetFrom(r.xsetController, r.xsetLabelMgr, instance, replaceRevision, originTargetId,
+			r.xsetController.GetXSetTemplatePatcher(instance))
 		if err != nil {
 			return err
 		}

xset/synccontrols/x_utils.go

@@ -17,6 +17,7 @@
 package synccontrols
 
 import (
+	"context"
 	"fmt"
 	"strconv"
 	"time"
@@ -26,6 +27,8 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"sigs.k8s.io/controller-runtime/pkg/client"
 
+	clientutils "kusionstack.io/kube-utils/client"
+	controllerutils "kusionstack.io/kube-utils/controller/utils"
 	"kusionstack.io/kube-utils/xset/api"
 )
 
@@ -180,3 +183,14 @@ func IsControlledByXSet(xsetLabelManager api.XSetLabelManager, obj client.Object
 	v, ok := xsetLabelManager.Get(obj.GetLabels(), api.EnumXSetControlledLabel)
 	return ok && v == "true"
 }
+
+func ApplyTemplatePatcher(ctx context.Context, xsetController api.XSetController, c client.Client, xset api.XSetObject, targets []targetWrapper) error {
+	_, patchErr := controllerutils.SlowStartBatch(len(targets), controllerutils.SlowStartInitialBatchSize, false, func(i int, _ error) error {
+		if targets[i].Object == nil || targets[i].PlaceHolder {
+			return nil
+		}
+		_, err := clientutils.UpdateOnConflict(ctx, c, c, targets[i].Object, xsetController.GetXSetTemplatePatcher(xset))
+		return err
+	})
+	return patchErr
+}

xset/xset_controller.go

@@ -242,8 +242,9 @@ func (r *xSetCommonReconciler) doSync(ctx context.Context, instance api.XSetObje
 
 	_, scaleRequeueAfter, scaleErr := r.syncControl.Scale(ctx, instance, syncContext)
 	_, updateRequeueAfter, updateErr := r.syncControl.Update(ctx, instance, syncContext)
+	patcherErr := synccontrols.ApplyTemplatePatcher(ctx, r.XSetController, r.Client, instance, syncContext.TargetWrappers)
 
-	err = errors.Join(scaleErr, updateErr)
+	err = errors.Join(scaleErr, updateErr, patcherErr)
 	if updateRequeueAfter != nil && (scaleRequeueAfter == nil || *updateRequeueAfter < *scaleRequeueAfter) {
 		return updateRequeueAfter, err
 	}