415 lines
11 KiB
Go
415 lines
11 KiB
Go
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
package log
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"runtime"
|
|
"runtime/debug"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"qoobing.com/gomod/str"
|
|
)
|
|
|
|
const (
|
|
PANIC LogLevel = 0
|
|
FATAL LogLevel = 1
|
|
ERROR LogLevel = 2
|
|
NOTICE LogLevel = 3
|
|
WARNING LogLevel = 4
|
|
INFO LogLevel = 5
|
|
DEBUG LogLevel = 9
|
|
ErrLogLevel = "~~~~~invalid loglevel~~~~~"
|
|
ErrLogPanic = "~~~~panic~~~~~~"
|
|
)
|
|
|
|
var (
|
|
mapStrToLevel = map[string]LogLevel{
|
|
"PANIC": PANIC,
|
|
"FATAL": FATAL,
|
|
"ERROR": ERROR,
|
|
"NOTICE": NOTICE,
|
|
"WARNING": WARNING,
|
|
"INFO": INFO,
|
|
"DEBUG": DEBUG,
|
|
}
|
|
mapLevelToStr = map[LogLevel]string{
|
|
PANIC: "PANIC",
|
|
FATAL: "FATAL",
|
|
ERROR: "ERROR",
|
|
NOTICE: "NOTICE",
|
|
WARNING: "WARNING",
|
|
INFO: "INFO",
|
|
DEBUG: "DEBUG",
|
|
}
|
|
)
|
|
|
|
var (
|
|
golog *log.Logger = nil
|
|
mylog *Logger = nil
|
|
logidCreator LogidCreator = nil
|
|
modulelogs map[string]*Logger = map[string]*Logger{}
|
|
)
|
|
|
|
type LogLevel int
|
|
type Logger struct {
|
|
module string
|
|
golog *log.Logger
|
|
loglevel LogLevel
|
|
skipfunc func(string) bool
|
|
calldepth int
|
|
}
|
|
|
|
type LogidCreator interface {
|
|
Cleanup()
|
|
GetLogid() string
|
|
SetLogid(logid string)
|
|
}
|
|
|
|
func New(module string) *Logger {
|
|
if module == "" {
|
|
module = "undefine"
|
|
}
|
|
if _, ok := modulelogs[module]; ok {
|
|
return modulelogs[module]
|
|
} else {
|
|
modulelogs[module] = &Logger{
|
|
module: module,
|
|
golog: golog,
|
|
loglevel: DEBUG,
|
|
}
|
|
return modulelogs[module]
|
|
}
|
|
}
|
|
|
|
// SetLogLevel set new log level
|
|
func (log *Logger) SetLogLevel(newlv LogLevel) (oldlv LogLevel) {
|
|
oldlv = log.loglevel
|
|
log.loglevel = newlv
|
|
return oldlv
|
|
}
|
|
|
|
// SetLogLevelByName set new log level by level name
|
|
func (log *Logger) SetLogLevelByName(newlv string) (oldlv string) {
|
|
if newlevel, ok := mapStrToLevel[newlv]; ok {
|
|
oldlevel := log.loglevel
|
|
log.loglevel = newlevel
|
|
return mapLevelToStr[oldlevel]
|
|
}
|
|
return ErrLogLevel
|
|
}
|
|
|
|
// SetCalldepth adjust the call depth
|
|
func (log *Logger) SetCalldepth(calldepth int) {
|
|
log.calldepth = calldepth
|
|
}
|
|
|
|
// SetSkipfunc set skip func to skip stack filename
|
|
func (log *Logger) SetSkipfunc(skipfunc func(string) bool) {
|
|
log.skipfunc = skipfunc
|
|
}
|
|
|
|
func (log *Logger) Print(prefix string, v interface{}) {
|
|
var str = ""
|
|
if pkg, err := json.Marshal(v); err != nil {
|
|
str = err.Error()
|
|
} else {
|
|
str = string(pkg)
|
|
}
|
|
log.logwrite(DEBUG, 3, prefix+"%s\n", str)
|
|
}
|
|
|
|
func (log *Logger) PrintPretty(prefix string, v interface{}) {
|
|
var preety = ""
|
|
if pkg, err := json.MarshalIndent(v, "==", " "); err != nil {
|
|
preety = err.Error()
|
|
} else {
|
|
preety = string(pkg)
|
|
}
|
|
log.logwrite(DEBUG, 3, prefix+"%s\n", preety)
|
|
}
|
|
|
|
func (log *Logger) Debugf(format string, v ...interface{}) {
|
|
log.logwrite(DEBUG, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) DebugfWithDepth(calldepth int, format string, v ...interface{}) {
|
|
calldepth += 3
|
|
log.logwrite(DEBUG, calldepth, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Noticef(format string, v ...interface{}) {
|
|
log.logwrite(NOTICE, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) NoticefWithDepth(calldepth int, format string, v ...interface{}) {
|
|
calldepth += 3
|
|
log.logwrite(NOTICE, calldepth, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Warningf(format string, v ...interface{}) {
|
|
log.logwrite(WARNING, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Infof(format string, v ...interface{}) {
|
|
log.logwrite(INFO, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Errorf(format string, v ...interface{}) {
|
|
log.logwrite(ERROR, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Fatalf(format string, v ...interface{}) {
|
|
log.logwrite(FATAL, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Panicf(format string, v ...interface{}) {
|
|
log.logwrite(PANIC, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Fatal(v ...interface{}) {
|
|
log.logwrite(FATAL, 3, "%s", fmt.Sprint(v...))
|
|
}
|
|
|
|
func (log *Logger) Printf(format string, v ...interface{}) {
|
|
log.logwrite(DEBUG, 3, format, v...)
|
|
}
|
|
|
|
func (log *Logger) Println(v ...interface{}) {
|
|
log.logwrite(DEBUG, 3, "%s", fmt.Sprint(v...))
|
|
}
|
|
|
|
func (log *Logger) GetLogidStr(format string) string {
|
|
if logidCreator != nil {
|
|
logid := logidCreator.GetLogid()
|
|
if logid != "" {
|
|
return fmt.Sprintf(format, logid)
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (log *Logger) logwrite(typ LogLevel, calldepth int, format string, v ...interface{}) {
|
|
if typ > log.loglevel {
|
|
return
|
|
}
|
|
if mylog == log {
|
|
calldepth = calldepth + 1
|
|
}
|
|
calldepth += log.calldepth
|
|
var (
|
|
idstr = log.GetLogidStr(" [%s]")
|
|
header = formatHeader(calldepth, log.skipfunc) + "$" + idstr
|
|
)
|
|
|
|
format = strings.Trim(format, "\n")
|
|
switch typ {
|
|
case INFO:
|
|
log.golog.SetPrefix("\x1b[2m" + "INFO " + header + "\x1b[0m" + " ")
|
|
case NOTICE:
|
|
log.golog.SetPrefix("\x1b[2m" + "NOTI " + header + "\x1b[0m" + " ")
|
|
case DEBUG:
|
|
log.golog.SetPrefix("\x1b[2m" + "DBUG " + header + "\x1b[0m" + " ")
|
|
case PANIC:
|
|
log.golog.SetPrefix("\x1b[31m" + "PANI " + header + "\x1b[0m" + " ")
|
|
case FATAL:
|
|
log.golog.SetPrefix("\x1b[31m" + "FATA " + header + "\x1b[0m" + " ")
|
|
case ERROR:
|
|
log.golog.SetPrefix("\x1b[41m" + "ERRO " + header + "\x1b[0m" + " ")
|
|
case WARNING:
|
|
log.golog.SetPrefix("\x1b[33m" + "WARN " + header + "\x1b[0m" + " ")
|
|
default:
|
|
log.golog.SetPrefix("\x1b[33m" + "UNKN " + header + "\x1b[0m" + " ")
|
|
}
|
|
|
|
if typ == FATAL || typ == WARNING || typ == ERROR {
|
|
log.golog.Output(calldepth, "\x1b[31m"+fmt.Sprintf(format, v...)+"\x1b[0m")
|
|
} else if typ == INFO || typ == DEBUG {
|
|
log.golog.Output(calldepth, fmt.Sprintf(format, v...))
|
|
} else if typ == NOTICE {
|
|
calldepth = calldepth + 2
|
|
log.golog.Output(calldepth, fmt.Sprintf(format, v...))
|
|
} else if typ == PANIC {
|
|
stackstr := strings.Replace(string(debug.Stack()), "\n", "\n== ", -1)
|
|
stackstr = "\n== Panic stack:\n" + str.SkipLine(stackstr, calldepth*2+1)
|
|
log.golog.Output(calldepth, fmt.Sprintf(format, v...)+stackstr)
|
|
panic(fmt.Sprintf(format, v...))
|
|
} else {
|
|
panic(ErrLogLevel)
|
|
}
|
|
}
|
|
|
|
func formatHeader(calldepth int, skipfunc func(string) bool) string {
|
|
const MAX_LENGTH = 35 /* + 22 */
|
|
// log time
|
|
now := time.Now()
|
|
year, month, day := now.Date()
|
|
hour, min, sec := now.Clock()
|
|
nowstr := fmt.Sprintf("%4d%02d%02dT%02d:%02d:%02d.%03d ",
|
|
year, month, day, hour, min, sec, now.Nanosecond()/1e6)
|
|
|
|
// log position
|
|
pcs := [13]uintptr{}
|
|
cnt := runtime.Callers(calldepth+1, pcs[:])
|
|
frames := runtime.CallersFrames(pcs[:cnt])
|
|
file, line := "???", 0
|
|
for i := 0; i < cnt; i++ {
|
|
frame, more := frames.Next()
|
|
if !more || skipfunc == nil || !skipfunc(frame.File) {
|
|
file, line = frame.File, frame.Line
|
|
break
|
|
}
|
|
}
|
|
|
|
// Case 1: too short path, return directly
|
|
i, header, linestr := 0, "", fmt.Sprintf("%d", line)
|
|
if len(file)+len(linestr) < MAX_LENGTH {
|
|
header = file + ":" + linestr
|
|
return nowstr + header
|
|
}
|
|
|
|
// Case 2: filename long enough, just return filename
|
|
var fileline = ""
|
|
for i = len(file) - 1; i >= 0; i-- {
|
|
if file[i] == '/' {
|
|
fileline = file[i:] + ":" + linestr
|
|
i--
|
|
break
|
|
}
|
|
}
|
|
if len(fileline) >= MAX_LENGTH && i > 0 {
|
|
//fileline = fileline[:4] + "*" + fileline[len(fileline)-MAX_LENGTH+6:]
|
|
return nowstr + fileline
|
|
} else if len(fileline) >= MAX_LENGTH && i <= 0 {
|
|
//fileline = fileline[:4] + "…" + fileline[len(fileline)-MAX_LENGTH+5:]
|
|
return nowstr + fileline
|
|
}
|
|
|
|
// Case 3: try complex middle path
|
|
header = fileline
|
|
tempheader := formatHeaderPath(file, i, 20, 2) + header
|
|
if len(tempheader) > MAX_LENGTH {
|
|
//tempheader = "…" + tempheader[len(tempheader)-MAX_LENGTH+1:]
|
|
tempheader = tempheader[len(tempheader)-MAX_LENGTH:]
|
|
return nowstr + tempheader
|
|
}
|
|
|
|
// Case 4: try complex middle path
|
|
tempheader = formatHeaderPath(file, i, 20, 3) + header
|
|
if len(tempheader) > MAX_LENGTH {
|
|
//tempheader = "…" + tempheader[len(tempheader)-MAX_LENGTH+1:]
|
|
tempheader = tempheader[len(tempheader)-MAX_LENGTH:]
|
|
return nowstr + tempheader
|
|
}
|
|
|
|
// Case 5: try complex middle path
|
|
tempheader = formatHeaderPath(file, i, 20, 4) + header
|
|
if len(tempheader) > MAX_LENGTH {
|
|
//tempheader = "…" + tempheader[len(tempheader)-MAX_LENGTH+1:]
|
|
tempheader = tempheader[len(tempheader)-MAX_LENGTH:]
|
|
return nowstr + tempheader
|
|
}
|
|
|
|
// Case 6: try complex middle path
|
|
tempheader = formatHeaderPath(file, i, 20, 5) + header
|
|
if len(tempheader) > MAX_LENGTH {
|
|
//tempheader = "…" + tempheader[len(tempheader)-MAX_LENGTH+1:]
|
|
tempheader = tempheader[len(tempheader)-MAX_LENGTH:]
|
|
return nowstr + tempheader
|
|
}
|
|
|
|
// Case 7: fallback
|
|
return nowstr + tempheader
|
|
}
|
|
|
|
func formatHeaderPath(file string, i, maxwidth, fullstep int) (paths string) {
|
|
ii, iw, step := i, i, 1
|
|
for i := ii; i >= 0; i-- {
|
|
if file[i] == '/' {
|
|
step++
|
|
if step < fullstep {
|
|
// dirname should not cut
|
|
paths = file[i:iw+1] + paths
|
|
} else if iw-i > maxwidth {
|
|
// dirname lenght more then width
|
|
paths = file[i:i+maxwidth] + paths // paths = file[i:i+width] + "..." + paths
|
|
} else {
|
|
// dirname lenght less then width
|
|
paths = file[i:iw+1] + paths
|
|
}
|
|
//fmt.Println("w", iw-i, "width", width, "paths", paths)
|
|
if step >= fullstep {
|
|
// decrease width
|
|
maxwidth = 2
|
|
}
|
|
iw = i - 1
|
|
continue
|
|
}
|
|
}
|
|
return paths
|
|
}
|
|
|
|
func needWriteLogToFile() bool {
|
|
if cls, ok := os.LookupEnv("QOOLOG_CLOSE_FILELOG"); ok && cls != "N" && cls != "n" {
|
|
return false
|
|
} else if ink8s, ok := os.LookupEnv("KUBERNETES_PORT"); ok && ink8s != "" {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func needWriteLogToStdout() bool {
|
|
if cls, ok := os.LookupEnv("QOOLOG_CLOSE_STDLOG"); ok && (cls == "Y" || cls == "y") {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// //////////////////////////////////////////////////////////////////////////////////
|
|
// ///////////////////////Initialize/////////////////////////////////////////////////
|
|
// //////////////////////////////////////////////////////////////////////////////////
|
|
var once sync.Once = sync.Once{}
|
|
|
|
// init
|
|
func init() {
|
|
once.Do(initlog)
|
|
}
|
|
|
|
// initlog
|
|
func initlog() {
|
|
// Step 1. create base writer & cache writer for ALL log
|
|
var (
|
|
logDir = "./logs/"
|
|
logName = getExeFilename()
|
|
logDirsOptional = []string{"./log", "/var/log"}
|
|
logSplitTag = "20060102" //20060102150405
|
|
baseWriter = NewBaseWriter(nil, os.Stdout, nil)
|
|
gologFlags = 0 /* log.Ldate | log.Lmicroseconds | log.Lshortfile */
|
|
)
|
|
|
|
// Step 2.1 add stdout to base writer if need
|
|
if needWriteLogToStdout() {
|
|
baseWriter.stdw = os.Stdout
|
|
}
|
|
// Step 2.2 rotate log(split log ) if need
|
|
if needWriteLogToFile() {
|
|
baseWriter.OpenLogFile(logDir, logName, logDirsOptional)
|
|
go func() {
|
|
for {
|
|
tag := time.Now().Local().Format(logSplitTag)
|
|
baseWriter.TryBackupLogFile(tag)
|
|
time.Sleep(2 * time.Minute)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// Step 3. init golog(golang std logger) & mylog(myself defined logger)
|
|
golog = log.New(NewCacheWriter(baseWriter), "", gologFlags)
|
|
mylog = New("system")
|
|
}
|