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.
35 lines
755 B
35 lines
755 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")
|
|
omitempty = flag.Bool("omitempty", true, "omit if google.api is empty")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *showVersion {
|
|
fmt.Printf("protoc-gen-go-http %v\n", kratos.Release)
|
|
return
|
|
}
|
|
protogen.Options{
|
|
ParamFunc: flag.CommandLine.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, *omitempty)
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|