From 81e60b1e501d746e0dd2980bb88cc6118a5bd318 Mon Sep 17 00:00:00 2001
From: Avesh Agarwal <avagarwa@redhat.com>
Date: Tue, 1 Aug 2017 18:31:52 -0400
Subject: [PATCH] Implement pod with local storage check.

---
 pkg/rescheduler/pod/pods.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/pkg/rescheduler/pod/pods.go b/pkg/rescheduler/pod/pods.go
index a28fe1510..45b7cc7e8 100644
--- a/pkg/rescheduler/pod/pods.go
+++ b/pkg/rescheduler/pod/pods.go
@@ -41,7 +41,7 @@ func ListPodsOnANode(client clientset.Interface, node *v1.Node) ([]*v1.Pod, erro
 }
 
 func IsCriticalPod(pod *v1.Pod) bool {
-	return types.IsCriticalPod(pod * v1.Pod)
+	return types.IsCriticalPod(pod)
 }
 
 func IsBestEffortPod(pod *v1.Pod) bool {
@@ -57,10 +57,21 @@ func IsGuaranteedPod(pod *v1.Pod) bool {
 }
 
 func IsDaemonsetPod(pod *v1.Pod) bool {
+	return false
 }
 
 // IsMirrorPod checks whether the pod is a mirror pod.
-func IsMirrorPod(pod *apiv1.Pod) bool {
+func IsMirrorPod(pod *v1.Pod) bool {
 	_, found := pod.ObjectMeta.Annotations[types.ConfigMirrorAnnotationKey]
 	return found
 }
+
+func IsPodWithLocalStorage(pod *v1.Pod) bool {
+	for _, volume := range pod.Spec.Volumes {
+		if volume.EmptyDir != nil {
+			return true
+		}
+	}
+
+	return false
+}