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.
32 lines
605 B
Go
32 lines
605 B
Go
4 years ago
|
package gogen
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
||
|
|
)
|
||
|
|
|
||
4 years ago
|
const svcTemplate = `package svc
|
||
|
4 years ago
|
|
||
|
|
import {{.imports}}
|
||
|
|
|
||
|
4 years ago
|
type ServiceContext struct {
|
||
|
|
c config.Config
|
||
|
|
}
|
||
|
4 years ago
|
|
||
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
||
|
|
return &ServiceContext{
|
||
|
|
c:c,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
func (g *defaultRpcGenerator) genSvc() error {
|
||
|
|
svcPath := g.dirM[dirSvc]
|
||
|
|
fileName := filepath.Join(svcPath, fileServiceContext)
|
||
|
|
return util.With("svc").GoFmt(true).Parse(svcTemplate).SaveTo(map[string]interface{}{
|
||
|
|
"imports": fmt.Sprintf(`"%v"`, g.mustGetPackage(dirConfig)),
|
||
|
|
}, fileName, false)
|
||
|
|
}
|