From ff9c0ea498a197e2967c93d32bc060f09d68a750 Mon Sep 17 00:00:00 2001
From: Phil <zgrubby@gmail.com>
Date: Wed, 28 Dec 2016 14:27:56 +0800
Subject: [PATCH] refactory generate log

---
 gollog.go       | 5 +----
 gollogOption.go | 4 +++-
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/gollog.go b/gollog.go
index 2aa95e5..52108d9 100644
--- 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
index c0d4cce..51c8f56 100644
--- 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
 }