Useless package names when response type and request type are in diff…

…erent packages (cloudwego#194) Signed-off-by: rogerogers <[email protected]>
yuxialuozi · Jun 14, 2024 · 2b831c5 · 2b831c5
1 parent 14939c0
commit 2b831c5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/tpl/hertz/server/standard/package.yaml b/tpl/hertz/server/standard/package.yaml
@@ -33,13 +33,15 @@ layouts:
        }
        {{end}}
         {{if eq $MethodInfo.OutputDir "" -}}
-          resp,err := service.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
+          resp := &{{$MethodInfo.ReturnTypeName}}{}
+          resp,err = service.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
           if err != nil {
                utils.SendErrResponse(ctx, c, consts.StatusOK, err)
                return
           }
         {{else}}
-          resp,err := {{$MethodInfo.OutputDir}}.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
+          resp := &{{$MethodInfo.ReturnTypeName}}{}
+          resp,err = {{$MethodInfo.OutputDir}}.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
           if err != nil {
                   utils.SendErrResponse(ctx, c, consts.StatusOK, err)
                   return
@@ -133,7 +135,7 @@ layouts:
         w := ut.PerformRequest(h.Engine, "{{.HTTPMethod}}", path, body,header)
         resp := w.Result()
         t.Log(string(resp.Body()))
-        
+
         // todo edit your unit test.
         // assert.DeepEqual(t, 200, resp.StatusCode())
         // assert.DeepEqual(t, "null", string(resp.Body()))
@@ -158,9 +160,9 @@ layouts:
         w := ut.PerformRequest(h.Engine, "{{$MethodInfo.HTTPMethod}}", path, body,header)
         resp := w.Result()
         t.Log(string(resp.Body()))
-      
+
         // todo edit your unit test.
         // assert.DeepEqual(t, 200, resp.StatusCode())
         // assert.DeepEqual(t, "null", string(resp.Body()))
         }
-      {{end}}
+      {{end}}
diff --git a/tpl/hertz/server/standard_v2/package.yaml b/tpl/hertz/server/standard_v2/package.yaml
@@ -34,13 +34,15 @@ layouts:
        }
        {{end}}
         {{if eq $MethodInfo.OutputDir "" -}}
-          resp,err := service.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
+          resp := &{{$MethodInfo.ReturnTypeName}}{}
+          resp,err = service.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
           if err != nil {
                utils.SendErrResponse(ctx, c, consts.StatusInternalServerError, err)
                return
           }
         {{else}}
-          resp,err := {{$MethodInfo.OutputDir}}.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
+          resp := &{{$MethodInfo.ReturnTypeName}}{}
+          resp,err = {{$MethodInfo.OutputDir}}.New{{$MethodInfo.Name}}Service(ctx, c).Run(&req)
           if err != nil {
                   utils.SendErrResponse(ctx, c, consts.StatusInternalServerError, err)
                   return
@@ -134,7 +136,7 @@ layouts:
         w := ut.PerformRequest(h.Engine, "{{.HTTPMethod}}", path, body,header)
         resp := w.Result()
         t.Log(string(resp.Body()))
-        
+
         // todo edit your unit test.
         // assert.DeepEqual(t, 200, resp.StatusCode())
         // assert.DeepEqual(t, "null", string(resp.Body()))
@@ -159,9 +161,9 @@ layouts:
         w := ut.PerformRequest(h.Engine, "{{$MethodInfo.HTTPMethod}}", path, body,header)
         resp := w.Result()
         t.Log(string(resp.Body()))
-      
+
         // todo edit your unit test.
         // assert.DeepEqual(t, 200, resp.StatusCode())
         // assert.DeepEqual(t, "null", string(resp.Body()))
         }
-      {{end}}
+      {{end}}