Refactor debug.go
This commit is contained in:
Florian Weingarten
parent
a906b9febe
commit
ca77655971
1 changed files with 79 additions and 71 deletions
150
debug/debug.go
150
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
|
||||
|
|
|
|||
Loading…
Reference in a new issue