227104d7d7
* Fix goctl completion expression * Fix code generation error if the pkg of pb/grpc is same to zrpc call client pkg * Remove deprecated comment on action goctl rpc new * Remove zrpc code generation on action goctl rpc proto * Remove zrpc code generation on action goctl rpc proto * Remove Generator interface Co-authored-by: anqiansong <anqiansong@bytedance.com>
35 lines
806 B
Go
35 lines
806 B
Go
package generator
|
|
|
|
import (
|
|
"log"
|
|
|
|
conf "github.com/zeromicro/go-zero/tools/goctl/config"
|
|
"github.com/zeromicro/go-zero/tools/goctl/env"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/console"
|
|
)
|
|
|
|
// Generator defines the environment needs of rpc service generation
|
|
type Generator struct {
|
|
log console.Console
|
|
cfg *conf.Config
|
|
}
|
|
|
|
// NewGenerator returns an instance of Generator
|
|
func NewGenerator(style string) *Generator {
|
|
cfg, err := conf.NewConfig(style)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
log := console.NewColorConsole()
|
|
return &Generator{
|
|
log: log,
|
|
cfg: cfg,
|
|
}
|
|
}
|
|
|
|
// Prepare provides environment detection generated by rpc service,
|
|
// including go environment, protoc, whether protoc-gen-go is installed or not
|
|
func (g *Generator) Prepare() error {
|
|
return env.Prepare(true, true)
|
|
}
|