rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: fix gen routes

Browse Source
This commit is contained in:
Rogee
2025-01-10 14:49:29 +08:00
parent 2eef4c35f4
commit be276b89ad
2 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cmd/gen_route.go
View File

@@ -4,7 +4,7 @@ import (
"io/fs" "io/fs"
"os" "os"
"path/filepath" "path/filepath"
"strings" "regexp"
"git.ipao.vip/rogeecn/atomctl/pkg/ast/route" "git.ipao.vip/rogeecn/atomctl/pkg/ast/route"
"git.ipao.vip/rogeecn/atomctl/pkg/utils/gomod" "git.ipao.vip/rogeecn/atomctl/pkg/utils/gomod"
@@ -51,11 +51,13 @@ func commandGenRouteE(cmd *cobra.Command, args []string) error {
log.Fatal("modules dir not exist, ", modulePath) log.Fatal("modules dir not exist, ", modulePath)
} }
controllerPattern := regexp.MustCompile(`controller(_?\w+)?\.go`)
err = filepath.WalkDir(modulePath, func(path string, d fs.DirEntry, err error) error { err = filepath.WalkDir(modulePath, func(path string, d fs.DirEntry, err error) error {
if d.IsDir() { if d.IsDir() {
return nil return nil
} }
if !strings.HasSuffix(path, "controller.go") {
if !controllerPattern.MatchString(d.Name()) {
return nil return nil
} }

7
templates/project/pkg/f/bind.go.tpl
View File

@@ -34,6 +34,13 @@ func Body[T any](name string) func(fiber.Ctx) (*T, error) {
} }
} }
func QueryParam[T fiber.GenericType](key string) func(fiber.Ctx) (T, error) {
return func(ctx fiber.Ctx) (T, error) {
v := fiber.Query[T](ctx, key)
return v, nil
}
}
func Query[T any](name string) func(fiber.Ctx) (*T, error) { func Query[T any](name string) func(fiber.Ctx) (*T, error) {
return func(ctx fiber.Ctx) (*T, error) { return func(ctx fiber.Ctx) (*T, error) {
p := new(T) p := new(T)