[Enhancement] support priorityClassName (#642)

config/crd/bases/starrocks.com_starrocksclusters.yaml

@@ -2634,6 +2634,12 @@ spec:
                       type: string
                     description: the pod labels for user select or classify pods.
                     type: object
+                  priorityClassName:
+                    description: |-
+                      If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+                      cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+                      More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+                    type: string
                   readOnlyRootFilesystem:
                     description: "Whether this container has a read-only root filesystem.\nDefault
                       is false.\nNote that:\n\t1. This field cannot be set when spec.os.name
@@ -7637,6 +7643,12 @@ spec:
                       type: string
                     description: the pod labels for user select or classify pods.
                     type: object
+                  priorityClassName:
+                    description: |-
+                      If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+                      cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+                      More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+                    type: string
                   readOnlyRootFilesystem:
                     description: "Whether this container has a read-only root filesystem.\nDefault
                       is false.\nNote that:\n\t1. This field cannot be set when spec.os.name
@@ -10534,6 +10546,12 @@ spec:
                       type: string
                     description: the pod labels for user select or classify pods.
                     type: object
+                  priorityClassName:
+                    description: |-
+                      If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+                      cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+                      More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+                    type: string
                   readinessProbeFailureSeconds:
                     description: |-
                       ReadinessProbeFailureSeconds defines the total failure seconds of readiness Probe.
@@ -13526,6 +13544,12 @@ spec:
                       type: string
                     description: the pod labels for user select or classify pods.
                     type: object
+                  priorityClassName:
+                    description: |-
+                      If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+                      cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+                      More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+                    type: string
                   readOnlyRootFilesystem:
                     description: "Whether this container has a read-only root filesystem.\nDefault
                       is false.\nNote that:\n\t1. This field cannot be set when spec.os.name

config/crd/bases/starrocks.com_starrockswarehouses.yaml

@@ -3230,6 +3230,12 @@ spec:
                       type: string
                     description: the pod labels for user select or classify pods.
                     type: object
+                  priorityClassName:
+                    description: |-
+                      If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+                      cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+                      More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+                    type: string
                   readOnlyRootFilesystem:
                     description: "Whether this container has a read-only root filesystem.\nDefault
                       is false.\nNote that:\n\t1. This field cannot be set when spec.os.name

deploy/starrocks.com_starrocksclusters.yaml

@@ -1289,6 +1289,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readOnlyRootFilesystem:
                     type: boolean
                   readinessProbeFailureSeconds:
@@ -3662,6 +3664,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readOnlyRootFilesystem:
                     type: boolean
                   readinessProbeFailureSeconds:
@@ -4994,6 +4998,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readinessProbeFailureSeconds:
                     format: int32
                     type: integer
@@ -6387,6 +6393,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readOnlyRootFilesystem:
                     type: boolean
                   readinessProbeFailureSeconds:

deploy/starrocks.com_starrockswarehouses.yaml

@@ -1603,6 +1603,8 @@ spec:
                     additionalProperties:
                       type: string
                     type: object
+                  priorityClassName:
+                    type: string
                   readOnlyRootFilesystem:
                     type: boolean
                   readinessProbeFailureSeconds:

helm-charts/charts/kube-starrocks/charts/starrocks/templates/starrockscluster.yaml

@@ -31,6 +31,9 @@ spec:
     {{- end }}
     replicas: {{ .Values.starrocksFESpec.replicas }}
     imagePullPolicy: {{ .Values.starrocksFESpec.imagePullPolicy }}
+    {{- if .Values.starrocksFESpec.priorityClassName }}
+    priorityClassName: {{ .Values.starrocksFESpec.priorityClassName }}
+    {{- end }}
     {{- if .Values.starrocksFESpec.maxUnavailablePods }}
     updateStrategy:
       rollingUpdate:
@@ -296,6 +299,9 @@ spec:
     {{- end }}
     replicas: {{ .Values.starrocksBeSpec.replicas }}
     imagePullPolicy: {{ .Values.starrocksBeSpec.imagePullPolicy }}
+    {{- if .Values.starrocksBeSpec.priorityClassName }}
+    priorityClassName: {{ .Values.starrocksBeSpec.priorityClassName }}
+    {{- end }}
     {{- if .Values.starrocksBeSpec.maxUnavailablePods }}
     updateStrategy:
       rollingUpdate:
@@ -593,6 +599,9 @@ spec:
     replicas: {{ .Values.starrocksCnSpec.replicas }}
     {{- end }}
     imagePullPolicy: {{ .Values.starrocksCnSpec.imagePullPolicy }}
+    {{- if .Values.starrocksCnSpec.priorityClassName }}
+    priorityClassName: {{ .Values.starrocksCnSpec.priorityClassName }}
+    {{- end }}
     {{- if .Values.starrocksCnSpec.maxUnavailablePods }}
     updateStrategy:
       rollingUpdate:
@@ -888,6 +897,9 @@ spec:
     {{- end }}
     replicas: {{ .Values.starrocksFeProxySpec.replicas }}
     imagePullPolicy: {{ .Values.starrocksFeProxySpec.imagePullPolicy }}
+    {{- if .Values.starrocksFeProxySpec.priorityClassName }}
+    priorityClassName: {{ .Values.starrocksFeProxySpec.priorityClassName }}
+    {{- end }}
     resolver: {{ .Values.starrocksFeProxySpec.resolver }}
     {{- if .Values.starrocksFeProxySpec.resources }}
       {{- toYaml .Values.starrocksFeProxySpec.resources | nindent 4 }}

helm-charts/charts/kube-starrocks/charts/starrocks/values.yaml

@@ -467,6 +467,10 @@ starrocksFESpec:
   # When this is set containers will be able to view and signal processes from other containers
   # in the same pod, and the first process in each container will not be assigned PID 1.
   shareProcessNamespace:
+  # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+  # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+  # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+  priorityClassName:
 
 # spec for compute node, compute node provide compute function.
 starrocksCnSpec:
@@ -817,6 +821,10 @@ starrocksCnSpec:
   # When this is set containers will be able to view and signal processes from other containers
   # in the same pod, and the first process in each container will not be assigned PID 1.
   shareProcessNamespace:
+  # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+  # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+  # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+  priorityClassName:
 
 # spec for component be, provide storage and compute function.
 starrocksBeSpec:
@@ -1128,6 +1136,10 @@ starrocksBeSpec:
   # When this is set containers will be able to view and signal processes from other containers
   # in the same pod, and the first process in each container will not be assigned PID 1.
   shareProcessNamespace:
+  # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+  # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+  # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+  priorityClassName:
 
 # create secrets if necessary.
 secrets: []
@@ -1247,3 +1259,7 @@ starrocksFeProxySpec:
       # e.g. mount an emptyDir volume to /tmp
       # - name: tmp-data
       #   mountPath: /tmp
+  # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+  # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+  # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+  priorityClassName:

helm-charts/charts/kube-starrocks/values.yaml

@@ -584,6 +584,10 @@ starrocks:
     # When this is set containers will be able to view and signal processes from other containers
     # in the same pod, and the first process in each container will not be assigned PID 1.
     shareProcessNamespace:
+    # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+    # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+    # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+    priorityClassName:
 
   # spec for compute node, compute node provide compute function.
   starrocksCnSpec:
@@ -934,6 +938,10 @@ starrocks:
     # When this is set containers will be able to view and signal processes from other containers
     # in the same pod, and the first process in each container will not be assigned PID 1.
     shareProcessNamespace:
+    # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+    # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+    # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+    priorityClassName:
 
   # spec for component be, provide storage and compute function.
   starrocksBeSpec:
@@ -1245,6 +1253,10 @@ starrocks:
     # When this is set containers will be able to view and signal processes from other containers
     # in the same pod, and the first process in each container will not be assigned PID 1.
     shareProcessNamespace:
+    # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+    # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+    # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+    priorityClassName:
 
   # create secrets if necessary.
   secrets: []
@@ -1364,3 +1376,7 @@ starrocks:
         # e.g. mount an emptyDir volume to /tmp
         # - name: tmp-data
         #   mountPath: /tmp
+    # priorityClassName allows you to assign a [Kubernetes PriorityClass]
+    # to the pods of this component. This determines the scheduling priority of pods and their eviction order under resource pressure.
+    # see https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption/#priorityclass for more details.
+    priorityClassName:

pkg/apis/starrocks/v1/component_type.go

@@ -307,3 +307,7 @@ const (
 	DRPhaseDoing DRPhase = "doing"
 	DRPhaseDone  DRPhase = "done"
 )
+
+func (spec *StarRocksComponentSpec) GetPriorityClassName() string {
+	return spec.PriorityClassName
+}

pkg/apis/starrocks/v1/load_type.go

@@ -28,6 +28,7 @@ type loadInterface interface {
 	GetImagePullPolicy() corev1.PullPolicy
 	GetCommand() []string
 	GetArgs() []string
+	GetPriorityClassName() string
 }
 
 type StarRocksLoadSpec struct {
@@ -169,6 +170,12 @@ type StarRocksLoadSpec struct {
 	// Optional: Default to false.
 	// +optional
 	ShareProcessNamespace *bool `json:"shareProcessNamespace,omitempty"`
+
+	//If specified, Pods can have priority. Priority indicates the importance of a Pod relative to other Pods. If a Pod
+	//cannot be scheduled, the scheduler tries to preempt (evict) lower priority Pods to make scheduling
+	//More info: https://kubernetes.io/docs/concepts/scheduling-eviction/pod-priority-preemption of the pending Pod possible.
+	// +optional
+	PriorityClassName string `json:"priorityClassName,omitempty"`
 }
 
 // StarRocksService defines external service for starrocks component.
@@ -333,3 +340,7 @@ func (spec *StarRocksLoadSpec) GetImagePullPolicy() corev1.PullPolicy {
 	}
 	return spec.ImagePullPolicy
 }
+
+func (spec *StarRocksLoadSpec) GetPriorityClassName() string {
+	return spec.PriorityClassName
+}

pkg/apis/starrocks/v1/starrockscluster_types.go

@@ -258,6 +258,10 @@ func (spec *StarRocksFeProxySpec) IsReadOnlyRootFilesystem() *bool {
 // that StarRocksFeProxySpec needs to implement SpecInterface interface
 func (spec *StarRocksFeProxySpec) GetSysctls() []corev1.Sysctl { return nil }
 
+func (spec *StarRocksFeProxySpec) GetPriorityClassName() string {
+	return spec.StarRocksLoadSpec.GetPriorityClassName()
+}
+
 // Phase is defined under status, e.g.
 // 1. StarRocksClusterStatus.Phase represents the phase of starrocks cluster.
 // 2. StarRocksWarehouseStatus.Phase represents the phase of starrocks warehouse.

pkg/k8sutils/templates/pod/spec.go

@@ -279,6 +279,7 @@ func Spec(spec v1.SpecInterface, container corev1.Container, volumes []corev1.Vo
 		SchedulerName:                 spec.GetSchedulerName(),
 		AutomountServiceAccountToken:  func() *bool { b := false; return &b }(),
 		ShareProcessNamespace:         spec.GetShareProcessNamespace(),
+		PriorityClassName:             spec.GetPriorityClassName(),
 	}
 	return podSpec
 }