From 062b8698c73120d4469c74b9894c962d9a5ea552 Mon Sep 17 00:00:00 2001
From: Avesh Agarwal <avagarwa@redhat.com>
Date: Tue, 1 Aug 2017 11:21:43 -0400
Subject: [PATCH] Implement mirror pod check.

---
 pkg/rescheduler/pod/pods.go | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

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
+}