From 208d65ab596b80644437578cede135a30f1a2599 Mon Sep 17 00:00:00 2001
From: He Simei <hesimei@zju.edu.cn>
Date: Thu, 12 Nov 2015 10:07:03 +0800
Subject: [PATCH] kube-down gracefully

---
 cluster/ubuntu/util.sh | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/cluster/ubuntu/util.sh b/cluster/ubuntu/util.sh
index 0c09b27effa26..58944715fd2a1 100755
--- a/cluster/ubuntu/util.sh
+++ b/cluster/ubuntu/util.sh
@@ -515,6 +515,20 @@ function provision-masterandnode() {
   }    
 }
 
+# check whether kubelet has torn down all of the pods
+function check-pods-torn-down() {
+  local kubectl="${KUBE_ROOT}/cluster/kubectl.sh"
+  local attempt=0
+  while [[ ! -z "$(kubectl get pods | tail -n +2)" ]]; do
+    if (( attempt > 120 )); then
+      echo "timeout waiting for tearing down pods" >> ~/kube/err.log
+    fi
+    echo "waiting for tearing down pods"
+    attempt=$((attempt+1))
+    sleep 5
+  done
+}
+
 # Delete a kubernetes cluster
 function kube-down() {
   
@@ -523,7 +537,8 @@ function kube-down() {
   source "${KUBE_ROOT}/cluster/common.sh"
 
   tear_down_alive_resources
-
+  check-pods-torn-down 
+  
   local ii=0
   for i in ${nodes}; do
       if [[ "${roles[${ii}]}" == "ai" || "${roles[${ii}]}" == "a" ]]; then
@@ -543,12 +558,18 @@ function kube-down() {
             rm -rf /srv/kubernetes
             '
         " || echo "Cleaning on master ${i#*@} failed"
+
+        if [[ "${roles[${ii}]}" == "ai" ]]; then
+          ssh $SSH_OPTS -t "$i" "sudo rm -rf /var/lib/kubelet"
+        fi
+        
       elif [[ "${roles[${ii}]}" == "i" ]]; then
         echo "Cleaning on node ${i#*@}"
         ssh $SSH_OPTS -t "$i" "
           pgrep flanneld && \
           sudo -p '[sudo] password to stop node: ' -- /bin/bash -c '
-            service flanneld stop            
+            service flanneld stop
+            rm -rf /var/lib/kubelet            
             '
           " || echo "Cleaning on node ${i#*@} failed"
       else
@@ -567,10 +588,8 @@ function kube-down() {
           /etc/default/flanneld
         
         rm -rf ~/kube
-        rm -rf /var/lib/kubelet 
         rm -f /run/flannel/subnet.env
       '" || echo "cleaning legacy files on ${i#*@} failed"
-
     ((ii=ii+1))
   done
 }