From ce17bbf58e8245a1cf5e7d54d31b09beae46d620 Mon Sep 17 00:00:00 2001
From: dworld <dworld2006@gmail.com>
Date: Wed, 12 Nov 2014 09:57:15 +0800
Subject: [PATCH] use debugLogger

---
 config.go |  2 ++
 log.go    | 17 +++++++++++++++++
 server.go |  2 +-
 3 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/config.go b/config.go
index 9828c0c..0439abd 100644
--- a/config.go
+++ b/config.go
@@ -5,6 +5,8 @@ const (
 )
 
 type Config struct {
+	EnableDebug bool
+
 	CookieSecret string
 
 	StaticDir       string
diff --git a/log.go b/log.go
index a5f3f48..91657e4 100644
--- a/log.go
+++ b/log.go
@@ -7,3 +7,20 @@ type Logger interface {
 	Debugf(s string, o ...interface{})
 	Tracef(s string, o ...interface{})
 }
+
+type debugLogger struct {
+	Logger
+	enableDebug bool
+}
+
+func (l *debugLogger) Debugf(s string, o ...interface{}) {
+	if l.enableDebug {
+		l.Logger.Debugf(s, o...)
+	}
+}
+
+func (l *debugLogger) Tracef(s string, o ...interface{}) {
+	if l.enableDebug {
+		l.Logger.Tracef(s, o...)
+	}
+}
diff --git a/server.go b/server.go
index 239180a..b393014 100644
--- a/server.go
+++ b/server.go
@@ -33,7 +33,7 @@ func NewServer(config *Config) *Server {
 	}
 	setDefaultConfig(config)
 	server := &Server{
-		Logger: logex.NewLogger(1),
+		Logger: &debugLogger{Logger: logex.NewLogger(1), enableDebug: config.EnableDebug},
 		Config: config,
 	}
 	server.PanicHandler = DefaultPanicHandler