Compare commits

..

No commits in common. "main" and "v1.1.1" have entirely different histories.
main ... v1.1.1

7 changed files with 144 additions and 234 deletions

View File

@ -25,14 +25,10 @@ func Cleanup() {
} }
} }
func SetLogLevel(newlv LogLevel) (oldlv LogLevel) { func SetLogLevel(newlv logLevel) (oldlv logLevel) {
return mylog.SetLogLevel(newlv) return mylog.SetLogLevel(newlv)
} }
func SetLogLevelByName(newlv string) (oldlv string) {
return mylog.SetLogLevelByName(newlv)
}
func PrintPretty(prefix string, v interface{}) { func PrintPretty(prefix string, v interface{}) {
mylog.PrintPretty(prefix, v) mylog.PrintPretty(prefix, v)
} }
@ -62,7 +58,7 @@ func Warningf(format string, v ...interface{}) {
} }
func Errorf(format string, v ...interface{}) { func Errorf(format string, v ...interface{}) {
mylog.Errorf(format, v...) mylog.Warningf(format, v...)
} }
func Panicf(format string, v ...interface{}) { func Panicf(format string, v ...interface{}) {
@ -73,14 +69,6 @@ func Fatalf(format string, v ...interface{}) {
mylog.Fatalf(format, v...) mylog.Fatalf(format, v...)
} }
func Fatal(v ...interface{}){
mylog.Fatal(v...)
}
func Println(v ...interface{}){
mylog.Println(v...)
}
func TraceInto(format string, v ...interface{}) string { func TraceInto(format string, v ...interface{}) string {
_, fn, line, _ := runtime.Caller(1) _, fn, line, _ := runtime.Caller(1)
strfn := fmt.Sprintf("%s:%d", fn, line) strfn := fmt.Sprintf("%s:%d", fn, line)

6
go.mod
View File

@ -2,8 +2,4 @@ module qoobing.com/gomod/log
go 1.16 go 1.16
require ( require qoobing.com/gomod/str v1.0.1
github.com/tylerb/gls v0.0.0-20150407001822-e606233f194d
github.com/tylerb/is v2.1.4+incompatible // indirect
qoobing.com/gomod/str v1.0.1
)

343
log.go
View File

@ -3,244 +3,185 @@
package log package log
import ( import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"log" "log"
"os" "os"
"runtime/debug" "runtime/debug"
"strings" "strings"
"sync" "sync"
"time" "time"
"qoobing.com/gomod/str" "qoobing.com/gomod/str"
) )
const ( const (
PANIC LogLevel = 0 PANIC logLevel = 0
FATAL LogLevel = 1 FATAL logLevel = 1
ERROR LogLevel = 2 ERROR logLevel = 2
NOTICE LogLevel = 3 NOTICE logLevel = 3
WARNING LogLevel = 4 WARNING logLevel = 4
INFO LogLevel = 5 INFO logLevel = 5
DEBUG LogLevel = 9 DEBUG logLevel = 9
ErrLogLevel = "~~~~~invalid loglevel~~~~~" ErrLogPanic = "~~~~panic~~~~~~"
ErrLogPanic = "~~~~panic~~~~~~"
) )
var ( var (
mapStrToLevel = map[string]LogLevel{ golog *log.Logger = nil
"PANIC": PANIC, mylog *Logger = nil
"FATAL": FATAL, logidCreator LogidCreator = nil
"ERROR": ERROR, modulelogs map[string]*Logger = map[string]*Logger{}
"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 ( type logLevel int
golog *log.Logger = nil
mylog *Logger = nil
logidCreator LogidCreator = nil
modulelogs map[string]*Logger = map[string]*Logger{}
)
type LogLevel int
type Logger struct { type Logger struct {
module string module string
golog *log.Logger golog *log.Logger
loglevel LogLevel loglevel logLevel
} }
type LogidCreator interface { type LogidCreator interface {
Cleanup() Cleanup()
GetLogid() string GetLogid() string
SetLogid(logid string) SetLogid(logid string)
} }
func New(module string) *Logger { func New(module string) *Logger {
if module == "" { if module == "" {
module = "undefine" module = "undefine"
} }
if _, ok := modulelogs[module]; ok { if _, ok := modulelogs[module]; ok {
return modulelogs[module] return modulelogs[module]
} else { } else {
modulelogs[module] = &Logger{ modulelogs[module] = &Logger{
module: module, module: module,
golog: golog, golog: golog,
loglevel: DEBUG, loglevel: DEBUG,
} }
return modulelogs[module] return modulelogs[module]
} }
} }
// SetLogLevel set new log level // ////////////////////////////////////////////////////////////////
func (log *Logger) SetLogLevel(newlv LogLevel) (oldlv LogLevel) { func (log *Logger) SetLogLevel(newlv logLevel) (oldlv logLevel) {
oldlv = log.loglevel oldlv = log.loglevel
log.loglevel = newlv log.loglevel = newlv
return oldlv 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
}
func (log *Logger) Print(prefix string, v interface{}) { func (log *Logger) Print(prefix string, v interface{}) {
var str = "" preety, _ := json.Marshal(v)
if pkg, err := json.Marshal(v); err != nil { log.logwrite(DEBUG, 3, prefix+"%s\n", preety)
str = err.Error()
} else {
str = string(pkg)
}
log.logwrite(DEBUG, 3, prefix+"%s\n", str)
} }
func (log *Logger) PrintPretty(prefix string, v interface{}) { func (log *Logger) PrintPretty(prefix string, v interface{}) {
var preety = "" preety, _ := json.MarshalIndent(v, "==", " ")
if pkg, err := json.MarshalIndent(v, "==", " "); err != nil { log.logwrite(DEBUG, 3, prefix+"%s\n", preety)
preety = err.Error()
} else {
preety = string(pkg)
}
log.logwrite(DEBUG, 3, prefix+"%s\n", preety)
} }
func (log *Logger) Debugf(format string, v ...interface{}) { func (log *Logger) Debugf(format string, v ...interface{}) {
log.logwrite(DEBUG, 3, format, v...) log.logwrite(DEBUG, 3, format, v...)
} }
func (log *Logger) DebugfWithDepth(calldepth int, format string, v ...interface{}) { func (log *Logger) DebugfWithDepth(calldepth int, format string, v ...interface{}) {
calldepth += 3 calldepth += 3
log.logwrite(DEBUG, calldepth, format, v...) log.logwrite(DEBUG, calldepth, format, v...)
} }
func (log *Logger) Noticef(format string, v ...interface{}) { func (log *Logger) Noticef(format string, v ...interface{}) {
log.logwrite(NOTICE, 3, format, v...) log.logwrite(NOTICE, 3, format, v...)
} }
func (log *Logger) NoticefWithDepth(calldepth int, format string, v ...interface{}) { func (log *Logger) NoticefWithDepth(calldepth int, format string, v ...interface{}) {
calldepth += 3 calldepth += 3
log.logwrite(NOTICE, calldepth, format, v...) log.logwrite(NOTICE, calldepth, format, v...)
} }
func (log *Logger) Warningf(format string, v ...interface{}) { func (log *Logger) Warningf(format string, v ...interface{}) {
log.logwrite(WARNING, 3, format, v...) log.logwrite(WARNING, 3, format, v...)
} }
func (log *Logger) Infof(format string, v ...interface{}) { func (log *Logger) Infof(format string, v ...interface{}) {
log.logwrite(INFO, 3, format, v...) log.logwrite(INFO, 3, format, v...)
} }
func (log *Logger) Errorf(format string, v ...interface{}) { func (log *Logger) Errorf(format string, v ...interface{}) {
log.logwrite(ERROR, 3, format, v...) log.logwrite(ERROR, 3, format, v...)
} }
func (log *Logger) Fatalf(format string, v ...interface{}) { func (log *Logger) Fatalf(format string, v ...interface{}) {
log.logwrite(FATAL, 3, format, v...) log.logwrite(FATAL, 3, format, v...)
} }
func (log *Logger) Panicf(format string, v ...interface{}) { func (log *Logger) Panicf(format string, v ...interface{}) {
log.logwrite(PANIC, 3, format, v...) log.logwrite(PANIC, 3, format, v...)
} }
func (log *Logger) Fatal(v ...interface{}){
log.logwrite(FATAL, 3, "%s", fmt.Sprintln(v...))
}
func (log *Logger) Println(v ...interface{}){
log.logwrite(DEBUG, 3, "%s", fmt.Sprintln(v...))
}
func (log *Logger) GetLogidStr(format string) string { func (log *Logger) GetLogidStr(format string) string {
if logidCreator != nil { if logidCreator != nil {
return fmt.Sprintf(format, logidCreator.GetLogid()) return fmt.Sprintf(format, logidCreator.GetLogid())
} }
return "" return ""
} }
func (log *Logger) logwrite(typ LogLevel, calldepth int, format string, v ...interface{}) { func (log *Logger) logwrite(typ logLevel, calldepth int, format string, v ...interface{}) {
if typ > log.loglevel { if typ > log.loglevel {
return return
} }
var ( var (
idstr = log.GetLogidStr("[%s] ") idstr = log.GetLogidStr("[%s] ")
prestr = idstr // + "[" + log.module + "] " prestr = idstr // + "[" + log.module + "] "
) )
format = strings.Trim(format, "\n") format = strings.Trim(format, "\n")
switch typ { switch typ {
case PANIC: case PANIC:
log.golog.SetPrefix("\x1b[31m" + "PANI ") log.golog.SetPrefix("\x1b[31m" + "PANI ")
case FATAL: case FATAL:
log.golog.SetPrefix("\x1b[31m" + "FATA ") log.golog.SetPrefix("\x1b[31m" + "FATA ")
case WARNING: case WARNING:
log.golog.SetPrefix("\x1b[32m" + "WARN ") log.golog.SetPrefix("\x1b[32m" + "WARN ")
case ERROR: case ERROR:
log.golog.SetPrefix("\x1b[33m" + "ERRO ") log.golog.SetPrefix("\x1b[32m" + "ERRO ")
case INFO: case INFO:
log.golog.SetPrefix("INFO ") log.golog.SetPrefix("INFO ")
case NOTICE: case NOTICE:
log.golog.SetPrefix("NOTI ") log.golog.SetPrefix("NOTI ")
case DEBUG: case DEBUG:
log.golog.SetPrefix("DBUG ") log.golog.SetPrefix("DBUG ")
default: default:
log.golog.SetPrefix("UNKN ") log.golog.SetPrefix("UNKN ")
} }
if mylog == log { if mylog == log {
calldepth = calldepth + 1 calldepth = calldepth + 1
} }
if typ == FATAL || typ == WARNING || typ == ERROR { if typ == FATAL || typ == WARNING || typ == ERROR {
log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\x1b[0m\n", v...)) log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\x1b[0m\n", v...))
} else if typ == NOTICE { } else if typ == NOTICE {
calldepth = calldepth + 2 calldepth = calldepth + 2
log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\n", v...)) log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\n", v...))
} else if typ == PANIC { } else if typ == PANIC {
stack := strings.Replace(string(debug.Stack()), "\n", "\n== ", -1) stack := strings.Replace(string(debug.Stack()), "\n", "\n== ", -1)
stack = str.SkipLine(stack, calldepth*2+1) stack = str.SkipLine(stack, calldepth*2+1)
v = append(v, stack) v = append(v, stack)
panicstr := fmt.Sprintf(prestr+format+"\x1b[0m. Panic stack:\n%s\n", v...) panicstr := fmt.Sprintf(prestr+format+"\x1b[0m. Panic stack:\n%s\n", v...)
log.golog.Output(calldepth, panicstr) log.golog.Output(calldepth, panicstr)
panic(ErrLogPanic) panic(ErrLogPanic)
} else { } else {
log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\n", v...)) log.golog.Output(calldepth, prestr+fmt.Sprintf(format+"\n", v...))
} }
} }
func needWriteLogToFile() bool { func needWriteLogToFile() bool {
if cls, ok := os.LookupEnv("QOOLOG_CLOSE_FILELOG"); ok && cls != "N" && cls != "n" { if cls, ok := os.LookupEnv("QOOLOG_CLOSE_FILELOG"); ok && cls != "N" && cls != "n" {
return false return false
} else if ink8s, ok := os.LookupEnv("KUBERNETES_PORT"); ok && ink8s != "" { } else if ink8s, ok := os.LookupEnv("KUBERNETES_PORT"); ok && ink8s != "" {
return false return false
} }
return true return true
}
func needWriteLogToStdout() bool {
if cls, ok := os.LookupEnv("QOOLOG_CLOSE_STDLOG"); ok && (cls == "Y" || cls == "y") {
return false
}
return true
} }
// ////////////////////////////////////////////////////////////////////////////////// // //////////////////////////////////////////////////////////////////////////////////
@ -250,38 +191,34 @@ var once sync.Once = sync.Once{}
// init // init
func init() { func init() {
once.Do(initlog) once.Do(initlog)
} }
// initlog // initlog
func initlog() { func initlog() {
// Step 1. create base writer & cache writer for ALL log // Step 1. create base writer & cache writer for ALL log
var ( var (
logDir = "./logs/" logDir = "./logs/"
logName = getExeFilename() logName = getExeFilename()
logDirsOptional = []string{"./log", "/var/log"} logDirsOptional = []string{"./log", "/var/log"}
logSplitTag = "20060102" //20060102150405 logSplitTag = "20060102" //20060102150405
baseWriter = NewBaseWriter(nil, os.Stdout, nil) baseWriter = NewBaseWriter(nil, os.Stdout, nil)
gologFlags = log.Ldate | log.Lmicroseconds | log.Lshortfile gologFlags = log.Ldate | log.Lmicroseconds | log.Lshortfile
) )
// Step 2.1 add stdout to base writer if need // Step 2. init golog(golang std logger) & mylog(myself defined logger)
if needWriteLogToStdout() { golog = log.New(NewCacheWriter(baseWriter), "", gologFlags)
baseWriter.stdw = os.Stdout mylog = New("system")
}
// 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) // Step 3. backgroud split log (log rotate) if need
golog = log.New(NewCacheWriter(baseWriter), "", gologFlags) if needWriteLogToFile() {
mylog = New("system") go func() {
baseWriter.OpenLogFile(logDir, logName, logDirsOptional)
for {
tag := time.Now().Local().Format(logSplitTag)
baseWriter.TryBackupLogFile(tag)
time.Sleep(2 * time.Minute)
}
}()
}
} }

View File

@ -1,9 +0,0 @@
package log
import "testing"
func TestLog(t *testing.T) {
Println("aaa", 1,3)
Fatal("aaa", 1,3)
Fatalf("aaa=%d", 1)
}

View File

@ -1,5 +1,4 @@
// +build glslogid // +build glslogid
package log package log
import ( import (

View File

@ -1,5 +1,4 @@
// +build runtimelogid // +build runtimelogid
package log package log
//// Format writes the source file path and line number of the record to the buf. //// Format writes the source file path and line number of the record to the buf.

View File

@ -130,5 +130,5 @@ func (w *cacheWriter) StartCacheLog() {
func (w *cacheWriter) StopCacheLog() { func (w *cacheWriter) StopCacheLog() {
w.cacheLock.Unlock() w.cacheLock.Unlock()
w.cacheing = false w.cacheing = true
} }