Merge pull request #387 from thockin/release-3.x

Clean up global flRoot use
kubernetes · May 22, 2021 · 61ded13 · 61ded13
2 parents 246584d + e9e682e
commit 61ded13
Showing 1 changed file with 13 additions and 14 deletions.
diff --git a/cmd/git-sync/main.go b/cmd/git-sync/main.go
@@ -164,6 +164,7 @@ const (
 
 type customLogger struct {
 	logr.Logger
+	root      string
 	errorFile string
 }
 
@@ -211,16 +212,16 @@ func (l *customLogger) exportError(content string) {
 
 // writeContent writes the error content to the error file.
 func (l *customLogger) writeContent(content []byte) {
-	if _, err := os.Stat(*flRoot); os.IsNotExist(err) {
+	if _, err := os.Stat(l.root); os.IsNotExist(err) {
 		fileMode := os.FileMode(0755)
-		if err := os.Mkdir(*flRoot, fileMode); err != nil {
-			l.Logger.Error(err, "can't create the root directory", "root", *flRoot)
+		if err := os.Mkdir(l.root, fileMode); err != nil {
+			l.Logger.Error(err, "can't create the root directory", "root", l.root)
 			return
 		}
 	}
-	tmpFile, err := ioutil.TempFile(*flRoot, "tmp-err-")
+	tmpFile, err := ioutil.TempFile(l.root, "tmp-err-")
 	if err != nil {
-		l.Logger.Error(err, "can't create temporary error-file", "directory", *flRoot, "prefix", "tmp-err-")
+		l.Logger.Error(err, "can't create temporary error-file", "directory", l.root, "prefix", "tmp-err-")
 		return
 	}
 	defer func() {
@@ -234,8 +235,9 @@ func (l *customLogger) writeContent(content []byte) {
 		return
 	}
 
-	if err := os.Rename(tmpFile.Name(), l.errorFile); err != nil {
-		l.Logger.Error(err, "can't rename to error-file", "temp-file", tmpFile.Name(), "error-file", l.errorFile)
+	errorFile := filepath.Join(l.root, l.errorFile)
+	if err := os.Rename(tmpFile.Name(), errorFile); err != nil {
+		l.Logger.Error(err, "can't rename to error-file", "temp-file", tmpFile.Name(), "error-file", errorFile)
 		return
 	}
 }
@@ -245,11 +247,12 @@ func (l *customLogger) deleteErrorFile() {
 	if l.errorFile == "" {
 		return
 	}
-	if err := os.Remove(l.errorFile); err != nil {
+	errorFile := filepath.Join(l.root, l.errorFile)
+	if err := os.Remove(errorFile); err != nil {
 		if os.IsNotExist(err) {
 			return
 		}
-		l.Logger.Error(err, "can't delete the error-file", "filename", l.errorFile)
+		l.Logger.Error(err, "can't delete the error-file", "filename", errorFile)
 	}
 }
 
@@ -338,11 +341,7 @@ func main() {
 	setFlagDefaults()
 	flag.Parse()
 
-	var errorFile string
-	if *flErrorFile != "" {
-		errorFile = filepath.Join(*flRoot, *flErrorFile)
-	}
-	log = &customLogger{glogr.New(), errorFile}
+	log = &customLogger{glogr.New(), *flRoot, *flErrorFile}
 
 	if *flVer {
 		fmt.Println(version.VERSION)