You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
678 B
33 lines
678 B
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
|
|
"github.com/go-kratos/kratos/v2"
|
|
"google.golang.org/protobuf/compiler/protogen"
|
|
"google.golang.org/protobuf/types/pluginpb"
|
|
)
|
|
|
|
var showVersion = flag.Bool("version", false, "print the version and exit")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *showVersion {
|
|
fmt.Printf("protoc-gen-go-errors %v\n", kratos.Release)
|
|
return
|
|
}
|
|
var flags flag.FlagSet
|
|
protogen.Options{
|
|
ParamFunc: flags.Set,
|
|
}.Run(func(gen *protogen.Plugin) error {
|
|
gen.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
|
|
for _, f := range gen.Files {
|
|
if !f.Generate {
|
|
continue
|
|
}
|
|
generateFile(gen, f)
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|