Sfoglia il codice sorgente

新增自定义返回

dcsunny 4 anni fa
parent
commit
0fd7c3ecc5
2 ha cambiato i file con 15 aggiunte e 1 eliminazioni
  1. 11 0
      codes/code.go
  2. 4 1
      http/handle.go

+ 11 - 0
codes/code.go

@@ -0,0 +1,11 @@
+package codes
+
+import "github.com/go-kratos/kratos/v2/errors"
+
+func SystemErr() error {
+	return errors.Error(10500, "", "")
+}
+
+func Customize() error {
+	return errors.Error(-1, "", "")
+}

+ 4 - 1
http/handle.go

@@ -15,10 +15,13 @@ func ErrHandle(w http.ResponseWriter, r *http.Request, err error) {
 	se, ok := errors.FromError(err)
 	if !ok {
 		se = &errors.StatusError{
-			Code:    2,
+			Code:    10500,
 			Message: err.Error(),
 		}
 	}
+	if se.Code == -1 {
+		return
+	}
 	codec := codecForRequest(r)
 	data, _ := codec.Marshal(se)
 	w.Header().Set(ContentTypeHeader, contentType(codec.Name()))