Browse Source

Exposed Color()

Signed-off-by: Vishal Rana <vr@labstack.com>
Vishal Rana 9 năm trước cách đây
mục cha
commit
45b8c98c5e
1 tập tin đã thay đổi với 8 bổ sung5 xóa
  1. 8 5
      log/log.go

+ 8 - 5
log/log.go

@@ -13,7 +13,6 @@ import (
 
 	"strconv"
 
-	"github.com/mattn/go-isatty"
 	"github.com/valyala/fasttemplate"
 
 	"github.com/labstack/gommon/color"
@@ -112,10 +111,6 @@ func (l *Logger) Output() io.Writer {
 	return l.output
 }
 
-func (l *Logger) SetHeader(h string) {
-	l.template = l.newTemplate(h)
-}
-
 func (l *Logger) SetOutput(w io.Writer) {
 	l.output = w
 	if w, ok := w.(*os.File); !ok || !isatty.IsTerminal(w.Fd()) {
@@ -123,6 +118,14 @@ func (l *Logger) SetOutput(w io.Writer) {
 	}
 }
 
+func (l *Logger) Color() *color.Color {
+	return l.color
+}
+
+func (l *Logger) SetHeader(h string) {
+	l.template = l.newTemplate(h)
+}
+
 func (l *Logger) Print(i ...interface{}) {
 	fmt.Fprintln(l.output, i...)
 }