Adds default file log option

blueszs · Apr 7, 2015 · 6c788a4 · 6c788a4
1 parent 0b7dce4
commit 6c788a4
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/logger.go b/logger.go
@@ -5,10 +5,9 @@
 package gin
 
 import (
-	"log"
+	"fmt"
+	"io"
 	"time"
-
-	"github.com/mattn/go-colorable"
 )
 
 var (
@@ -39,9 +38,10 @@ func ErrorLoggerT(typ uint32) HandlerFunc {
 }
 
 func Logger() HandlerFunc {
-	stdlogger := log.New(colorable.NewColorableStdout(), "", 0)
-	//errlogger := log.New(os.Stderr, "", 0)
+	return LoggerInFile(DefaultLogFile)
+}
 
+func LoggerInFile(out io.Writer) HandlerFunc {
 	return func(c *Context) {
 		// Start timer
 		start := time.Now()
@@ -58,15 +58,16 @@ func Logger() HandlerFunc {
 		statusCode := c.Writer.Status()
 		statusColor := colorForStatus(statusCode)
 		methodColor := colorForMethod(method)
+		comment := c.Errors.String()
 
-		stdlogger.Printf("[GIN] %v |%s %3d %s| %12v | %s |%s  %s %-7s %s\n%s",
+		fmt.Fprintf(out, "[GIN] %v |%s %3d %s| %12v | %s |%s  %s %-7s %s\n%s",
 			end.Format("2006/01/02 - 15:04:05"),
 			statusColor, statusCode, reset,
 			latency,
 			clientIP,
 			methodColor, reset, method,
 			c.Request.URL.Path,
-			c.Errors.String(),
+			comment,
 		)
 	}
 }

diff --git a/mode.go b/mode.go
@@ -7,6 +7,8 @@ package gin
 import (
 	"log"
 	"os"
+
+	"github.com/mattn/go-colorable"
 )
 
 const GIN_MODE = "GIN_MODE"
@@ -22,6 +24,7 @@ const (
 	testCode    = iota
 )
 
+var DefaultLogFile = colorable.NewColorableStdout()
 var ginMode int = debugCode
 var modeName string = DebugMode