when call middle.Exception should set w.WriteHeader first

2013-12-18 21:00:45 +08:00 · 2013-12-18 21:00:45 +08:00 · 079a41136e
commit 079a41136e
parent e01fbd497c
1 changed files with 5 additions and 0 deletions
--- a/middleware/error.go
+++ b/middleware/error.go
@ -294,6 +294,11 @@ func RegisterErrorHander() {

 func Exception(errcode string, w http.ResponseWriter, r *http.Request, msg string) {
 	if h, ok := ErrorMaps[errcode]; ok {
+		isint, err := strconv.Atoi(errcode)
+		if err != nil {
+			isint = 500
+		}
+		w.WriteHeader(isint)
 		h(w, r)
 		return
 	} else {