forked from Shiloh/githaven
596 lines
14 KiB
Go
596 lines
14 KiB
Go
package log
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// These flags define which text to prefix to each log entry generated by the Logger.
|
|
const (
|
|
// Bits or'ed together to control what's printed. There is no control over the
|
|
// order they appear (the order listed here) or the format they present (as
|
|
// described in the comments). A colon appears after these items:
|
|
// 2009/0123 01:23:23.123123 /a/b/c/d.go:23: message
|
|
Ldate = 1 << iota // the date: 2009/0123
|
|
Ltime // the time: 01:23:23
|
|
Lmicroseconds // microsecond resolution: 01:23:23.123123. assumes Ltime.
|
|
Llongfile // full file name and line number: /a/b/c/d.go:23
|
|
Lshortfile // final file name element and line number: d.go:23. overrides Llongfile
|
|
Lmodule // module name
|
|
Llevel // level: 0(Debug), 1(Info), 2(Warn), 3(Error), 4(Panic), 5(Fatal)
|
|
Llongcolor // color will start [info] end of line
|
|
Lshortcolor // color only include [info]
|
|
LstdFlags = Ldate | Ltime // initial values for the standard logger
|
|
//Ldefault = Llevel | LstdFlags | Lshortfile | Llongcolor
|
|
) // [prefix][time][level][module][shortfile|longfile]
|
|
|
|
func Ldefault() int {
|
|
if runtime.GOOS == "windows" {
|
|
return Llevel | LstdFlags | Lshortfile
|
|
}
|
|
return Llevel | LstdFlags | Lshortfile | Llongcolor
|
|
}
|
|
|
|
func Version() string {
|
|
return "0.2.0.1121"
|
|
}
|
|
|
|
const (
|
|
Lall = iota
|
|
)
|
|
const (
|
|
Ldebug = iota
|
|
Linfo
|
|
Lwarn
|
|
Lerror
|
|
Lpanic
|
|
Lfatal
|
|
Lnone
|
|
)
|
|
|
|
const (
|
|
ForeBlack = iota + 30 //30
|
|
ForeRed //31
|
|
ForeGreen //32
|
|
ForeYellow //33
|
|
ForeBlue //34
|
|
ForePurple //35
|
|
ForeCyan //36
|
|
ForeWhite //37
|
|
)
|
|
|
|
const (
|
|
BackBlack = iota + 40 //40
|
|
BackRed //41
|
|
BackGreen //42
|
|
BackYellow //43
|
|
BackBlue //44
|
|
BackPurple //45
|
|
BackCyan //46
|
|
BackWhite //47
|
|
)
|
|
|
|
var levels = []string{
|
|
"[Debug]",
|
|
"[Info]",
|
|
"[Warn]",
|
|
"[Error]",
|
|
"[Panic]",
|
|
"[Fatal]",
|
|
}
|
|
|
|
// MUST called before all logs
|
|
func SetLevels(lvs []string) {
|
|
levels = lvs
|
|
}
|
|
|
|
var colors = []int{
|
|
ForeCyan,
|
|
ForeGreen,
|
|
ForeYellow,
|
|
ForeRed,
|
|
ForePurple,
|
|
ForeBlue,
|
|
}
|
|
|
|
// MUST called before all logs
|
|
func SetColors(cls []int) {
|
|
colors = cls
|
|
}
|
|
|
|
// A Logger represents an active logging object that generates lines of
|
|
// output to an io.Writer. Each logging operation makes a single call to
|
|
// the Writer's Write method. A Logger can be used simultaneously from
|
|
// multiple goroutines; it guarantees to serialize access to the Writer.
|
|
type Logger struct {
|
|
mu sync.Mutex // ensures atomic writes; protects the following fields
|
|
prefix string // prefix to write at beginning of each line
|
|
flag int // properties
|
|
Level int
|
|
out io.Writer // destination for output
|
|
buf bytes.Buffer // for accumulating text to write
|
|
levelStats [6]int64
|
|
loc *time.Location
|
|
}
|
|
|
|
// New creates a new Logger. The out variable sets the
|
|
// destination to which log data will be written.
|
|
// The prefix appears at the beginning of each generated log line.
|
|
// The flag argument defines the logging properties.
|
|
func New(out io.Writer, prefix string, flag int) *Logger {
|
|
l := &Logger{out: out, prefix: prefix, Level: 1, flag: flag, loc: time.Local}
|
|
if out != os.Stdout {
|
|
l.flag = RmColorFlags(l.flag)
|
|
}
|
|
return l
|
|
}
|
|
|
|
var Std = New(os.Stderr, "", Ldefault())
|
|
|
|
// Cheap integer to fixed-width decimal ASCII. Give a negative width to avoid zero-padding.
|
|
// Knows the buffer has capacity.
|
|
func itoa(buf *bytes.Buffer, i int, wid int) {
|
|
var u uint = uint(i)
|
|
if u == 0 && wid <= 1 {
|
|
buf.WriteByte('0')
|
|
return
|
|
}
|
|
|
|
// Assemble decimal in reverse order.
|
|
var b [32]byte
|
|
bp := len(b)
|
|
for ; u > 0 || wid > 0; u /= 10 {
|
|
bp--
|
|
wid--
|
|
b[bp] = byte(u%10) + '0'
|
|
}
|
|
|
|
// avoid slicing b to avoid an allocation.
|
|
for bp < len(b) {
|
|
buf.WriteByte(b[bp])
|
|
bp++
|
|
}
|
|
}
|
|
|
|
func moduleOf(file string) string {
|
|
pos := strings.LastIndex(file, "/")
|
|
if pos != -1 {
|
|
pos1 := strings.LastIndex(file[:pos], "/src/")
|
|
if pos1 != -1 {
|
|
return file[pos1+5 : pos]
|
|
}
|
|
}
|
|
return "UNKNOWN"
|
|
}
|
|
|
|
func (l *Logger) formatHeader(buf *bytes.Buffer, t time.Time,
|
|
file string, line int, lvl int, reqId string) {
|
|
if l.prefix != "" {
|
|
buf.WriteString(l.prefix)
|
|
}
|
|
if l.flag&(Ldate|Ltime|Lmicroseconds) != 0 {
|
|
if l.flag&Ldate != 0 {
|
|
year, month, day := t.Date()
|
|
itoa(buf, year, 4)
|
|
buf.WriteByte('/')
|
|
itoa(buf, int(month), 2)
|
|
buf.WriteByte('/')
|
|
itoa(buf, day, 2)
|
|
buf.WriteByte(' ')
|
|
}
|
|
if l.flag&(Ltime|Lmicroseconds) != 0 {
|
|
hour, min, sec := t.Clock()
|
|
itoa(buf, hour, 2)
|
|
buf.WriteByte(':')
|
|
itoa(buf, min, 2)
|
|
buf.WriteByte(':')
|
|
itoa(buf, sec, 2)
|
|
if l.flag&Lmicroseconds != 0 {
|
|
buf.WriteByte('.')
|
|
itoa(buf, t.Nanosecond()/1e3, 6)
|
|
}
|
|
buf.WriteByte(' ')
|
|
}
|
|
}
|
|
if reqId != "" {
|
|
buf.WriteByte('[')
|
|
buf.WriteString(reqId)
|
|
buf.WriteByte(']')
|
|
buf.WriteByte(' ')
|
|
}
|
|
|
|
if l.flag&(Lshortcolor|Llongcolor) != 0 {
|
|
buf.WriteString(fmt.Sprintf("\033[1;%dm", colors[lvl]))
|
|
}
|
|
if l.flag&Llevel != 0 {
|
|
buf.WriteString(levels[lvl])
|
|
buf.WriteByte(' ')
|
|
}
|
|
if l.flag&Lshortcolor != 0 {
|
|
buf.WriteString("\033[0m")
|
|
}
|
|
|
|
if l.flag&Lmodule != 0 {
|
|
buf.WriteByte('[')
|
|
buf.WriteString(moduleOf(file))
|
|
buf.WriteByte(']')
|
|
buf.WriteByte(' ')
|
|
}
|
|
if l.flag&(Lshortfile|Llongfile) != 0 {
|
|
if l.flag&Lshortfile != 0 {
|
|
short := file
|
|
for i := len(file) - 1; i > 0; i-- {
|
|
if file[i] == '/' {
|
|
short = file[i+1:]
|
|
break
|
|
}
|
|
}
|
|
file = short
|
|
}
|
|
buf.WriteString(file)
|
|
buf.WriteByte(':')
|
|
itoa(buf, line, -1)
|
|
buf.WriteByte(' ')
|
|
}
|
|
}
|
|
|
|
// Output writes the output for a logging event. The string s contains
|
|
// the text to print after the prefix specified by the flags of the
|
|
// Logger. A newline is appended if the last character of s is not
|
|
// already a newline. Calldepth is used to recover the PC and is
|
|
// provided for generality, although at the moment on all pre-defined
|
|
// paths it will be 2.
|
|
func (l *Logger) Output(reqId string, lvl int, calldepth int, s string) error {
|
|
if lvl < l.Level {
|
|
return nil
|
|
}
|
|
now := time.Now().In(l.loc) // get this early.
|
|
var file string
|
|
var line int
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
if l.flag&(Lshortfile|Llongfile|Lmodule) != 0 {
|
|
// release lock while getting caller info - it's expensive.
|
|
l.mu.Unlock()
|
|
var ok bool
|
|
_, file, line, ok = runtime.Caller(calldepth)
|
|
if !ok {
|
|
file = "???"
|
|
line = 0
|
|
}
|
|
l.mu.Lock()
|
|
}
|
|
l.levelStats[lvl]++
|
|
l.buf.Reset()
|
|
l.formatHeader(&l.buf, now, file, line, lvl, reqId)
|
|
l.buf.WriteString(s)
|
|
if l.flag&Llongcolor != 0 {
|
|
l.buf.WriteString("\033[0m")
|
|
}
|
|
if len(s) > 0 && s[len(s)-1] != '\n' {
|
|
l.buf.WriteByte('\n')
|
|
}
|
|
_, err := l.out.Write(l.buf.Bytes())
|
|
return err
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
// Printf calls l.Output to print to the logger.
|
|
// Arguments are handled in the manner of fmt.Printf.
|
|
func (l *Logger) Printf(format string, v ...interface{}) {
|
|
l.Output("", Linfo, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
// Print calls l.Output to print to the logger.
|
|
// Arguments are handled in the manner of fmt.Print.
|
|
func (l *Logger) Print(v ...interface{}) {
|
|
l.Output("", Linfo, 2, fmt.Sprint(v...))
|
|
}
|
|
|
|
// Println calls l.Output to print to the logger.
|
|
// Arguments are handled in the manner of fmt.Println.
|
|
func (l *Logger) Println(v ...interface{}) {
|
|
l.Output("", Linfo, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func (l *Logger) Debugf(format string, v ...interface{}) {
|
|
l.Output("", Ldebug, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func (l *Logger) Debug(v ...interface{}) {
|
|
l.Output("", Ldebug, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
func (l *Logger) Infof(format string, v ...interface{}) {
|
|
l.Output("", Linfo, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func (l *Logger) Info(v ...interface{}) {
|
|
l.Output("", Linfo, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
func (l *Logger) Warnf(format string, v ...interface{}) {
|
|
l.Output("", Lwarn, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func (l *Logger) Warn(v ...interface{}) {
|
|
l.Output("", Lwarn, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func (l *Logger) Errorf(format string, v ...interface{}) {
|
|
l.Output("", Lerror, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func (l *Logger) Error(v ...interface{}) {
|
|
l.Output("", Lerror, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func (l *Logger) Fatal(v ...interface{}) {
|
|
l.Output("", Lfatal, 2, fmt.Sprintln(v...))
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Fatalf is equivalent to l.Printf() followed by a call to os.Exit(1).
|
|
func (l *Logger) Fatalf(format string, v ...interface{}) {
|
|
l.Output("", Lfatal, 2, fmt.Sprintf(format, v...))
|
|
os.Exit(1)
|
|
}
|
|
|
|
// -----------------------------------------
|
|
// Panic is equivalent to l.Print() followed by a call to panic().
|
|
func (l *Logger) Panic(v ...interface{}) {
|
|
s := fmt.Sprintln(v...)
|
|
l.Output("", Lpanic, 2, s)
|
|
panic(s)
|
|
}
|
|
|
|
// Panicf is equivalent to l.Printf() followed by a call to panic().
|
|
func (l *Logger) Panicf(format string, v ...interface{}) {
|
|
s := fmt.Sprintf(format, v...)
|
|
l.Output("", Lpanic, 2, s)
|
|
panic(s)
|
|
}
|
|
|
|
// -----------------------------------------
|
|
func (l *Logger) Stack(v ...interface{}) {
|
|
s := fmt.Sprint(v...)
|
|
s += "\n"
|
|
buf := make([]byte, 1024*1024)
|
|
n := runtime.Stack(buf, true)
|
|
s += string(buf[:n])
|
|
s += "\n"
|
|
l.Output("", Lerror, 2, s)
|
|
}
|
|
|
|
// -----------------------------------------
|
|
func (l *Logger) Stat() (stats []int64) {
|
|
l.mu.Lock()
|
|
v := l.levelStats
|
|
l.mu.Unlock()
|
|
return v[:]
|
|
}
|
|
|
|
// Flags returns the output flags for the logger.
|
|
func (l *Logger) Flags() int {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
return l.flag
|
|
}
|
|
|
|
func RmColorFlags(flag int) int {
|
|
// for un std out, it should not show color since almost them don't support
|
|
if flag&Llongcolor != 0 {
|
|
flag = flag ^ Llongcolor
|
|
}
|
|
if flag&Lshortcolor != 0 {
|
|
flag = flag ^ Lshortcolor
|
|
}
|
|
return flag
|
|
}
|
|
|
|
func (l *Logger) Location() *time.Location {
|
|
return l.loc
|
|
}
|
|
|
|
func (l *Logger) SetLocation(loc *time.Location) {
|
|
l.loc = loc
|
|
}
|
|
|
|
// SetFlags sets the output flags for the logger.
|
|
func (l *Logger) SetFlags(flag int) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
if l.out != os.Stdout {
|
|
flag = RmColorFlags(flag)
|
|
}
|
|
l.flag = flag
|
|
}
|
|
|
|
// Prefix returns the output prefix for the logger.
|
|
func (l *Logger) Prefix() string {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
return l.prefix
|
|
}
|
|
|
|
// SetPrefix sets the output prefix for the logger.
|
|
func (l *Logger) SetPrefix(prefix string) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
l.prefix = prefix
|
|
}
|
|
|
|
// SetOutputLevel sets the output level for the logger.
|
|
func (l *Logger) SetOutputLevel(lvl int) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
l.Level = lvl
|
|
}
|
|
|
|
func (l *Logger) OutputLevel() int {
|
|
return l.Level
|
|
}
|
|
|
|
func (l *Logger) SetOutput(w io.Writer) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
l.out = w
|
|
if w != os.Stdout {
|
|
l.flag = RmColorFlags(l.flag)
|
|
}
|
|
}
|
|
|
|
// SetOutput sets the output destination for the standard logger.
|
|
func SetOutput(w io.Writer) {
|
|
Std.SetOutput(w)
|
|
}
|
|
|
|
func SetLocation(loc *time.Location) {
|
|
Std.SetLocation(loc)
|
|
}
|
|
|
|
func Location() *time.Location {
|
|
return Std.Location()
|
|
}
|
|
|
|
// Flags returns the output flags for the standard logger.
|
|
func Flags() int {
|
|
return Std.Flags()
|
|
}
|
|
|
|
// SetFlags sets the output flags for the standard logger.
|
|
func SetFlags(flag int) {
|
|
Std.SetFlags(flag)
|
|
}
|
|
|
|
// Prefix returns the output prefix for the standard logger.
|
|
func Prefix() string {
|
|
return Std.Prefix()
|
|
}
|
|
|
|
// SetPrefix sets the output prefix for the standard logger.
|
|
func SetPrefix(prefix string) {
|
|
Std.SetPrefix(prefix)
|
|
}
|
|
|
|
func SetOutputLevel(lvl int) {
|
|
Std.SetOutputLevel(lvl)
|
|
}
|
|
|
|
func OutputLevel() int {
|
|
return Std.OutputLevel()
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
// Print calls Output to print to the standard logger.
|
|
// Arguments are handled in the manner of fmt.Print.
|
|
func Print(v ...interface{}) {
|
|
Std.Output("", Linfo, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// Printf calls Output to print to the standard logger.
|
|
// Arguments are handled in the manner of fmt.Printf.
|
|
func Printf(format string, v ...interface{}) {
|
|
Std.Output("", Linfo, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
// Println calls Output to print to the standard logger.
|
|
// Arguments are handled in the manner of fmt.Println.
|
|
func Println(v ...interface{}) {
|
|
Std.Output("", Linfo, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func Debugf(format string, v ...interface{}) {
|
|
Std.Output("", Ldebug, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func Debug(v ...interface{}) {
|
|
Std.Output("", Ldebug, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func Infof(format string, v ...interface{}) {
|
|
Std.Output("", Linfo, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func Info(v ...interface{}) {
|
|
Std.Output("", Linfo, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func Warnf(format string, v ...interface{}) {
|
|
Std.Output("", Lwarn, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func Warn(v ...interface{}) {
|
|
Std.Output("", Lwarn, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func Errorf(format string, v ...interface{}) {
|
|
Std.Output("", Lerror, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
func Error(v ...interface{}) {
|
|
Std.Output("", Lerror, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
// Fatal is equivalent to Print() followed by a call to os.Exit(1).
|
|
func Fatal(v ...interface{}) {
|
|
Std.Output("", Lfatal, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// Fatalf is equivalent to Printf() followed by a call to os.Exit(1).
|
|
func Fatalf(format string, v ...interface{}) {
|
|
Std.Output("", Lfatal, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
// Panic is equivalent to Print() followed by a call to panic().
|
|
func Panic(v ...interface{}) {
|
|
Std.Output("", Lpanic, 2, fmt.Sprintln(v...))
|
|
}
|
|
|
|
// Panicf is equivalent to Printf() followed by a call to panic().
|
|
func Panicf(format string, v ...interface{}) {
|
|
Std.Output("", Lpanic, 2, fmt.Sprintf(format, v...))
|
|
}
|
|
|
|
// -----------------------------------------
|
|
|
|
func Stack(v ...interface{}) {
|
|
s := fmt.Sprint(v...)
|
|
s += "\n"
|
|
buf := make([]byte, 1024*1024)
|
|
n := runtime.Stack(buf, true)
|
|
s += string(buf[:n])
|
|
s += "\n"
|
|
Std.Output("", Lerror, 2, s)
|
|
}
|
|
|
|
// -----------------------------------------
|