diff --git a/debug/debug.go b/debug/debug.go index eb6fb01d4..aab6d9853 100644 --- a/debug/debug.go +++ b/debug/debug.go @@ -14,101 +14,109 @@ import ( ) var opts struct { + logger *log.Logger tags map[string]bool breaks map[string]bool m sync.Mutex } -var debugLogger = initDebugLogger() - -func initDebugLogger() (lgr *log.Logger) { - opts.tags = make(map[string]bool) - opts.breaks = make(map[string]bool) +func init() { + initDebugLogger() + initDebugTags() + initDebugBreaks() fmt.Fprintf(os.Stderr, "debug enabled\n") +} +func initDebugLogger() { debugfile := os.Getenv("DEBUG_LOG") - if debugfile != "" { - fmt.Fprintf(os.Stderr, "debug log file %v\n", debugfile) - - // open logfile - f, err := os.OpenFile(debugfile, os.O_WRONLY|os.O_APPEND, 0600) - - if err == nil { - // seek to the end - _, err = f.Seek(2, 0) - if err != nil { - fmt.Fprintf(os.Stderr, "unable to seek to the end of %v: %v\n", debugfile, err) - os.Exit(3) - } - } - - if err != nil && os.IsNotExist(err) { - // create logfile - f, err = os.OpenFile(debugfile, os.O_WRONLY|os.O_CREATE, 0600) - } - - if err != nil { - fmt.Fprintf(os.Stderr, "unable to open debug log file: %v\n", err) - os.Exit(2) - } - - // open logger - lgr = log.New(f, "", log.LstdFlags) + if debugfile == "" { + return } + fmt.Fprintf(os.Stderr, "debug log file %v\n", debugfile) + + f, err := os.OpenFile(debugfile, os.O_WRONLY|os.O_APPEND, 0600) + + if err == nil { + _, err = f.Seek(2, 0) + if err != nil { + fmt.Fprintf(os.Stderr, "unable to seek to the end of %v: %v\n", debugfile, err) + os.Exit(3) + } + } + + if err != nil && os.IsNotExist(err) { + f, err = os.OpenFile(debugfile, os.O_WRONLY|os.O_CREATE, 0600) + } + + if err != nil { + fmt.Fprintf(os.Stderr, "unable to open debug log file: %v\n", err) + os.Exit(2) + } + + opts.logger = log.New(f, "", log.LstdFlags) +} + +func initDebugTags() { + opts.tags = make(map[string]bool) + // defaults opts.tags["break"] = true // initialize tags env := os.Getenv("DEBUG_TAGS") - if len(env) > 0 { - tags := []string{} - - for _, tag := range strings.Split(env, ",") { - t := strings.TrimSpace(tag) - val := true - if t[0] == '-' { - val = false - t = t[1:] - } else if t[0] == '+' { - val = true - t = t[1:] - } - - // test pattern - _, err := path.Match(t, "") - if err != nil { - fmt.Fprintf(os.Stderr, "error: invalid pattern %q: %v\n", t, err) - os.Exit(5) - } - - opts.tags[t] = val - tags = append(tags, tag) - } - - fmt.Fprintf(os.Stderr, "debug log enabled for: %v\n", tags) + if len(env) == 0 { + return } - // initialize break tags - env = os.Getenv("DEBUG_BREAK") - if len(env) > 0 { - breaks := []string{} + tags := []string{} - for _, tag := range strings.Split(env, ",") { - t := strings.TrimSpace(tag) - opts.breaks[t] = true - breaks = append(breaks, t) + for _, tag := range strings.Split(env, ",") { + t := strings.TrimSpace(tag) + val := true + if t[0] == '-' { + val = false + t = t[1:] + } else if t[0] == '+' { + val = true + t = t[1:] } - fmt.Fprintf(os.Stderr, "debug breaks enabled for: %v\n", breaks) + // test pattern + _, err := path.Match(t, "") + if err != nil { + fmt.Fprintf(os.Stderr, "error: invalid pattern %q: %v\n", t, err) + os.Exit(5) + } + + opts.tags[t] = val + tags = append(tags, tag) } - return + fmt.Fprintf(os.Stderr, "debug log enabled for: %v\n", tags) +} + +func initDebugBreaks() { + opts.breaks = make(map[string]bool) + + env := os.Getenv("DEBUG_BREAK") + if len(env) == 0 { + return + } + + breaks := []string{} + + for _, tag := range strings.Split(env, ",") { + t := strings.TrimSpace(tag) + opts.breaks[t] = true + breaks = append(breaks, t) + } + + fmt.Fprintf(os.Stderr, "debug breaks enabled for: %v\n", breaks) } func Log(tag string, f string, args ...interface{}) { - // synchronize log writes opts.m.Lock() defer opts.m.Unlock() @@ -120,8 +128,8 @@ func Log(tag string, f string, args ...interface{}) { fmt.Fprintf(os.Stderr, "DEBUG["+tag+"]: "+f, args...) } - if debugLogger != nil { - debugLogger.Printf("["+tag+"] "+f, args...) + if opts.logger != nil { + opts.logger.Printf("["+tag+"] "+f, args...) } // check if tag is enabled directly