修复SetStatus错误

This commit is contained in:
eson
2023-06-20 19:52:38 +08:00
parent 37f2326b21
commit b2062d6448
3 changed files with 10 additions and 30 deletions

View File

@@ -64,13 +64,17 @@ func (resp *Response) SetStatus(sr *StatusResponse, data ...interface{}) *Respon
Code: sr.Code,
Message: sr.Message,
}
if len(data) == 1 {
switch len(data) {
case 0:
// 0 直接返回
case 1:
if err, ok := data[0].(error); ok {
newResp.Message = err.Error()
} else {
newResp.Data = data[0]
}
} else if len(data) == 2 {
case 2:
var dataIndex = 0
if err, ok := data[dataIndex].(error); ok {
newResp.Message = err.Error()
@@ -79,7 +83,7 @@ func (resp *Response) SetStatus(sr *StatusResponse, data ...interface{}) *Respon
panic("data只接收1或2个 data 或 error.Error() 和 data ")
}
newResp.Data = data[1]
} else {
default:
panic("data只接收1或2个 data, error.Error() 和 data ")
}