set Status when newing codeint.Error

xgfone · Apr 21, 2024 · fefeb89 · fefeb89
1 parent 5449dcf
commit fefeb89
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/result/codeint/error.go b/result/codeint/error.go
@@ -39,7 +39,7 @@ type Error struct {
 }
 
 // NewError returns a new Error with the code.
-func NewError(code int) Error { return Error{Code: code} }
+func NewError(code int) Error { return Error{Code: code}.WithStatus(code) }
 
 // IsZero reports whether e is ZERO.
 func (e Error) IsZero() bool {
@@ -98,7 +98,11 @@ func (e Error) WithMessage(msg string, args ...interface{}) Error {
 
 // WithStatus returns a new Error with the status.
 func (e Error) WithStatus(status int) Error {
-	e.Status = status
+	if status < 600 {
+		e.Status = status
+	} else {
+		e.Status = 500
+	}
 	return e
 }