Skip to content

Commit 5e9516d

Browse files
authored
Merge pull request #386 from thockin/master
Clean up global flRoot use (v4)
2 parents 53e0c0d + 295cfd4 commit 5e9516d

File tree

1 file changed

+13
-14
lines changed

1 file changed

+13
-14
lines changed

cmd/git-sync/main.go

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,7 @@ const (
183183

184184
type customLogger struct {
185185
logr.Logger
186+
root string
186187
errorFile string
187188
}
188189

@@ -230,16 +231,16 @@ func (l *customLogger) exportError(content string) {
230231

231232
// writeContent writes the error content to the error file.
232233
func (l *customLogger) writeContent(content []byte) {
233-
if _, err := os.Stat(*flRoot); os.IsNotExist(err) {
234+
if _, err := os.Stat(l.root); os.IsNotExist(err) {
234235
fileMode := os.FileMode(0755)
235-
if err := os.Mkdir(*flRoot, fileMode); err != nil {
236-
l.Logger.Error(err, "can't create the root directory", "root", *flRoot)
236+
if err := os.Mkdir(l.root, fileMode); err != nil {
237+
l.Logger.Error(err, "can't create the root directory", "root", l.root)
237238
return
238239
}
239240
}
240-
tmpFile, err := ioutil.TempFile(*flRoot, "tmp-err-")
241+
tmpFile, err := ioutil.TempFile(l.root, "tmp-err-")
241242
if err != nil {
242-
l.Logger.Error(err, "can't create temporary error-file", "directory", *flRoot, "prefix", "tmp-err-")
243+
l.Logger.Error(err, "can't create temporary error-file", "directory", l.root, "prefix", "tmp-err-")
243244
return
244245
}
245246
defer func() {
@@ -253,8 +254,9 @@ func (l *customLogger) writeContent(content []byte) {
253254
return
254255
}
255256

256-
if err := os.Rename(tmpFile.Name(), l.errorFile); err != nil {
257-
l.Logger.Error(err, "can't rename to error-file", "temp-file", tmpFile.Name(), "error-file", l.errorFile)
257+
errorFile := filepath.Join(l.root, l.errorFile)
258+
if err := os.Rename(tmpFile.Name(), errorFile); err != nil {
259+
l.Logger.Error(err, "can't rename to error-file", "temp-file", tmpFile.Name(), "error-file", errorFile)
258260
return
259261
}
260262
}
@@ -264,11 +266,12 @@ func (l *customLogger) deleteErrorFile() {
264266
if l.errorFile == "" {
265267
return
266268
}
267-
if err := os.Remove(l.errorFile); err != nil {
269+
errorFile := filepath.Join(l.root, l.errorFile)
270+
if err := os.Remove(errorFile); err != nil {
268271
if os.IsNotExist(err) {
269272
return
270273
}
271-
l.Logger.Error(err, "can't delete the error-file", "filename", l.errorFile)
274+
l.Logger.Error(err, "can't delete the error-file", "filename", errorFile)
272275
}
273276
}
274277

@@ -385,11 +388,7 @@ func main() {
385388
flag.CommandLine.Parse(nil) // Otherwise glog complains
386389
setGlogFlags()
387390

388-
var errorFile string
389-
if *flErrorFile != "" {
390-
errorFile = filepath.Join(*flRoot, *flErrorFile)
391-
}
392-
log = &customLogger{glogr.New(), errorFile}
391+
log = &customLogger{glogr.New(), *flRoot, *flErrorFile}
393392

394393
if *flVersion {
395394
fmt.Println(version.VERSION)

0 commit comments

Comments
 (0)