refactory generate log

philchia · Dec 28, 2016 · ff9c0ea · ff9c0ea
1 parent 200e7f8
commit ff9c0ea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/gollog.go b/gollog.go
@@ -39,10 +39,7 @@ func (l *gollog) put(buf *bytes.Buffer) {
 }
 
 func (l *gollog) output(callDepth int, level LogLevel, msg string) {
-	buf := bufferPoolGet()
-	l.generateLog(buf, callDepth, level, msg)
-
-	l.put(buf)
+	l.put(l.generateLog(callDepth, level, msg))
 }
 
 // Debug will prinnt log as DEBUG level

diff --git a/gollogOption.go b/gollogOption.go
@@ -97,7 +97,8 @@ func getCaller(depth int) (file string, line int) {
 	return
 }
 
-func (l *gollog) generateLog(buf *bytes.Buffer, callDepth int, level LogLevel, msg string) {
+func (l *gollog) generateLog(callDepth int, level LogLevel, msg string) *bytes.Buffer {
+	buf := bufferPoolGet()
 	l.generatePrefix(buf, callDepth)
 
 	buf.Write(level.ColorString())
@@ -106,4 +107,5 @@ func (l *gollog) generateLog(buf *bytes.Buffer, callDepth int, level LogLevel, m
 	buf.WriteByte(' ')
 	buf.WriteString(msg)
 	buf.WriteByte('\n')
+	return buf
 }