You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package gogen
|
|
|
|
import (
|
|
_ "embed"
|
|
"strings"
|
|
|
|
"github.com/zeromicro/go-zero/tools/goctl/api/spec"
|
|
"github.com/zeromicro/go-zero/tools/goctl/config"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/format"
|
|
)
|
|
|
|
//go:embed middleware.tpl
|
|
var middlewareImplementCode string
|
|
|
|
func genMiddleware(dir string, cfg *config.Config, api *spec.ApiSpec) error {
|
|
middlewares := getMiddleware(api)
|
|
for _, item := range middlewares {
|
|
middlewareFilename := strings.TrimSuffix(strings.ToLower(item), "middleware") + "_middleware"
|
|
filename, err := format.FileNamingFormat(cfg.NamingFormat, middlewareFilename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
name := strings.TrimSuffix(item, "Middleware") + "Middleware"
|
|
err = genFile(fileGenConfig{
|
|
dir: dir,
|
|
subdir: middlewareDir,
|
|
filename: filename + ".go",
|
|
templateName: "contextTemplate",
|
|
category: category,
|
|
templateFile: middlewareImplementCodeFile,
|
|
builtinTemplate: middlewareImplementCode,
|
|
data: map[string]string{
|
|
"name": strings.Title(name),
|
|
},
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|