gitroast/modules/log/log.go

349 lines
7.7 KiB
Go
Raw Normal View History

2014-02-12 19:54:09 +00:00
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package log
import (
2014-06-20 05:14:54 +00:00
"fmt"
2014-05-26 00:11:25 +00:00
"os"
2014-06-20 05:14:54 +00:00
"path"
2014-07-26 04:24:27 +00:00
"path/filepath"
"runtime"
"strings"
"golang.org/x/sync/syncmap"
2014-02-12 19:54:09 +00:00
)
2014-03-25 11:47:04 +00:00
var (
2016-11-26 11:53:29 +00:00
loggers []*Logger
// GitLogger logger for git
2014-07-26 04:24:27 +00:00
GitLogger *Logger
2014-03-25 11:47:04 +00:00
)
2014-02-12 19:54:09 +00:00
2016-11-26 11:53:29 +00:00
// NewLogger create a logger
2014-03-19 08:08:25 +00:00
func NewLogger(bufLen int64, mode, config string) {
2014-07-26 04:24:27 +00:00
logger := newLogger(bufLen)
isExist := false
for i, l := range loggers {
2014-07-26 04:24:27 +00:00
if l.adapter == mode {
isExist = true
loggers[i] = logger
}
}
if !isExist {
loggers = append(loggers, logger)
}
2014-06-20 04:25:23 +00:00
if err := logger.SetLogger(mode, config); err != nil {
Fatal(2, "Failed to set logger (%s): %v", mode, err)
2014-06-20 04:25:23 +00:00
}
2014-02-12 19:54:09 +00:00
}
// DelLogger removes loggers that are for the given mode
func DelLogger(mode string) error {
for _, l := range loggers {
if _, ok := l.outputs.Load(mode); ok {
return l.DelLogger(mode)
}
}
Trace("Log adapter %s not found, no need to delete", mode)
return nil
}
2016-11-26 11:53:29 +00:00
// NewGitLogger create a logger for git
2016-02-16 04:11:22 +00:00
// FIXME: use same log level as other loggers.
2014-06-20 05:14:54 +00:00
func NewGitLogger(logPath string) {
2016-11-30 23:56:15 +00:00
path := path.Dir(logPath)
if err := os.MkdirAll(path, os.ModePerm); err != nil {
Fatal(4, "Failed to create dir %s: %v", path, err)
2016-11-30 23:56:15 +00:00
}
2014-07-26 04:24:27 +00:00
GitLogger = newLogger(0)
2014-06-20 05:14:54 +00:00
GitLogger.SetLogger("file", fmt.Sprintf(`{"level":0,"filename":"%s","rotate":false}`, logPath))
}
2016-11-26 11:53:29 +00:00
// Trace records trace log
2014-02-18 22:48:02 +00:00
func Trace(format string, v ...interface{}) {
for _, logger := range loggers {
logger.Trace(format, v...)
}
2014-02-18 22:48:02 +00:00
}
2016-11-26 11:53:29 +00:00
// Debug records debug log
2014-03-23 16:16:17 +00:00
func Debug(format string, v ...interface{}) {
for _, logger := range loggers {
logger.Debug(format, v...)
}
2014-03-23 16:16:17 +00:00
}
2016-11-26 11:53:29 +00:00
// Info records info log
2014-02-12 19:54:09 +00:00
func Info(format string, v ...interface{}) {
for _, logger := range loggers {
logger.Info(format, v...)
}
2014-02-12 19:54:09 +00:00
}
2014-02-18 22:31:16 +00:00
// Warn records warning log
2014-07-26 04:24:27 +00:00
func Warn(format string, v ...interface{}) {
for _, logger := range loggers {
2014-07-26 04:24:27 +00:00
logger.Warn(format, v...)
}
2014-02-18 22:31:16 +00:00
}
2016-11-26 11:53:29 +00:00
// Error records error log
2014-07-26 04:24:27 +00:00
func Error(skip int, format string, v ...interface{}) {
for _, logger := range loggers {
2014-07-26 04:24:27 +00:00
logger.Error(skip, format, v...)
}
2014-02-18 22:31:16 +00:00
}
2014-02-18 22:48:02 +00:00
2016-11-26 11:53:29 +00:00
// Critical records critical log
2014-07-26 04:24:27 +00:00
func Critical(skip int, format string, v ...interface{}) {
for _, logger := range loggers {
2014-07-26 04:24:27 +00:00
logger.Critical(skip, format, v...)
}
2014-02-18 22:48:02 +00:00
}
2014-05-26 00:11:25 +00:00
2016-11-26 11:53:29 +00:00
// Fatal records error log and exit process
2014-07-26 04:24:27 +00:00
func Fatal(skip int, format string, v ...interface{}) {
Error(skip, format, v...)
2014-05-26 00:11:25 +00:00
for _, l := range loggers {
l.Close()
}
2014-07-26 04:24:27 +00:00
os.Exit(1)
}
2016-11-26 11:53:29 +00:00
// Close closes all the loggers
func Close() {
for _, l := range loggers {
l.Close()
}
}
// .___ __ _____
// | | _____/ |_ ____________/ ____\____ ____ ____
// | |/ \ __\/ __ \_ __ \ __\\__ \ _/ ___\/ __ \
// | | | \ | \ ___/| | \/| | / __ \\ \__\ ___/
// |___|___| /__| \___ >__| |__| (____ /\___ >___ >
// \/ \/ \/ \/ \/
2014-07-26 04:24:27 +00:00
2016-11-26 11:53:29 +00:00
// LogLevel level type for log
//type LogLevel int
2014-07-26 04:24:27 +00:00
2016-11-26 11:53:29 +00:00
// log levels
2014-07-26 04:24:27 +00:00
const (
TRACE = iota
DEBUG
INFO
WARN
ERROR
CRITICAL
FATAL
)
// LoggerInterface represents behaviors of a logger provider.
type LoggerInterface interface {
Init(config string) error
WriteMsg(msg string, skip, level int) error
Destroy()
Flush()
}
type loggerType func() LoggerInterface
var adapters = make(map[string]loggerType)
// Register registers given logger provider to adapters.
func Register(name string, log loggerType) {
if log == nil {
panic("log: register provider is nil")
}
if _, dup := adapters[name]; dup {
panic("log: register called twice for provider \"" + name + "\"")
}
adapters[name] = log
}
type logMsg struct {
skip, level int
msg string
}
// Logger is default logger in beego application.
// it can contain several providers and log message into all providers.
type Logger struct {
adapter string
level int
msg chan *logMsg
outputs syncmap.Map
2014-07-26 04:24:27 +00:00
quit chan bool
}
// newLogger initializes and returns a new logger.
func newLogger(buffer int64) *Logger {
l := &Logger{
msg: make(chan *logMsg, buffer),
quit: make(chan bool),
2014-07-26 04:24:27 +00:00
}
go l.StartLogger()
return l
}
2014-12-07 01:22:48 +00:00
// SetLogger sets new logger instance with given logger adapter and config.
2014-07-26 04:24:27 +00:00
func (l *Logger) SetLogger(adapter string, config string) error {
if log, ok := adapters[adapter]; ok {
lg := log()
if err := lg.Init(config); err != nil {
return err
}
l.outputs.Store(adapter, lg)
2014-07-26 04:24:27 +00:00
l.adapter = adapter
} else {
panic("log: unknown adapter \"" + adapter + "\" (forgotten register?)")
2014-07-26 04:24:27 +00:00
}
return nil
}
// DelLogger removes a logger adapter instance.
func (l *Logger) DelLogger(adapter string) error {
if lg, ok := l.outputs.Load(adapter); ok {
lg.(LoggerInterface).Destroy()
l.outputs.Delete(adapter)
2014-07-26 04:24:27 +00:00
} else {
panic("log: unknown adapter \"" + adapter + "\" (forgotten register?)")
2014-07-26 04:24:27 +00:00
}
return nil
}
func (l *Logger) writerMsg(skip, level int, msg string) error {
if l.level > level {
return nil
}
lm := &logMsg{
skip: skip,
level: level,
}
// Only error information needs locate position for debugging.
if lm.level >= ERROR {
pc, file, line, ok := runtime.Caller(skip)
if ok {
// Get caller function name.
fn := runtime.FuncForPC(pc)
var fnName string
if fn == nil {
fnName = "?()"
} else {
fnName = strings.TrimLeft(filepath.Ext(fn.Name()), ".") + "()"
}
fileName := file
if len(fileName) > 20 {
fileName = "..." + fileName[len(fileName)-20:]
}
lm.msg = fmt.Sprintf("[%s:%d %s] %s", fileName, line, fnName, msg)
2014-07-26 04:24:27 +00:00
} else {
lm.msg = msg
}
} else {
lm.msg = msg
}
l.msg <- lm
return nil
}
// StartLogger starts logger chan reading.
func (l *Logger) StartLogger() {
for {
select {
case bm := <-l.msg:
l.outputs.Range(func(k, v interface{}) bool {
if err := v.(LoggerInterface).WriteMsg(bm.msg, bm.skip, bm.level); err != nil {
2014-08-27 08:39:36 +00:00
fmt.Println("ERROR, unable to WriteMsg:", err)
}
return true
})
2014-07-26 04:24:27 +00:00
case <-l.quit:
return
}
}
}
// Flush flushes all chan data.
2014-07-26 04:24:27 +00:00
func (l *Logger) Flush() {
l.outputs.Range(func(k, v interface{}) bool {
v.(LoggerInterface).Flush()
return true
})
2014-07-26 04:24:27 +00:00
}
// Close closes logger, flush all chan data and destroy all adapter instances.
func (l *Logger) Close() {
l.quit <- true
for {
if len(l.msg) > 0 {
bm := <-l.msg
l.outputs.Range(func(k, v interface{}) bool {
if err := v.(LoggerInterface).WriteMsg(bm.msg, bm.skip, bm.level); err != nil {
2014-08-27 08:39:36 +00:00
fmt.Println("ERROR, unable to WriteMsg:", err)
}
return true
})
2014-07-26 04:24:27 +00:00
} else {
break
}
}
l.outputs.Range(func(k, v interface{}) bool {
v.(LoggerInterface).Flush()
v.(LoggerInterface).Destroy()
return true
})
2014-07-26 04:24:27 +00:00
}
2016-11-26 11:53:29 +00:00
// Trace records trace log
2014-07-26 04:24:27 +00:00
func (l *Logger) Trace(format string, v ...interface{}) {
msg := fmt.Sprintf("[T] "+format, v...)
l.writerMsg(0, TRACE, msg)
}
2016-11-26 11:53:29 +00:00
// Debug records debug log
2014-07-26 04:24:27 +00:00
func (l *Logger) Debug(format string, v ...interface{}) {
msg := fmt.Sprintf("[D] "+format, v...)
l.writerMsg(0, DEBUG, msg)
}
2016-11-26 11:53:29 +00:00
// Info records information log
2014-07-26 04:24:27 +00:00
func (l *Logger) Info(format string, v ...interface{}) {
msg := fmt.Sprintf("[I] "+format, v...)
l.writerMsg(0, INFO, msg)
}
// Warn records warning log
2014-07-26 04:24:27 +00:00
func (l *Logger) Warn(format string, v ...interface{}) {
msg := fmt.Sprintf("[W] "+format, v...)
l.writerMsg(0, WARN, msg)
}
2016-11-26 11:53:29 +00:00
// Error records error log
2014-07-26 04:24:27 +00:00
func (l *Logger) Error(skip int, format string, v ...interface{}) {
msg := fmt.Sprintf("[E] "+format, v...)
l.writerMsg(skip, ERROR, msg)
}
2016-11-26 11:53:29 +00:00
// Critical records critical log
2014-07-26 04:24:27 +00:00
func (l *Logger) Critical(skip int, format string, v ...interface{}) {
msg := fmt.Sprintf("[C] "+format, v...)
l.writerMsg(skip, CRITICAL, msg)
}
2016-11-26 11:53:29 +00:00
// Fatal records error log and exit the process
2014-07-26 04:24:27 +00:00
func (l *Logger) Fatal(skip int, format string, v ...interface{}) {
msg := fmt.Sprintf("[F] "+format, v...)
l.writerMsg(skip, FATAL, msg)
l.Close()
os.Exit(1)
2014-05-26 00:11:25 +00:00
}