diff --git a/pkg/rescheduler/pod/pods.go b/pkg/rescheduler/pod/pods.go index 9c76d1f87..a28fe1510 100644 --- a/pkg/rescheduler/pod/pods.go +++ b/pkg/rescheduler/pod/pods.go @@ -22,10 +22,7 @@ import ( "k8s.io/kubernetes/pkg/api/v1" "k8s.io/kubernetes/pkg/api/v1/helper/qos" "k8s.io/kubernetes/pkg/client/clientset_generated/clientset" -) - -const ( - criticalPodAnnotation = "scheduler.alpha.kubernetes.io/critical-pod" + "k8s.io/kubernetes/pkg/kubelet/types" ) func ListPodsOnANode(client clientset.Interface, node *v1.Node) ([]*v1.Pod, error) { @@ -44,8 +41,7 @@ func ListPodsOnANode(client clientset.Interface, node *v1.Node) ([]*v1.Pod, erro } func IsCriticalPod(pod *v1.Pod) bool { - _, found := pod.ObjectMeta.Annotations[criticalPodAnnotation] - return found + return types.IsCriticalPod(pod * v1.Pod) } func IsBestEffortPod(pod *v1.Pod) bool { @@ -59,3 +55,12 @@ func IsBurstablePod(pod *v1.Pod) bool { func IsGuaranteedPod(pod *v1.Pod) bool { return qos.GetPodQOS(pod) == v1.PodQOSGuaranteed } + +func IsDaemonsetPod(pod *v1.Pod) bool { +} + +// IsMirrorPod checks whether the pod is a mirror pod. +func IsMirrorPod(pod *apiv1.Pod) bool { + _, found := pod.ObjectMeta.Annotations[types.ConfigMirrorAnnotationKey] + return found +}