重构应用未处理异常代码，add DotWeb.DefaultHTTPErrorHandler，add DotWeb.SetDebugMode

devfeel · devfeel · commit 7ea5af2c1dfb · 2017-03-09T15:28:34.000+08:00
diff --git a/dotweb.go b/dotweb.go
@@ -17,6 +17,7 @@ type Dotweb struct {
 	Modules          []*HttpModule
 	logpath          string
 	ExceptionHandler ExceptionHandle
+	DebugMode        bool
 }
 
 type ExceptionHandle func(*HttpContext, interface{})
@@ -28,6 +29,7 @@ func New() *Dotweb {
 	dotweb := &Dotweb{
 		HttpServer: NewHttpServer(),
 		Modules:    make([]*HttpModule, 0, 10),
+		DebugMode:  false,
 	}
 	dotweb.HttpServer.setDotweb(dotweb)
 
@@ -41,6 +43,13 @@ func (ds *Dotweb) RegisterModule(module *HttpModule) {
 	ds.Modules = append(ds.Modules, module)
 }
 
+/*
+设置Debug模式，默认为false
+*/
+func (ds *Dotweb) SetDebugMode(isDebug bool) {
+	ds.DebugMode = isDebug
+}
+
 /*
 * 设置异常处理函数
  */
@@ -80,7 +89,7 @@ func (ds *Dotweb) StartServer(httpport int) error {
 	ds.HttpServer.GET("/dotweb/query/:key", showQuery)
 
 	if ds.ExceptionHandler == nil {
-		ds.SetExceptionHandle(DefaultHTTPErrorHandler)
+		ds.SetExceptionHandle(ds.DefaultHTTPErrorHandler)
 	}
 
 	port := ":" + strconv.Itoa(httpport)
@@ -89,6 +98,18 @@ func (ds *Dotweb) StartServer(httpport int) error {
 	return err
 }
 
+//默认异常处理
+func (ds *Dotweb) DefaultHTTPErrorHandler(ctx *HttpContext, errinfo interface{}) {
+	//输出内容
+	ctx.Response.WriteHeader(http.StatusInternalServerError)
+	ctx.Response.Header().Set(HeaderContentType, CharsetUTF8)
+	if ds.DebugMode {
+		ctx.WriteString(fmt.Sprintln(errinfo))
+	} else {
+		ctx.WriteString("Internal Server Error")
+	}
+}
+
 //query pprof debug info
 //key:heap goroutine threadcreate block
 func initPProf(ctx *HttpContext) {
@@ -118,11 +139,3 @@ func showQuery(ctx *HttpContext) {
 		ctx.WriteString("not support key => " + querykey)
 	}
 }
-
-//默认异常处理
-func DefaultHTTPErrorHandler(ctx *HttpContext, errinfo interface{}) {
-	//输出内容
-	ctx.Response.WriteHeader(http.StatusInternalServerError)
-	ctx.Response.Header().Set(HeaderContentType, CharsetUTF8)
-	ctx.WriteString(fmt.Sprintln(errinfo))
-}
diff --git a/example/main.go b/example/main.go
@@ -14,6 +14,9 @@ func main() {
 	//设置dotserver日志目录
 	dotserver.SetLogPath(file.GetCurrentDirectory())
 
+	//设置Debug模式
+	dotserver.SetDebugMode(true)
+
 	//设置路由
 	InitRoute(dotserver)
 
@@ -39,7 +42,7 @@ func IndexReg(ctx *dotweb.HttpContext) {
 }
 
 func DefaultError(ctx *dotweb.HttpContext) {
-	panic("DefaultError")
+	panic("my panic error!")
 }
 
 func InitRoute(dotserver *dotweb.Dotweb) {

Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,9 @@ func main() {`
`14`	`14`	`//设置dotserver日志目录`
`15`	`15`	`dotserver.SetLogPath(file.GetCurrentDirectory())`
`16`	`16`
	`17`	`+ //设置Debug模式`
	`18`	`+ dotserver.SetDebugMode(true)`
	`19`	`+`
`17`	`20`	`//设置路由`
`18`	`21`	`InitRoute(dotserver)`
`19`	`22`
`@@ -39,7 +42,7 @@ func IndexReg(ctx *dotweb.HttpContext) {`
`39`	`42`	`}`
`40`	`43`
`41`	`44`	`func DefaultError(ctx *dotweb.HttpContext) {`
`42`		`- panic("DefaultError")`
	`45`	`+ panic("my panic error!")`
`43`	`46`	`}`
`44`	`47`
`45`	`48`	`func InitRoute(dotserver *dotweb.Dotweb) {`