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.
38 lines
799 B
Go
38 lines
799 B
Go
package generator
|
|
|
|
import (
|
|
"os/exec"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util/console"
|
|
)
|
|
|
|
// DefaultGenerator defines the environment needs of rpc service generation
|
|
type DefaultGenerator struct {
|
|
log console.Console
|
|
}
|
|
|
|
// NewDefaultGenerator returns an instance of DefaultGenerator
|
|
func NewDefaultGenerator() *DefaultGenerator {
|
|
log := console.NewColorConsole()
|
|
return &DefaultGenerator{
|
|
log: log,
|
|
}
|
|
}
|
|
|
|
// Prepare provides environment detection generated by rpc service,
|
|
// including go environment, protoc, whether protoc-gen-go is installed or not
|
|
func (g *DefaultGenerator) Prepare() error {
|
|
_, err := exec.LookPath("go")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = exec.LookPath("protoc")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = exec.LookPath("protoc-gen-go")
|
|
return err
|
|
}
|