mirror of
https://github.com/kubernetes-sigs/descheduler.git
synced 2026-01-26 21:31:18 +01:00
6e953b2ff3
* Remove log level from Errors Every error printed via Errors is expected to be important and always printable. * Invoke first Deschedule and then Balance extension points (breaking change) * Separate plugin arg conversion from pluginsMap * Seperate profile population from plugin execution * Convert strategy params into profiles outside the main descheduling loop Strategy params are static and do not change in time. * Bump the internal DeschedulerPolicy to v1alpha2 Drop conversion from v1alpha1 to internal * add tests to v1alpha1 to internal conversion * add tests to strategyParamsToPluginArgs params wiring * in v1alpha1 evictableNamespaces are still Namespaces * add test passing in all params Co-authored-by: Lucas Severo Alves <lseveroa@redhat.com>
95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
"sigs.k8s.io/descheduler/pkg/api"
|
|
)
|
|
|
|
const SystemCriticalPriority = 2 * int32(1000000000)
|
|
|
|
// GetNamespacesFromPodAffinityTerm returns a set of names
|
|
// according to the namespaces indicated in podAffinityTerm.
|
|
// If namespaces is empty it considers the given pod's namespace.
|
|
func GetNamespacesFromPodAffinityTerm(pod *v1.Pod, podAffinityTerm *v1.PodAffinityTerm) sets.String {
|
|
names := sets.String{}
|
|
if len(podAffinityTerm.Namespaces) == 0 {
|
|
names.Insert(pod.Namespace)
|
|
} else {
|
|
names.Insert(podAffinityTerm.Namespaces...)
|
|
}
|
|
return names
|
|
}
|
|
|
|
// PodMatchesTermsNamespaceAndSelector returns true if the given <pod>
|
|
// matches the namespace and selector defined by <affinityPod>`s <term>.
|
|
func PodMatchesTermsNamespaceAndSelector(pod *v1.Pod, namespaces sets.String, selector labels.Selector) bool {
|
|
if !namespaces.Has(pod.Namespace) {
|
|
return false
|
|
}
|
|
|
|
if !selector.Matches(labels.Set(pod.Labels)) {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// GetPriorityFromPriorityClass gets priority from the given priority class.
|
|
// If no priority class is provided, it will return SystemCriticalPriority by default.
|
|
func GetPriorityFromPriorityClass(ctx context.Context, client clientset.Interface, name string) (int32, error) {
|
|
if name != "" {
|
|
priorityClass, err := client.SchedulingV1().PriorityClasses().Get(ctx, name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return priorityClass.Value, nil
|
|
}
|
|
return SystemCriticalPriority, nil
|
|
}
|
|
|
|
// GetPriorityFromStrategyParams gets priority from the given StrategyParameters.
|
|
// It will return SystemCriticalPriority by default.
|
|
func GetPriorityFromStrategyParams(ctx context.Context, client clientset.Interface, priorityThreshold *api.PriorityThreshold) (priority int32, err error) {
|
|
if priorityThreshold == nil {
|
|
return SystemCriticalPriority, nil
|
|
}
|
|
if priorityThreshold.Value != nil {
|
|
priority = *priorityThreshold.Value
|
|
} else {
|
|
priority, err = GetPriorityFromPriorityClass(ctx, client, priorityThreshold.Name)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
if priority > SystemCriticalPriority {
|
|
return 0, fmt.Errorf("Priority threshold can't be greater than %d", SystemCriticalPriority)
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetPriorityValueFromPriorityThreshold gets priority from the given PriorityThreshold.
|
|
// It will return SystemCriticalPriority by default.
|
|
func GetPriorityValueFromPriorityThreshold(ctx context.Context, client clientset.Interface, priorityThreshold *api.PriorityThreshold) (priority int32, err error) {
|
|
if priorityThreshold == nil {
|
|
return SystemCriticalPriority, nil
|
|
}
|
|
if priorityThreshold.Value != nil {
|
|
priority = *priorityThreshold.Value
|
|
} else {
|
|
priority, err = GetPriorityFromPriorityClass(ctx, client, priorityThreshold.Name)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("unable to get priority value from the priority class: %v", err)
|
|
}
|
|
}
|
|
if priority > SystemCriticalPriority {
|
|
return 0, fmt.Errorf("priority threshold can't be greater than %d", SystemCriticalPriority)
|
|
}
|
|
return
|
|
}
|