From 965de9098f4cd0d71ce1fec255248849a8efaf93 Mon Sep 17 00:00:00 2001
From: Wojciech Tyczynski <wojtekt@google.com>
Date: Thu, 3 Mar 2016 11:35:00 +0100
Subject: [PATCH] Fix update codecgen

---
 hack/update-codecgen.sh | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/hack/update-codecgen.sh b/hack/update-codecgen.sh
index 37ecc41476f..a43c4d77a15 100755
--- a/hack/update-codecgen.sh
+++ b/hack/update-codecgen.sh
@@ -32,12 +32,21 @@ generated_files=($(
         -o -wholename './target' \
         -o -wholename '*/third_party/*' \
         -o -wholename '*/Godeps/*' \
+        -o -wholename '*/codecgen-*-1234.generated.go' \
       \) -prune \
     \) -name '*.generated.go'))
 
-# Build codecgen binary from Godeps.
+# Register function to be called on EXIT to remove codecgen
+# binary and also to touch the files that should be regenerated
+# since they are first removed.
+# This is necessary to make the script work after previous failure.
 function cleanup {
   rm -f "${CODECGEN:-}"
+  pushd "${KUBE_ROOT}" > /dev/null
+  for (( i=0; i < number; i++ )); do
+    touch "${generated_files[${i}]}" || true
+  done
+  popd > /dev/null
 }
 trap cleanup EXIT