添加windows彩色终端输出

Author: eruca

console.go

@@ -6,8 +6,29 @@ import (
 	"github.com/mattn/go-colorable"
 )
 
+type Brush func(string) string
+
+func NewBrush(color string) Brush {
+	pre := "\033["
+	reset := "\033[0m"
+	return func(text string) string {
+		return pre + color + "m" + text + reset
+	}
+}
+
+var colors = []Brush{
+	NewBrush("1;36"), // Trace      cyan
+	NewBrush("1;34"), // Debug      blue
+	NewBrush("1;32"), // Info       green
+	NewBrush("1;33"), // Warn       yellow
+	NewBrush("1;31"), // Error      red
+	NewBrush("1;35"), // Critical   purple
+	NewBrush("1;31"), // Fatal      red
+}
+
 type console struct {
-	lg *log.Logger
+	level LogLevel
+	lg    *log.Logger
 }
 
 func newConsole() LoggerInterface {
@@ -16,11 +37,15 @@ func newConsole() LoggerInterface {
 	}
 }
 
-func (c *console) WriteMsg(msg string) error {
-	c.lg.Println(msg)
+func (c *console) WriteMsg(msg string, level LogLevel) error {
+	if c.level > level {
+		return nil
+	}
+
+	c.lg.Println(colors[level](msg))
 	return nil
 }
 
 func (*console) Flush() {}
 
-func (*console) Destroy() {}
+func (*console) Close() {}

log_test.go

@@ -0,0 +1,18 @@
+package logs
+
+import (
+	"reflect"
+	"testing"
+)
+
+func expect(t *testing.T, a, b interface{}) {
+	if a != b {
+		t.Errorf("Expected %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
+	}
+}
+
+func refute(t *testing.T, a, b interface{}) {
+	if a == b {
+		t.Errorf("Did not expect %v (type %v) - Got %v (type %v)", b, reflect.TypeOf(b), a, reflect.TypeOf(a))
+	}
+}

logs.go

@@ -38,17 +38,22 @@ const (
 )
 
 type LoggerInterface interface {
-	Write(data []byte) (int, error)
+	Init()
+	WriteMsg(msg string, level LogLevel) error
 	Close()
 	Flush()
 }
 
 // log instance generator
 type LoggerGenerator func() LoggerInterface
 
-var (
+var adapters map[LogType]LoggerGenerator
+
+func init() {
 	adapters = make(map[LogType]LoggerGenerator)
-)
+
+	adapters[LogConsole] = newConsole
+}
 
 type logMsg struct {
 	skip  int
@@ -64,25 +69,34 @@ type Logger struct {
 	quit    chan bool
 }
 
-func newLogger(buffer int64) *Logger {
+func NewLogger() *Logger {
+	numCPU := runtime.NumCPU()
+
+	l := newLogger(numCPU)
+	l.AddLogger(LogConsole)
+
+	return l
+}
+
+func newLogger(buffer int) *Logger {
 	l := &Logger{
 		level:   TRACE,
 		outputs: make(map[LogType]LoggerInterface),
 		msgChan: make(chan *logMsg, buffer),
 		quit:    make(chan bool),
 	}
 
-	go l.StartLogger()
+	go l.startLogger()
 	return l
 }
 
-func (l *Logger) AddLogger(t LogType, config string) error {
+func (l *Logger) AddLogger(t LogType) error {
 	l.Lock()
 	defer l.Unlock()
 
 	if logGen, ok := adapters[t]; ok {
 		logInst := logGen()
-		if err := logInst.Init(config); err != nil {
+		if err := logInst.Init(); err != nil {
 			return err
 		}
 		l.outputs[t] = logInst
@@ -142,7 +156,8 @@ func (l *Logger) writeMsg(skip int, level LogLevel, msg string) error {
 	return nil
 }
 
-func (l *Logger) StartLogger() {
+// 还需处理意外的情况，ctrl+c退出
+func (l *Logger) startLogger() {
 	for {
 		select {
 		case bm := <-l.msgChan: