This commit is contained in:
laodaming
2023-06-06 16:14:02 +08:00
parent 245155d10e
commit 29f6e2ddbf
4 changed files with 49 additions and 43 deletions

View File

@@ -8,55 +8,48 @@ import (
{{.types}}
// Set 设置Response的Code和Message值
func (resp *Response) Set(Code int, Message string) *Response{
func (resp *Response) Set(Code int, Message string) {
resp.Code = Code
resp.Message = Message
return resp
}
// Set 设置整个Response
func (resp *Response) SetWithData(Code int, Message string, Data interface{}) *Response{
func (resp *Response) SetWithData(Code int, Message string, Data interface{}) {
resp.Code = Code
resp.Message = Message
resp.Data = Data
return resp
}
// SetMessage 设置Response的Message
func (resp *Response) SetMessage(msg string) *Response{
func (resp *Response) SetMessage(msg string) {
resp.Message = msg
return resp
}
// SetWithData 设置Data
func (resp *Response) SetData(Data interface{}) *Response{
func (resp *Response) SetData(Data interface{}) {
resp.Data = Data
return resp
}
// SetWithData 设置Response的Code和Message值 带Data入参数
func (resp *Response) SetCode(Code int) *Response{
func (resp *Response) SetCode(Code int) {
resp.Code = Code
return resp
}
// SetStatus 设置默认StatusResponse(内部自定义) 默认msg, 可以带data, data只使用一个参数
func (resp *Response) SetStatus(sr *basic.StatusResponse, data ...interface{}) *Response{
func (resp *Response) SetStatus(sr *basic.StatusResponse, data ...interface{}) {
resp.Code = sr.Code
resp.Message = sr.Message
if len(data) == 1 {
resp.Data = data[0]
}
return resp
}
// SetStatusWithMessage 设置默认StatusResponse(内部自定义) 非默认msg, 可以带data, data只使用一个参数
func (resp *Response) SetStatusWithMessage(sr *basic.StatusResponse, msg string, data ...interface{}) *Response{
func (resp *Response) SetStatusWithMessage(sr *basic.StatusResponse, msg string, data ...interface{}) {
resp.Code = sr.Code
resp.Message = msg
if len(data) == 1 {
resp.Data = data[0]
}
return resp
}