Skip to content

Commit 0bb806b

Browse files
qloogLoong
andauthored
chore(cli): improve handler gen (#188)
* feat: improve kafka component * chore: revert test password * docs: add changelog * chore: upgrade sarama to IBM/sarama * chore: improve handler generation --------- Co-authored-by: Loong <richard@Loongs-MacBook-Pro.local>
1 parent 2ba4371 commit 0bb806b

4 files changed

Lines changed: 24 additions & 11 deletions

File tree

cmd/eagle/internal/handler/add/add.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ func run(cmd *cobra.Command, args []string) {
4343
Path: targetDir,
4444
Version: version,
4545
Method: method,
46+
ModName: utils.ModName(),
4647
}
4748
if err := c.Generate(); err != nil {
4849
fmt.Println(err)

cmd/eagle/internal/handler/add/handler.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ type Handler struct {
1616
Path string
1717
Version string
1818
Method string
19+
ModName string
1920
}
2021

2122
// Generate generate a Handler template.

cmd/eagle/internal/handler/add/template.go

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,8 @@ package add
22

33
import (
44
"bytes"
5+
"html/template"
56
"strings"
6-
7-
"github.com/alecthomas/template"
87
)
98

109
const handlerTemplate = `
@@ -15,48 +14,60 @@ import (
1514
"github.com/go-eagle/eagle/pkg/app"
1615
"github.com/go-eagle/eagle/pkg/errcode"
1716
18-
"github.com/go-eagle/eagle-layout/internal/service"
19-
"github.com/go-eagle/eagle-layout/internal/types"
17+
// "{{.ModName}}/internal/service"
18+
"{{.ModName}}/internal/types"
2019
)
2120
21+
// {{.Name}}Handler {{.LcName}}
22+
type {{.Name}}Handler struct {
23+
// here you can add your service
24+
// example:
25+
// UserService service.UserService
26+
}
27+
28+
// New{{.Name}}Handler create a new {{.Name}}Handler
29+
func New{{.Name}}Handler() *{{.Name}}Handler {
30+
return &{{.Name}}Handler{}
31+
}
32+
2233
// {{.Name}} {{.LcName}}
2334
// @Summary {{.LcName}}
2435
// @Description {{.LcName}}
2536
// @Tags system
2637
// @Accept json
2738
// @Produce json
2839
// @Router /{{.UsName}} {{.Method}}
29-
func {{.Name}}(c *gin.Context) {
40+
func (h *{{.Name}}Handler) {{.Name}}(c *gin.Context) {
3041
var req types.{{.Name}}Request
31-
{{- if .Method eq "GET" }}
42+
{{- if eq .Method "GET" }}
3243
if err := c.ShouldBindQuery(&req); err != nil {
3344
app.Error(c, errcode.ErrInvalidParam.WithDetails(err.Error()))
3445
return
3546
}
3647
{{- end }}
3748
38-
{{- if .Method eq "POST" }}
49+
{{- if eq .Method "POST" }}
3950
if err := c.ShouldBindJSON(&req); err != nil {
4051
app.Error(c, errcode.ErrInvalidParam.WithDetails(err.Error()))
4152
return
4253
}
4354
{{- end }}
4455
45-
{{- if .Method eq "PUT" }}
56+
{{- if eq .Method "PUT" }}
4657
if err := c.ShouldBindJSON(&req); err != nil {
4758
app.Error(c, errcode.ErrInvalidParam.WithDetails(err.Error()))
4859
return
4960
}
5061
{{- end }}
5162
52-
{{- if .Method eq "PATCH" }}
63+
{{- if eq .Method "PATCH" }}
5364
if err := c.ShouldBindJSON(&req); err != nil {
5465
app.Error(c, errcode.ErrInvalidParam.WithDetails(err.Error()))
5566
return
5667
}
5768
{{- end }}
5869
59-
{{- if .Method eq "DELETE" }}
70+
{{- if eq .Method "DELETE" }}
6071
if err := c.ShouldBindJSON(&req); err != nil {
6172
app.Error(c, errcode.ErrInvalidParam.WithDetails(err.Error()))
6273
return

cmd/eagle/main.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ import (
1919

2020
var (
2121
// Version is the version of the compiled software.
22-
Version = "v1.0.3"
22+
Version = "v1.0.4"
2323

2424
rootCmd = &cobra.Command{
2525
Use: "eagle",

0 commit comments

Comments
 (0)