cleanup created bundle when creating failed

Signed-off-by: Ma Shimiao <[email protected]>
opencontainers · Aug 22, 2017 · 5e6f503 · 5e6f503
1 parent 50ad08a
commit 5e6f503
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/image/image.go b/image/image.go
@@ -301,6 +301,13 @@ func createBundle(w walker, m *manifest, dest, rootfs string) error {
 			if err2 := os.MkdirAll(dest, 0755); err2 != nil {
 				return err2
 			}
+			defer func() {
+				if err != nil {
+					if err3 := os.RemoveAll(dest); err3 != nil {
+						fmt.Printf("Failed to clean up %q: %s\n", dest, err3.Error())
+					}
+				}
+			}()
 		} else {
 			return err
 		}