echo
/
framework1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改provider 实现

This commit is contained in:
dandan 2024-10-24 23:56:48 +08:00
parent ec23a6a72a
commit b40fc75627
1 changed files with 26 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
framework/command/provider.go
View File

@ -105,6 +105,12 @@ var providerCreateCommand = &cobra.Command{
if err := os.Mkdir(filepath.Join(pFolder, folder), 0700); err != nil { if err := os.Mkdir(filepath.Join(pFolder, folder), 0700); err != nil {
return err return err
} }
// 模板数据
config := container.MustMake(contract.ConfigKey).(contract.Config)
data := map[string]interface{}{
"appName": config.GetAppName(),
"packageName": name,
}
// 创建title这个模版方法 // 创建title这个模版方法
funcs := template.FuncMap{"title": strings.Title} funcs := template.FuncMap{"title": strings.Title}
{ {
@ -115,10 +121,10 @@ var providerCreateCommand = &cobra.Command{
return errors.Cause(err) return errors.Cause(err)
} }
// 使用contractTmp模版来初始化template，并且让这个模版支持title方法，即支持{{.|title}} // 使用contractTmp模版来初始化template，并且让这个模版支持title方法，即支持{{.packageName | title}}
t := template.Must(template.New("contract").Funcs(funcs).Parse(contractTmp)) t := template.Must(template.New("contract").Funcs(funcs).Parse(contractTmp))
// 将name传递进入到template中渲染，并且输出到contract.go 中 // 将name传递进入到template中渲染，并且输出到contract.go 中
if err := t.Execute(f, name); err != nil { if err := t.Execute(f, data); err != nil {
return errors.Cause(err) return errors.Cause(err)
} }
} }
@ -130,7 +136,7 @@ var providerCreateCommand = &cobra.Command{
return err return err
} }
t := template.Must(template.New("provider").Funcs(funcs).Parse(providerTmp)) t := template.Must(template.New("provider").Funcs(funcs).Parse(providerTmp))
if err := t.Execute(f, name); err != nil { if err := t.Execute(f, data); err != nil {
return err return err
} }
} }
@ -142,7 +148,7 @@ var providerCreateCommand = &cobra.Command{
return err return err
} }
t := template.Must(template.New("service").Funcs(funcs).Parse(serviceTmp)) t := template.Must(template.New("service").Funcs(funcs).Parse(serviceTmp))
if err := t.Execute(f, name); err != nil { if err := t.Execute(f, data); err != nil {
return err return err
} }
} }
@ -152,9 +158,9 @@ var providerCreateCommand = &cobra.Command{
}, },
} }
var contractTmp = `package {{.}} var contractTmp = `package {{.packageName}}
const {{.|title}}Key = "{{.}}" const {{.packageName | title}}Key = "{{.appName}}:{{.packageName}}"
type Service interface { type Service interface {
// 请在这里定义你的方法 // 请在这里定义你的方法
@ -162,54 +168,54 @@ type Service interface {
} }
` `
var providerTmp = `package {{.}} var providerTmp = `package {{.packageName}}
import ( import (
"github.com/Superdanda/hade/framework" "github.com/Superdanda/hade/framework"
) )
type {{.|title}}Provider struct { type {{.packageName | title}}Provider struct {
framework.ServiceProvider framework.ServiceProvider
c framework.Container c framework.Container
} }
func (sp *{{.|title}}Provider) Name() string { func (sp *{{.packageName | title}}Provider) Name() string {
return {{.|title}}Key return {{.packageName | title}}Key
} }
func (sp *{{.|title}}Provider) Register(c framework.Container) framework.NewInstance { func (sp *{{.packageName | title}}Provider) Register(c framework.Container) framework.NewInstance {
return New{{.|title}}Service return New{{.packageName | title}}Service
} }
func (sp *{{.|title}}Provider) IsDefer() bool { func (sp *{{.packageName | title}}Provider) IsDefer() bool {
return false return false
} }
func (sp *{{.|title}}Provider) Params(c framework.Container) []interface{} { func (sp *{{.packageName | title}}Provider) Params(c framework.Container) []interface{} {
return []interface{}{c} return []interface{}{c}
} }
func (sp *{{.|title}}Provider) Boot(c framework.Container) error { func (sp *{{.packageName | title}}Provider) Boot(c framework.Container) error {
return nil return nil
} }
` `
var serviceTmp = `package {{.}} var serviceTmp = `package {{.packageName}}
import "github.com/Superdanda/hade/framework" import "github.com/Superdanda/hade/framework"
type {{.|title}}Service struct { type {{.packageName | title}}Service struct {
container framework.Container container framework.Container
} }
func New{{.|title}}Service(params ...interface{}) (interface{}, error) { func New{{.packageName | title}}Service(params ...interface{}) (interface{}, error) {
container := params[0].(framework.Container) container := params[0].(framework.Container)
return &{{.|title}}Service{container: container}, nil return &{{.packageName | title}}Service{container: container}, nil
} }
func (s *{{.|title}}Service) Foo() string { func (s *{{.packageName | title}}Service) Foo() string {
return "" return ""
} }
` `