83 lines
1.5 KiB
Go
83 lines
1.5 KiB
Go
package log
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
type priority int
|
|
|
|
const (
|
|
errorFormat = "[%s] %s:%d %s\n"
|
|
logFormat = "[%s] %s\n"
|
|
|
|
fatal priority = iota
|
|
error
|
|
info
|
|
debug
|
|
)
|
|
|
|
// A common interface to access the Fatal method of
|
|
// both testing.B and testing.T.
|
|
type Fataler interface {
|
|
Fatal(args ...interface{})
|
|
}
|
|
|
|
func (p priority) String() string {
|
|
switch p {
|
|
case fatal:
|
|
return "fatal"
|
|
case error:
|
|
return "error"
|
|
case info:
|
|
return "info"
|
|
case debug:
|
|
return "debug"
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
// Debug function, if the debug flag is set, then display. Do nothing otherwise
|
|
// If Docker is in damon mode, also send the debug info on the socket
|
|
func Debugf(format string, a ...interface{}) {
|
|
if os.Getenv("DEBUG") != "" {
|
|
logf(os.Stderr, debug, format, a...)
|
|
}
|
|
}
|
|
|
|
func Infof(format string, a ...interface{}) {
|
|
logf(os.Stdout, info, format, a...)
|
|
}
|
|
|
|
func Errorf(format string, a ...interface{}) {
|
|
logf(os.Stderr, error, format, a...)
|
|
}
|
|
|
|
func Fatalf(format string, a ...interface{}) {
|
|
logf(os.Stderr, fatal, format, a...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func logf(stream io.Writer, level priority, format string, a ...interface{}) {
|
|
var prefix string
|
|
|
|
if level <= error || level == debug {
|
|
// Retrieve the stack infos
|
|
_, file, line, ok := runtime.Caller(2)
|
|
if !ok {
|
|
file = "<unknown>"
|
|
line = -1
|
|
} else {
|
|
file = file[strings.LastIndex(file, "/")+1:]
|
|
}
|
|
prefix = fmt.Sprintf(errorFormat, level.String(), file, line, format)
|
|
} else {
|
|
prefix = fmt.Sprintf(logFormat, level.String(), format)
|
|
}
|
|
|
|
fmt.Fprintf(stream, prefix, a...)
|
|
}
|