Commit cc5abc6b authored by JessonChan's avatar JessonChan

default atoi func to handle exception

parent 2a96f335
...@@ -391,19 +391,22 @@ func ErrorController(c ControllerInterface) *App { ...@@ -391,19 +391,22 @@ func ErrorController(c ControllerInterface) *App {
// show error string as simple text message. // show error string as simple text message.
// if error string is empty, show 503 or 500 error as default. // if error string is empty, show 503 or 500 error as default.
func exception(errCode string, ctx *context.Context) { func exception(errCode string, ctx *context.Context) {
for _, ec := range []string{errCode, "503", "500"} { atoi := func(code string) int {
code, _ := strconv.Atoi(ec) v, err := strconv.Atoi(code)
if code == 0 { if err == nil {
code = 503 return v
}
return 503
} }
for _, ec := range []string{errCode, "503", "500"} {
if h, ok := ErrorMaps[ec]; ok { if h, ok := ErrorMaps[ec]; ok {
executeError(h, ctx, code) executeError(h, ctx, atoi(ec))
return return
} }
} }
//if 50x error has been removed from errorMap //if 50x error has been removed from errorMap
//set 503 as default ctx.ResponseWriter.WriteHeader(atoi(errCode))
ctx.ResponseWriter.WriteHeader(503)
ctx.WriteString(errCode) ctx.WriteString(errCode)
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment