Add multiple update selectors

api/v1/ytsaurus_types.go

@@ -20,6 +20,8 @@ import (
 	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/ytsaurus/ytsaurus-k8s-operator/pkg/consts"
 )
 
 // EmbeddedPersistentVolumeClaim is an embedded version of k8s.io/api/core/v1.PersistentVolumeClaim.
@@ -645,11 +647,14 @@ type YtsaurusSpec struct {
 	//+optional
 	EnableFullUpdate bool `json:"enableFullUpdate"`
 	//+optional
-	//+kubebuilder:validation:Enum={"","Nothing","MasterOnly","DataNodesOnly","TabletNodesOnly","ExecNodesOnly","StatelessOnly","Everything"}
-	// UpdateSelector is an experimental field. Behaviour may change.
-	// If UpdateSelector is not empty EnableFullUpdate is ignored.
+	//+optional
+	// Deprecated: UpdateSelector is an experimental field. Behaviour may change.
 	UpdateSelector UpdateSelector `json:"updateSelector"`
 
+	//+optional
+	// Controls the components that should be updated during the update process.
+	UpdateSelectors []ComponentUpdateSelector `json:"updateSelectors,omitempty"`
+
 	NodeSelector map[string]string   `json:"nodeSelector,omitempty"`
 	Tolerations  []corev1.Toleration `json:"tolerations,omitempty"`
 
@@ -727,26 +732,16 @@ type TabletCellBundleInfo struct {
 
 type UpdateSelector string
 
-const (
-	// UpdateSelectorUnspecified means that selector is disabled and would be ignored completely.
-	UpdateSelectorUnspecified UpdateSelector = ""
-	// UpdateSelectorNothing means that no component could be updated.
-	UpdateSelectorNothing UpdateSelector = "Nothing"
-	// UpdateSelectorMasterOnly means that only master could be updated.
-	UpdateSelectorMasterOnly UpdateSelector = "MasterOnly"
-	// UpdateSelectorTabletNodesOnly means that only data nodes could be updated
-	UpdateSelectorDataNodesOnly UpdateSelector = "DataNodesOnly"
-	// UpdateSelectorTabletNodesOnly means that only tablet nodes could be updated
-	UpdateSelectorTabletNodesOnly UpdateSelector = "TabletNodesOnly"
-	// UpdateSelectorExecNodesOnly means that only tablet nodes could be updated
-	UpdateSelectorExecNodesOnly UpdateSelector = "ExecNodesOnly"
-	// UpdateSelectorStatelessOnly means that only stateless components (everything but master, data nodes, and tablet nodes)
-	// could be updated.
-	UpdateSelectorStatelessOnly UpdateSelector = "StatelessOnly"
-	// UpdateSelectorEverything means that all components could be updated.
-	// With this setting and if master or tablet nodes need update all the components would be updated.
-	UpdateSelectorEverything UpdateSelector = "Everything"
-)
+type ComponentUpdateSelector struct {
+	//+optional
+	ComponentType consts.ComponentType `json:"componentType,omitempty"`
+	//+optional
+	ComponentGroup consts.ComponentGroup `json:"componentGroup,omitempty"`
+	//+optional
+	ComponentName string `json:"componentName,omitempty"`
+
+	// TODO(#325): Add rolling options
+}
 
 type UpdateFlow string
 

config/crd/bases/cluster.ytsaurus.tech_ytsaurus.yaml

@@ -34657,18 +34657,22 @@ spec:
               uiImage:
                 type: string
               updateSelector:
-                description: UpdateSelector is an experimental field. Behaviour may
-                  change.
-                enum:
-                - ""
-                - Nothing
-                - MasterOnly
-                - DataNodesOnly
-                - TabletNodesOnly
-                - ExecNodesOnly
-                - StatelessOnly
-                - Everything
+                description: 'Deprecated: UpdateSelector is an experimental field.
+                  Behaviour may change.'
                 type: string
+              updateSelectors:
+                description: Controls the components that should be updated during
+                  the update process.
+                items:
+                  properties:
+                    componentGroup:
+                      type: string
+                    componentName:
+                      type: string
+                    componentType:
+                      type: string
+                  type: object
+                type: array
               useIpv4:
                 default: false
                 type: boolean

controllers/sync.go

@@ -433,61 +433,81 @@
 	componentNames []string
 }
 
-func canUpdateComponent(selector ytv1.UpdateSelector, component consts.ComponentType) bool {
-	switch selector {
-	case ytv1.UpdateSelectorNothing:
-		return false
-	case ytv1.UpdateSelectorMasterOnly:
-		return component == consts.MasterType
-	case ytv1.UpdateSelectorDataNodesOnly:
-		return component == consts.DataNodeType
-	case ytv1.UpdateSelectorTabletNodesOnly:
-		return component == consts.TabletNodeType
-	case ytv1.UpdateSelectorExecNodesOnly:
-		return component == consts.ExecNodeType
-	case ytv1.UpdateSelectorStatelessOnly:
-		switch component {
-		case consts.MasterType:
-			return false
-		case consts.DataNodeType:
-			return false
-		case consts.TabletNodeType:
-			return false
-		}
-		return true
-	case ytv1.UpdateSelectorEverything:
-		return true
-	default:
-		return false
+func getFlowFromComponent(component consts.ComponentType) ytv1.UpdateFlow {
+	if component == consts.MasterType {
+		return ytv1.UpdateFlowMaster
 	}
+	if component == consts.TabletNodeType {
+		return ytv1.UpdateFlowTabletNodes
+	}
+	if component == consts.DataNodeType || component == consts.ExecNodeType {
+		return ytv1.UpdateFlowFull
+	}
+	return ytv1.UpdateFlowStateless
+}
+
+func canUpdateComponent(selectors []ytv1.ComponentUpdateSelector, component components.Component) (bool, error) {
+	for _, selector := range selectors {
+		if selector.ComponentType != "" {
+			if selector.ComponentType == component.GetType() {
+				return true, nil
+			}
+		} else if selector.ComponentName != "" {
+			if selector.ComponentName == component.GetName() {
+				return true, nil
+			}
+		} else {
+			switch selector.ComponentGroup {
+			case consts.ComponentGroupEverything:
+				return true, nil
+			case consts.ComponentGroupNothing:
+				return false, nil
+			case consts.ComponentGroupStateful:
+				if component.GetType() == consts.DataNodeType || component.GetType() == consts.TabletNodeType {
+					return true, nil
+				}
+			case consts.ComponentGroupStateless:
+				if component.GetType() != consts.DataNodeType && component.GetType() != consts.TabletNodeType && component.GetType() != consts.MasterType {
+					return true, nil
+				}
+			default:
+				return false, fmt.Errorf("unexpected component group %s", selector.ComponentGroup)
+			}
+		}
+	}
+	return false, nil
 }
 
 // chooseUpdateFlow considers spec and decides if operator should proceed with update or block.
 // Block case is indicated with non-empty blockMsg.
 // If update is not blocked, updateMeta containing a chosen flow and the component names to update returned.
 func chooseUpdateFlow(spec ytv1.YtsaurusSpec, needUpdate []components.Component) (meta updateMeta, blockMsg string) {
-	configuredSelector := spec.UpdateSelector
-	if configuredSelector == ytv1.UpdateSelectorUnspecified {
-		if spec.EnableFullUpdate {
-			configuredSelector = ytv1.UpdateSelectorEverything
-		} else {
-			configuredSelector = ytv1.UpdateSelectorStatelessOnly
-		}
+	configuredSelectors := spec.UpdateSelectors
+	if len(configuredSelectors) == 0 && spec.EnableFullUpdate {
+		configuredSelectors = []ytv1.ComponentUpdateSelector{{ComponentGroup: consts.ComponentGroupEverything}}
 	}
+	needFullUpdate := false
 
 	var canUpdate []string
 	var cannotUpdate []string
-	needFullUpdate := false
+	flows := make(map[ytv1.UpdateFlow]struct{})
 
 	for _, comp := range needUpdate {
 		componentType := comp.GetType()
 		componentName := comp.GetName()
-		if canUpdateComponent(configuredSelector, componentType) {
+		upd, err := canUpdateComponent(configuredSelectors, comp)
+		if err != nil {
+			return updateMeta{}, err.Error()
+		}
+		if upd {
 			canUpdate = append(canUpdate, componentName)
+			flows[getFlowFromComponent(componentType)] = struct{}{}
 		} else {
 			cannotUpdate = append(cannotUpdate, componentName)
 		}
-		if !canUpdateComponent(ytv1.UpdateSelectorStatelessOnly, componentType) && componentType != consts.DataNodeType {
+
+		statelessCheck, err := canUpdateComponent([]ytv1.ComponentUpdateSelector{{ComponentGroup: consts.ComponentGroupStateless}}, comp)
+		if !statelessCheck && componentType != consts.DataNodeType {
 			needFullUpdate = true
 		}
 	}
@@ -499,37 +519,25 @@
 		return updateMeta{}, "All components are uptodate"
 	}
 
-	switch configuredSelector {
-	case ytv1.UpdateSelectorEverything:
+	if len(configuredSelectors) == 1 && configuredSelectors[0].ComponentGroup == consts.ComponentGroupEverything {
 		if needFullUpdate {
-			return updateMeta{
-				flow:           ytv1.UpdateFlowFull,
-				componentNames: nil,
-			}, ""
+			return updateMeta{flow: ytv1.UpdateFlowFull, componentNames: nil}, ""
 		} else {
-			return updateMeta{
-				flow:           ytv1.UpdateFlowStateless,
-				componentNames: canUpdate,
-			}, ""
-		}
-	case ytv1.UpdateSelectorMasterOnly:
-		return updateMeta{
-			flow:           ytv1.UpdateFlowMaster,
-			componentNames: canUpdate,
-		}, ""
-	case ytv1.UpdateSelectorTabletNodesOnly:
-		return updateMeta{
-			flow:           ytv1.UpdateFlowTabletNodes,
-			componentNames: canUpdate,
-		}, ""
-	case ytv1.UpdateSelectorDataNodesOnly, ytv1.UpdateSelectorExecNodesOnly, ytv1.UpdateSelectorStatelessOnly:
-		return updateMeta{
-			flow:           ytv1.UpdateFlowStateless,
-			componentNames: canUpdate,
-		}, ""
-	default:
-		return updateMeta{}, fmt.Sprintf("Unexpected update selector %s", configuredSelector)
+			return updateMeta{flow: ytv1.UpdateFlowStateless, componentNames: canUpdate}, ""
+		}
+	}
+
+	if len(flows) == 0 {
+		return updateMeta{}, fmt.Sprintf("Unexpected state: no flows for components {%s}", strings.Join(canUpdate, ", "))
+	}
+
+	if len(flows) == 1 {
+		for flow := range flows {
+			return updateMeta{flow: flow, componentNames: canUpdate}, ""
+		}
 	}
+	// If more than one flow is possible, we choose to follow full update flow.
+	return updateMeta{flow: ytv1.UpdateFlowFull, componentNames: canUpdate}, ""
 }
 
 func (r *YtsaurusReconciler) Sync(ctx context.Context, resource *ytv1.Ytsaurus) (ctrl.Result, error) {

docs/api.md

@@ -327,6 +327,23 @@ _Appears in:_
 | `imagePullSecrets` _[LocalObjectReference](https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.28/#localobjectreference-v1-core) array_ |  |  |  |
 
 
+#### ComponentUpdateSelector
+
+
+
+
+
+
+
+_Appears in:_
+- [YtsaurusSpec](#ytsaurusspec)
+
+| Field | Description | Default | Validation |
+| --- | --- | --- | --- |
+| `componentKind` _[ComponentType](#componenttype)_ |  |  |  |
+| `componentGroup` _[ComponentGroup](#componentgroup)_ |  |  |  |
+
+
 #### ControllerAgentsSpec
 
 
@@ -2048,16 +2065,6 @@ _Underlying type:_ _string_
 _Appears in:_
 - [YtsaurusSpec](#ytsaurusspec)
 
-| Field | Description |
-| --- | --- |
-| `` | UpdateSelectorUnspecified means that selector is disabled and would be ignored completely.<br /> |
-| `Nothing` | UpdateSelectorNothing means that no component could be updated.<br /> |
-| `MasterOnly` | UpdateSelectorMasterOnly means that only master could be updated.<br /> |
-| `DataNodesOnly` | UpdateSelectorTabletNodesOnly means that only data nodes could be updated<br /> |
-| `TabletNodesOnly` | UpdateSelectorTabletNodesOnly means that only tablet nodes could be updated<br /> |
-| `ExecNodesOnly` | UpdateSelectorExecNodesOnly means that only tablet nodes could be updated<br /> |
-| `StatelessOnly` | UpdateSelectorStatelessOnly means that only stateless components (everything but master, data nodes, and tablet nodes)<br />could be updated.<br /> |
-| `Everything` | UpdateSelectorEverything means that all components could be updated.<br />With this setting and if master or tablet nodes need update all the components would be updated.<br /> |
 
 
 #### UpdateState
@@ -2206,7 +2213,8 @@ _Appears in:_
 | `oauthService` _[OauthServiceSpec](#oauthservicespec)_ |  |  |  |
 | `isManaged` _boolean_ |  | true |  |
 | `enableFullUpdate` _boolean_ |  | true |  |
-| `updateSelector` _[UpdateSelector](#updateselector)_ | UpdateSelector is an experimental field. Behaviour may change.<br />If UpdateSelector is not empty EnableFullUpdate is ignored. |  | Enum: [ Nothing MasterOnly DataNodesOnly TabletNodesOnly ExecNodesOnly StatelessOnly Everything] <br /> |
+| `updateSelector` _[UpdateSelector](#updateselector)_ | Deprecated: UpdateSelector is an experimental field. Behaviour may change. |  |  |
+| `updateSelectors` _[ComponentUpdateSelector](#componentupdateselector) array_ | Controls the components that should be updated during the update process. |  |  |
 | `nodeSelector` _object (keys:string, values:string)_ |  |  |  |
 | `tolerations` _[Toleration](https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.28/#toleration-v1-core) array_ |  |  |  |
 | `bootstrap` _[BootstrapSpec](#bootstrapspec)_ |  |  |  |

pkg/consts/types.go

@@ -23,3 +23,12 @@ const (
 	ChytType                 ComponentType = "CHYT"
 	SpytType                 ComponentType = "SPYT"
 )
+
+type ComponentGroup string
+
+const (
+	ComponentGroupStateless  ComponentGroup = "Stateless"
+	ComponentGroupStateful   ComponentGroup = "Stateful"
+	ComponentGroupEverything ComponentGroup = "Everything"
+	ComponentGroupNothing    ComponentGroup = "Nothing"
+)