From 43f6d901819dc254d900610ec30ccaddc6382f86 Mon Sep 17 00:00:00 2001 From: yaocw2020 Date: Tue, 25 May 2021 16:42:13 +0800 Subject: [PATCH] Delete temporary files generated by controller-gen --- pkg/controller-gen/main.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/pkg/controller-gen/main.go b/pkg/controller-gen/main.go index f74f509c..3264bf1a 100644 --- a/pkg/controller-gen/main.go +++ b/pkg/controller-gen/main.go @@ -142,14 +142,18 @@ func copyGoPathToModules(customArgs *cgargs.CustomArgs) error { continue } - return filepath.Walk(pkg, func(path string, info os.FileInfo, err error) error { + if err := filepath.Walk(pkg, func(path string, info os.FileInfo, err error) error { newPath := strings.Replace(path, pkg, ".", 1) if info.IsDir() { return os.MkdirAll(newPath, info.Mode()) } return copyFile(path, newPath) - }) + }); err != nil { + return err + } + + return deleteFile(pkg) } return nil @@ -180,6 +184,10 @@ func copyFile(src, dst string) error { return os.Chmod(dst, srcinfo.Mode()) } +func deleteFile(path string) error { + return os.RemoveAll(path) +} + func generateDeepcopy(groups map[string]bool, customArgs *cgargs.CustomArgs) error { if len(groups) == 0 { return nil