fix

goctl_template/api/types.tpl

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