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.
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package new
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/gogen"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const apiTemplate = `
|
|
type Request struct {
|
|
Name string ` + "`" + `path:"name,options=you|me"` + "`" + ` // 框架自动验证请求参数是否合法
|
|
}
|
|
|
|
type Response struct {
|
|
Message string ` + "`" + `json:"message"` + "`" + `
|
|
}
|
|
|
|
service {{.name}}-api {
|
|
@handler GreetHandler
|
|
get /greet/from/:name(Request) returns (Response);
|
|
}
|
|
`
|
|
|
|
func NewService(c *cli.Context) error {
|
|
args := c.Args()
|
|
dirName := "greet"
|
|
if len(args) > 0 {
|
|
dirName = args.First()
|
|
}
|
|
|
|
abs, err := filepath.Abs(dirName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = util.MkdirIfNotExist(abs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dirName = filepath.Base(filepath.Clean(abs))
|
|
filename := dirName + ".api"
|
|
apiFilePath := filepath.Join(abs, filename)
|
|
fp, err := os.Create(apiFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer fp.Close()
|
|
t := template.Must(template.New("template").Parse(apiTemplate))
|
|
if err := t.Execute(fp, map[string]string{
|
|
"name": dirName,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
err = gogen.DoGenProject(apiFilePath, abs, true)
|
|
return err
|
|
}
|