add log

Author: c-wind

log/log.go

@@ -0,0 +1,64 @@
+package log
+
+import (
+	"fmt"
+	"os"
+)
+
+var mlog = &Logger{out: os.Stdout, level: LogDebug, color: true}
+
+func SetLevel(level LogLevel) {
+	mlog.SetLevel(level)
+}
+
+func GetLogLevel() LogLevel {
+	return mlog.level
+}
+
+func Info(v ...interface{}) {
+	mlog.write(LogInfo, fmt.Sprint(v...))
+}
+
+func Infof(format string, v ...interface{}) {
+	mlog.write(LogInfo, format, v...)
+}
+
+func Debug(v ...interface{}) {
+	mlog.write(LogDebug,fmt.Sprint( v...))
+}
+
+func Debugf(format string, v ...interface{}) {
+	mlog.write(LogDebug, format, v...)
+}
+
+func Warning(v ...interface{}) {
+	mlog.write(LogWarning, fmt.Sprint( v...))
+}
+
+func Warningf(format string, v ...interface{}) {
+	mlog.write(LogWarning, format, v...)
+}
+
+func Error(v ...interface{}) {
+	mlog.write(LogError, fmt.Sprint( v...))
+}
+
+func Errorf(format string, v ...interface{}) {
+	mlog.write(LogError, format, v...)
+}
+
+func Fatal(v ...interface{}) {
+	mlog.write(LogFatal, fmt.Sprint( v...))
+}
+
+func Fatalf(format string, v ...interface{}) {
+	mlog.write(LogFatal, format, v...)
+}
+
+func SetLevelByString(level string) {
+	mlog.SetLevelByString(level)
+}
+
+func SetColor(color bool) {
+	mlog.SetColor(color)
+}

log/log_level.go

@@ -0,0 +1,66 @@
+package log
+
+import ()
+
+type LogLevel int
+
+const (
+	LogFatal LogLevel = iota
+	LogError
+	LogWarning
+	LogInfo
+	LogDebug
+)
+
+//StringToLogLevel 字符串转LogLevel.
+func StringToLogLevel(level string) LogLevel {
+	switch level {
+	case "fatal":
+		return LogFatal
+	case "error":
+		return LogError
+	case "warn":
+		return LogWarning
+	case "warning":
+		return LogWarning
+	case "debug":
+		return LogDebug
+	case "info":
+		return LogInfo
+	}
+	return LogDebug
+}
+
+//LogLevel loglevel 转字符串.
+func (l LogLevel) String() string {
+	switch l {
+	case LogFatal:
+		return "fatal"
+	case LogError:
+		return "error"
+	case LogWarning:
+		return "warning"
+	case LogDebug:
+		return "debug"
+	case LogInfo:
+		return "info"
+	}
+	return "unknown"
+}
+
+//LogLevel loglevel转颜色.
+func (l LogLevel) Color() string {
+	switch l {
+	case LogFatal:
+		return "\033[0;31m"
+	case LogError:
+		return "\033[0;31m"
+	case LogWarning:
+		return "\033[0;33m"
+	case LogDebug:
+		return "\033[0;36m"
+	case LogInfo:
+		return "\033[0;32m"
+	}
+	return "\033[0;37m"
+}

log/log_test.go

@@ -0,0 +1,19 @@
+package log
+
+import (
+	"testing"
+	"time"
+)
+
+func TestLog(t *testing.T) {
+	Debug("default log begin 11111111")
+	Debugf("%s, %+v", "abc", mlog)
+	Infof("%v test log", time.Now())
+
+	l := NewLogger()
+	l.Info("logger 2222222222")
+	l.Errorf("logger color %v xxxxxx", time.Now().UnixNano())
+    l.SetColor(false)
+	l.Errorf("logger no color %v yyyyyy", time.Now().UnixNano())
+	Infof("%v default has color test log", time.Now())
+}

log/log_writer.go

@@ -0,0 +1,152 @@
+package log
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+	"runtime"
+	"sync"
+	"time"
+)
+
+type Logger struct {
+	out   *os.File
+	level LogLevel
+	color bool
+	mu    sync.Mutex
+}
+
+//NewLogger 创建日志对象.
+func NewLogger() *Logger {
+	return &Logger{
+		out:   os.Stdout,
+		level: LogDebug,
+		color: true,
+	}
+}
+
+//SetColor 开启/关闭颜色.
+func (l *Logger) SetColor(on bool) *Logger {
+	l.color = on
+	return l
+}
+
+//SetLevel 设置日志级别.
+func (l *Logger) SetLevel(level LogLevel) {
+	l.level = level
+}
+
+//SetLevelByString 设置字符串格式的日志级别.
+func (l *Logger) SetLevelByString(level string) {
+	l.level = StringToLogLevel(level)
+}
+
+type pos struct {
+	file string
+	line int
+	name string
+}
+
+type posCache struct {
+	ps map[uintptr]pos
+	sync.RWMutex
+}
+
+var mpc = posCache{ps: make(map[uintptr]pos)}
+
+func (l *Logger) caller() (string, int, string) {
+	pc, file, line, _ := runtime.Caller(3)
+
+	mpc.RLock()
+	p, ok := mpc.ps[pc]
+	mpc.RUnlock()
+
+	if ok {
+		return p.file, p.line, p.name
+	}
+
+	name := runtime.FuncForPC(pc).Name()
+	if i := bytes.LastIndexAny([]byte(name), "."); i != -1 {
+		name = name[i+1:]
+	}
+	if i := bytes.LastIndexAny([]byte(file), "/"); i != -1 {
+		file = file[i+1:]
+	}
+
+	mpc.Lock()
+	mpc.ps[pc] = pos{file: file, line: line, name: name}
+	mpc.Unlock()
+
+	return file, line, name
+}
+
+func (l *Logger) write(t LogLevel, format string, argv ...interface{}) {
+	if t > l.level {
+		return
+	}
+
+	date := time.Now().Format("2006/01/02 15:04:05")
+
+	file, line, name := l.caller()
+
+	//时间，源码文件，源码列
+	fmt.Fprintf(l.out, "%s %s:%d ", date, file, line)
+	if l.color {
+		//颜色开始
+		fmt.Fprint(l.out, t.Color())
+	}
+
+	//函数名
+	fmt.Fprintf(l.out, "%s %s ", t.String(), name)
+
+	fmt.Fprintf(l.out, format, argv...)
+
+	if l.color {
+		//颜色结束
+		fmt.Fprint(l.out, "\033[0m")
+	}
+
+	l.out.WriteString("\n")
+}
+
+func (l *Logger) Info(v ...interface{}) {
+	l.write(LogInfo, fmt.Sprint(v...))
+}
+
+func (l *Logger) Infof(format string, v ...interface{}) {
+	l.write(LogInfo, format, v...)
+}
+
+func (l *Logger) Debug(v ...interface{}) {
+	l.write(LogDebug, fmt.Sprint(v...))
+}
+
+func (l *Logger) Debugf(format string, v ...interface{}) {
+	l.write(LogDebug, format, v...)
+}
+
+func (l *Logger) Warning(v ...interface{}) {
+	l.write(LogWarning, fmt.Sprint(v...))
+}
+
+func (l *Logger) Warningf(format string, v ...interface{}) {
+	l.write(LogWarning, format, v...)
+}
+
+func (l *Logger) Error(v ...interface{}) {
+	l.write(LogError, fmt.Sprint(v...))
+}
+
+func (l *Logger) Errorf(format string, v ...interface{}) {
+	l.write(LogError, format, v...)
+}
+
+func (l *Logger) Fatal(v ...interface{}) {
+	l.write(LogFatal, fmt.Sprint(v...))
+	os.Exit(-1)
+}
+
+func (l *Logger) Fatalf(format string, v ...interface{}) {
+	l.write(LogFatal, format, v...)
+	os.Exit(-1)
+}