update stub generator template

Author: ehsannm

desc/service.go

@@ -177,9 +177,11 @@ func (s Service) dtoStub(stub *Stub) error {
 		if err != nil {
 			return err
 		}
-		err = stub.addDTO(reflect.TypeOf(c.Output))
-		if err != nil {
-			return err
+		if c.Output != nil {
+			err = stub.addDTO(reflect.TypeOf(c.Output))
+			if err != nil {
+				return err
+			}
 		}
 
 		for _, pe := range s.PossibleErrors {

exmples/simple-rest-server/stub/sampleservice.go

@@ -58,70 +58,58 @@ func NewSampleServiceStub(hostPort string, opts ...stub.Option) *SampleServiceSt
 	return s
 }
 
-type cannedEchoResponse struct {
-	EchoResponse *EchoResponse
-}
-
-func (s SampleServiceStub) Echo(req *EchoRequest) (cannedEchoResponse, error) {
-	res := cannedEchoResponse{
-		EchoResponse: &EchoResponse{},
-	}
+func (s SampleServiceStub) Echo(req *EchoRequest) (*EchoResponse, error) {
+	res := &EchoResponse{}
 	err := s.s.REST().
 		SetMethod("GET").
 		DefaultResponseHandler(nil).
 		AutoRun("/echo/:randomID", ronykit.JSON, req).
 		Err()
+	if err != nil {
+		return nil, err
+	}
 
-	return res, err
-}
-
-type cannedSum1Response struct {
-	SumResponse *SumResponse
+	return res, nil
 }
 
-func (s SampleServiceStub) Sum1(req *SumRequest) (cannedSum1Response, error) {
-	res := cannedSum1Response{
-		SumResponse: &SumResponse{},
-	}
+func (s SampleServiceStub) Sum1(req *SumRequest) (*SumResponse, error) {
+	res := &SumResponse{}
 	err := s.s.REST().
 		SetMethod("GET").
 		DefaultResponseHandler(nil).
 		AutoRun("/sum/:val1/:val2", ronykit.JSON, req).
 		Err()
+	if err != nil {
+		return nil, err
+	}
 
-	return res, err
-}
-
-type cannedSum2Response struct {
-	SumResponse *SumResponse
+	return res, nil
 }
 
-func (s SampleServiceStub) Sum2(req *SumRequest) (cannedSum2Response, error) {
-	res := cannedSum2Response{
-		SumResponse: &SumResponse{},
-	}
+func (s SampleServiceStub) Sum2(req *SumRequest) (*SumResponse, error) {
+	res := &SumResponse{}
 	err := s.s.REST().
 		SetMethod("POST").
 		DefaultResponseHandler(nil).
 		AutoRun("/sum", ronykit.JSON, req).
 		Err()
+	if err != nil {
+		return nil, err
+	}
 
-	return res, err
-}
-
-type cannedSumRedirectResponse struct {
-	SumResponse *SumResponse
+	return res, nil
 }
 
-func (s SampleServiceStub) SumRedirect(req *SumRequest) (cannedSumRedirectResponse, error) {
-	res := cannedSumRedirectResponse{
-		SumResponse: &SumResponse{},
-	}
+func (s SampleServiceStub) SumRedirect(req *SumRequest) (*SumResponse, error) {
+	res := &SumResponse{}
 	err := s.s.REST().
 		SetMethod("GET").
 		DefaultResponseHandler(nil).
 		AutoRun("/sum-redirect/:val1/:val2", ronykit.JSON, req).
 		Err()
+	if err != nil {
+		return nil, err
+	}
 
-	return res, err
+	return res, nil
 }

stub/gen/go/tpl/stub.gotmpl

@@ -51,18 +51,8 @@ func New{{$serviceName}}Stub(hostPort string, opts ...stub.Option) *{{$serviceNa
 {{ range .RESTs }}
 {{$methodName := .Name}}
 {{- if ne $methodName "" }}
-{{$cannedResponse := lowerCamelCase (print "canned" $methodName "Response")}}
-type {{$cannedResponse}} struct {
-	{{ range .Response -}}
-	{{.Name}} *{{.Name}}
-	{{- end }}
-}
-func (s {{$serviceName}}Stub) {{$methodName}}(req *{{.Request.Name}}) (*{{$cannedResponse}}, error){
-	res := {{$cannedResponse}}{
-		{{ range .Response -}}
-		{{.Name}}: &{{.Name}}{},
-		{{- end }}
-	}
+func (s {{$serviceName}}Stub) {{$methodName}}(req *{{.Request.Name}}) (*{{.Response.Name}}, error){
+	res := &{{.Response.Name}}{}
 	err := s.s.REST().
 		SetMethod("{{.Method}}").
 		DefaultResponseHandler(nil).