release: new version
This commit is contained in:
parent
91a5eb75b8
commit
d1f399b03d
GPG Key ID:
A23CF9620CBB55F9
20
README.md
Normal file
20
README.md
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
# protoc-gen-coco
|
||||||
|
|
||||||
|
## install
|
||||||
|
|
||||||
|
```shell
|
||||||
|
go install gitter.top/coco/protoc-gen-coco@latest
|
||||||
|
```
|
||||||
|
|
||||||
|
## usage
|
||||||
|
|
||||||
|
```shell
|
||||||
|
protoc example.proto --coco_out=. --go_out=.
|
||||||
|
```
|
||||||
|
|
||||||
|
## debug
|
||||||
|
|
||||||
|
```shell
|
||||||
|
go install .
|
||||||
|
protoc --coco_out=. .\example_errcode.proto --go_out=.
|
||||||
|
```
|
||||||
115
comment_parse.go
115
comment_parse.go
@ -1,9 +1,10 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"google.golang.org/protobuf/compiler/protogen"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
)
|
)
|
||||||
|
|
||||||
func IsCommentRouterGroup(comments protogen.CommentSet) bool {
|
func IsCommentRouterGroup(comments protogen.CommentSet) bool {
|
||||||
@ -41,9 +42,26 @@ func GetCommentGenerateTo(comments protogen.CommentSet) string {
|
|||||||
if len(match[0]) != 2 {
|
if len(match[0]) != 2 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return match[0][1]
|
return trim(match[0][1])
|
||||||
}
|
}
|
||||||
return "/"
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentRpcGenerateTo(comments protogen.CommentSet) string {
|
||||||
|
raw := comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@rpc_to:\\s*([\\w|/|\\.]*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return trim(match[0][1])
|
||||||
|
}
|
||||||
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCommentBaseURL(comments protogen.CommentSet) string {
|
func GetCommentBaseURL(comments protogen.CommentSet) string {
|
||||||
@ -58,7 +76,7 @@ func GetCommentBaseURL(comments protogen.CommentSet) string {
|
|||||||
if len(match[0]) != 2 {
|
if len(match[0]) != 2 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return match[0][1]
|
return trim(match[0][1])
|
||||||
}
|
}
|
||||||
return "/"
|
return "/"
|
||||||
}
|
}
|
||||||
@ -75,7 +93,7 @@ func GetCommentHttpMethod(comments protogen.CommentSet) string {
|
|||||||
if len(match[0]) != 2 {
|
if len(match[0]) != 2 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return match[0][1]
|
return trim(match[0][1])
|
||||||
}
|
}
|
||||||
return "GET"
|
return "GET"
|
||||||
}
|
}
|
||||||
@ -92,7 +110,7 @@ func GetCommentAuthor(comments protogen.CommentSet) string {
|
|||||||
if len(match[0]) != 2 {
|
if len(match[0]) != 2 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return match[0][1]
|
return trim(match[0][1])
|
||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
@ -109,7 +127,7 @@ func GetCommentDescribe(comments protogen.CommentSet) string {
|
|||||||
if len(match[0]) != 2 {
|
if len(match[0]) != 2 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
return match[0][1]
|
return trim(match[0][1])
|
||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
@ -130,3 +148,86 @@ func GetCommentApiURL(method *protogen.Method) string {
|
|||||||
}
|
}
|
||||||
return "/" + CamelCaseToJavascriptCase(method.GoName)
|
return "/" + CamelCaseToJavascriptCase(method.GoName)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetCommentTableName(message *protogen.Message) string {
|
||||||
|
raw := message.Comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@table_name:\\s*(.*)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return trim(match[0][1])
|
||||||
|
}
|
||||||
|
return CamelCaseToUnderscore(message.GoIdent.GoName)
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetCommentBsonName(field *protogen.Field) string {
|
||||||
|
raw := field.Comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@bson:\\s*(.*)")
|
||||||
|
var value string
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
value = trim(match[0][1])
|
||||||
|
}
|
||||||
|
if value == "" {
|
||||||
|
value = CamelCaseToUnderscore(field.GoName)
|
||||||
|
}
|
||||||
|
if strings.EqualFold(value, "id") {
|
||||||
|
value = "_id"
|
||||||
|
}
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetRPCInfoList(services *protogen.Service) []*rpcInfo {
|
||||||
|
if len(services.Methods) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var list []*rpcInfo
|
||||||
|
for _, method := range services.Methods {
|
||||||
|
node := &rpcInfo{
|
||||||
|
FuncName: method.GoName,
|
||||||
|
RouterPath: GetCommentApiURL(method),
|
||||||
|
Method: GetCommentHttpMethod(method.Comments),
|
||||||
|
Author: GetCommentAuthor(method.Comments),
|
||||||
|
Describe: GetCommentDescribe(method.Comments),
|
||||||
|
ReqName: method.Input.GoIdent.GoName,
|
||||||
|
RespName: method.Output.GoIdent.GoName,
|
||||||
|
}
|
||||||
|
list = append(list, node)
|
||||||
|
}
|
||||||
|
return list
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsCommentModel(message *protogen.Message) bool {
|
||||||
|
raw := message.Comments.Leading.String()
|
||||||
|
split := strings.Split(raw, "\n")
|
||||||
|
re := regexp.MustCompile("@model:\\s*(true|false)")
|
||||||
|
for _, s := range split {
|
||||||
|
match := re.FindAllStringSubmatch(s, -1)
|
||||||
|
if len(match) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(match[0]) != 2 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if match[0][1] == "true" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(message.GoIdent.GoName, "Model") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|||||||
87
error_code_generate.go
Normal file
87
error_code_generate.go
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *Coco) generateErrorCode(plugin *protogen.Plugin) {
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
for _, e := range pbFile.Enums {
|
||||||
|
if e.Desc.Name() != "ErrCode" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
values, err := c.errcodeDefine(string(pbFile.GoPackageName), e)
|
||||||
|
if err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate error code failed: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
filename := fmt.Sprintf("%s/autogen_errcode_%s.go",
|
||||||
|
filepath.Dir(pbFile.GeneratedFilenamePrefix), pbFile.GoPackageName)
|
||||||
|
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
|
||||||
|
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
|
||||||
|
g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
|
||||||
|
g.P("// generate at: ", time.Now().Format(time.DateTime))
|
||||||
|
g.P()
|
||||||
|
g.P("package ", pbFile.GoPackageName)
|
||||||
|
g.P()
|
||||||
|
g.P(`import (
|
||||||
|
"gitter.top/coco/coco/core"
|
||||||
|
)`)
|
||||||
|
g.P()
|
||||||
|
g.P(values)
|
||||||
|
g.P()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) errcodeDefine(pkgName string, enum *protogen.Enum) (string, error) {
|
||||||
|
tpl, err := template.New("errcode_map").Parse(generateErrorCode)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("parse error code template failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
type T struct {
|
||||||
|
ErrorCode int32
|
||||||
|
ErrorMsg string
|
||||||
|
ErrorDefine string
|
||||||
|
}
|
||||||
|
var ts []T
|
||||||
|
for _, enumItem := range enum.Values {
|
||||||
|
ts = append(ts, T{
|
||||||
|
ErrorCode: int32(enumItem.Desc.Number()),
|
||||||
|
ErrorMsg: getCommentFromEnumItem(enumItem.Comments),
|
||||||
|
ErrorDefine: string(enumItem.Desc.Name()),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
type TT struct {
|
||||||
|
PkgName string
|
||||||
|
ErrorCodes []T
|
||||||
|
}
|
||||||
|
|
||||||
|
var tt = TT{PkgName: JavascriptCaseToCamelCase(pkgName), ErrorCodes: ts}
|
||||||
|
tpl.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = tpl.Execute(&buf, tt); err != nil {
|
||||||
|
return "", fmt.Errorf("execute error code template failed: %v", err)
|
||||||
|
}
|
||||||
|
return buf.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const generateErrorCode = `var (
|
||||||
|
{{.PkgName}}ErrcodeMap = map[int32]string{ {{range $errcode := .ErrorCodes}}
|
||||||
|
int32(ErrCode_{{$errcode.ErrorDefine}}): "{{$errcode.ErrorMsg}}",{{end}}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
{{range $errcode := .ErrorCodes}}{{$errcode.ErrorDefine}} = &core.ErrMsg{ErrCode: {{$errcode.ErrorCode}}, ErrMsg: "{{$errcode.ErrorMsg}}"}
|
||||||
|
{{end}}
|
||||||
|
)
|
||||||
|
`
|
||||||
36
go.mod
36
go.mod
@ -1,34 +1,12 @@
|
|||||||
module gitter.top/coco/protoc-gen-coco
|
module gitter.top/coco/protoc-gen-coco
|
||||||
|
|
||||||
go 1.19
|
go 1.21
|
||||||
|
|
||||||
|
toolchain go1.21.5
|
||||||
|
|
||||||
require (
|
require (
|
||||||
gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955
|
github.com/emicklei/proto v1.13.2
|
||||||
google.golang.org/protobuf v1.28.1
|
gitter.top/common/goast v0.0.1
|
||||||
)
|
gitter.top/common/protofmt v0.0.1
|
||||||
|
google.golang.org/protobuf v1.33.0
|
||||||
require (
|
|
||||||
github.com/bytedance/sonic v1.8.0 // indirect
|
|
||||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
|
|
||||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
|
||||||
github.com/gin-gonic/gin v1.9.0 // indirect
|
|
||||||
github.com/go-playground/locales v0.14.1 // indirect
|
|
||||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
|
||||||
github.com/go-playground/validator/v10 v10.11.2 // indirect
|
|
||||||
github.com/goccy/go-json v0.10.0 // indirect
|
|
||||||
github.com/json-iterator/go v1.1.12 // indirect
|
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9 // indirect
|
|
||||||
github.com/leodido/go-urn v1.2.1 // indirect
|
|
||||||
github.com/mattn/go-isatty v0.0.17 // indirect
|
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 // indirect
|
|
||||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
|
||||||
github.com/pelletier/go-toml/v2 v2.0.6 // indirect
|
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
|
||||||
github.com/ugorji/go/codec v1.2.9 // indirect
|
|
||||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670 // indirect
|
|
||||||
golang.org/x/crypto v0.5.0 // indirect
|
|
||||||
golang.org/x/net v0.7.0 // indirect
|
|
||||||
golang.org/x/sys v0.5.0 // indirect
|
|
||||||
golang.org/x/text v0.7.0 // indirect
|
|
||||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
|
||||||
)
|
)
|
||||||
|
|||||||
83
go.sum
83
go.sum
@ -1,83 +1,20 @@
|
|||||||
github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
|
|
||||||
github.com/bytedance/sonic v1.8.0 h1:ea0Xadu+sHlu7x5O3gKhRpQ1IKiMrSiHttPF0ybECuA=
|
|
||||||
github.com/bytedance/sonic v1.8.0/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
|
|
||||||
github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
|
|
||||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
|
|
||||||
github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
|
|
||||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
github.com/emicklei/proto v1.13.2 h1:z/etSFO3uyXeuEsVPzfl56WNgzcvIr42aQazXaQmFZY=
|
||||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
github.com/emicklei/proto v1.13.2/go.mod h1:rn1FgRS/FANiZdD2djyH7TMA9jdRDcYQ9IEN9yvjX0A=
|
||||||
github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8=
|
|
||||||
github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k=
|
|
||||||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
|
||||||
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
|
||||||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
|
||||||
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
|
||||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
|
||||||
github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyhcJ38oeKGACXohU=
|
|
||||||
github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s=
|
|
||||||
github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA=
|
|
||||||
github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
|
||||||
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
|
|
||||||
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||||
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
|
||||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
|
||||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4=
|
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
|
||||||
github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
|
|
||||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
|
||||||
github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w=
|
|
||||||
github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY=
|
|
||||||
github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng=
|
|
||||||
github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc=
|
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
|
||||||
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
|
||||||
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
|
||||||
github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU=
|
|
||||||
github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
|
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
gitter.top/common/goast v0.0.1 h1:YYpEOVXdlkORHAExgZSSUuT1MaebGE46EGcJLBxu/Sw=
|
||||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
gitter.top/common/goast v0.0.1/go.mod h1:+7nJ3ablVgA5+56qmTnIqrQhD9AjIN28YVFQYylDtm0=
|
||||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
gitter.top/common/protofmt v0.0.1 h1:aIRLb4vFKu1u3R7e/foijaNzls6N31F2yhnw2yUSXl0=
|
||||||
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
gitter.top/common/protofmt v0.0.1/go.mod h1:LrTenAqDqQQ74bBmpVwEOqZhmhcobtE79j0hQbeairs=
|
||||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
|
||||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
|
||||||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
|
||||||
github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
|
|
||||||
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
|
||||||
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
|
||||||
github.com/ugorji/go/codec v1.2.9 h1:rmenucSohSTiyL09Y+l2OCk+FrMxGMzho2+tjr5ticU=
|
|
||||||
github.com/ugorji/go/codec v1.2.9/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
|
||||||
gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955 h1:dLnS34ByD6KgTiUHBPQSqhvh1yk9pR10P3So/Gz9Zo0=
|
|
||||||
gitter.top/coco/coco v0.0.0-20230312134419-0dd59eb0e955/go.mod h1:IptVJiCXf87Xywfw26zrvXQjTjF8ZsIHUTJ8yPWSpWE=
|
|
||||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU=
|
|
||||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
|
||||||
golang.org/x/crypto v0.5.0 h1:U/0M97KRkSFvyD/3FSmdP5W5swImpNgle/EHFhOsQPE=
|
|
||||||
golang.org/x/crypto v0.5.0/go.mod h1:NK/OQwhpMQP3MwtdjgLlYHnH9ebylxKWv3e0fK+mkQU=
|
|
||||||
golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g=
|
|
||||||
golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
|
|
||||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU=
|
|
||||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo=
|
|
||||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
|
google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
|
||||||
google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
|
google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||||
google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
|
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
|
||||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
|
|
||||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
|
||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
|
||||||
|
|||||||
270
main.go
270
main.go
@ -1,44 +1,122 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"google.golang.org/protobuf/compiler/protogen"
|
|
||||||
"os"
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/emicklei/proto"
|
||||||
|
"gitter.top/common/protofmt"
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Coco struct{}
|
type Coco struct {
|
||||||
|
DisableGenerateRouter bool // 禁止路由信息生成
|
||||||
|
DisableGenerateMongoModel bool // 禁止mongo信息生成
|
||||||
|
DisableGenerateErrorCode bool // 禁止错误码信息生成
|
||||||
|
DisableGenerateRouterWire bool // 禁止wire路由信息生成
|
||||||
|
|
||||||
|
Prefix string // Prefix 当遇到proto是相对路径生成的时候 指定prefix
|
||||||
|
ProjectName string // 项目名称
|
||||||
|
}
|
||||||
|
|
||||||
// Generate generate coco router map
|
// Generate generate coco router map
|
||||||
func (c *Coco) Generate(plugin *protogen.Plugin) error {
|
func (c *Coco) Generate(plugin *protogen.Plugin) error {
|
||||||
if len(plugin.Files) == 0 {
|
if len(plugin.Files) == 0 {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c.Prefix != "" {
|
||||||
|
if !strings.HasSuffix(c.Prefix, "/") {
|
||||||
|
c.Prefix += "/"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
c.format(plugin)
|
||||||
|
|
||||||
|
if !c.DisableGenerateRouter {
|
||||||
|
c.generateRouterMap(plugin)
|
||||||
|
c.generateRouterImpl(plugin)
|
||||||
|
c.generateGrpcImpl(plugin)
|
||||||
|
c.generateRouterWire(plugin)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !c.DisableGenerateMongoModel {
|
||||||
|
c.generateMongoModel(plugin)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !c.DisableGenerateErrorCode {
|
||||||
|
c.generateErrorCode(plugin)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) format(plugin *protogen.Plugin) {
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
filename := pbFile.Desc.Path()
|
||||||
|
if c.Prefix != "" {
|
||||||
|
filename = c.Prefix + filename
|
||||||
|
}
|
||||||
|
file, err := os.Open(filename)
|
||||||
|
if err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "format pbFile %s failed: %v\n", filename, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
parser := proto.NewParser(file)
|
||||||
|
parser.Filename(filename)
|
||||||
|
def, err := parser.Parse()
|
||||||
|
if err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "parse pbFile %s failed: %v\n", filename, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
protofmt.NewFormatter(buf, " ").Format(def)
|
||||||
|
if err := os.WriteFile(filename, buf.Bytes(), os.ModePerm); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "rewrite pbFile %s failed: %v\n", filename, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) generateRouterMap(plugin *protogen.Plugin) {
|
||||||
for _, pbFile := range plugin.Files {
|
for _, pbFile := range plugin.Files {
|
||||||
// service empty
|
// service empty
|
||||||
if len(pbFile.Services) == 0 {
|
if len(pbFile.Services) == 0 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// write file header
|
|
||||||
filename := fmt.Sprintf("autogen_router_%s.go", pbFile.GeneratedFilenamePrefix)
|
|
||||||
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
|
|
||||||
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
|
|
||||||
g.P()
|
|
||||||
g.P("package ", pbFile.GoPackageName)
|
|
||||||
g.P()
|
|
||||||
g.P(`import (
|
|
||||||
"gitter.top/coco/coco/core"
|
|
||||||
)`)
|
|
||||||
g.P()
|
|
||||||
|
|
||||||
for _, service := range pbFile.Services {
|
for _, service := range pbFile.Services {
|
||||||
// service router group
|
// is service router group
|
||||||
if !IsCommentRouterGroup(pbFile.Services[0].Comments) {
|
if !IsCommentRouterGroup(pbFile.Services[0].Comments) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if len(service.Methods) == 0 {
|
if len(service.Methods) == 0 {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if GetCommentGenerateTo(service.Comments) == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// write file header
|
||||||
|
filename := fmt.Sprintf("%s/autogen_router_%s.go",
|
||||||
|
separator(filepath.Dir(pbFile.GeneratedFilenamePrefix)), CamelCaseToUnderscore(service.GoName))
|
||||||
|
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
|
||||||
|
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
|
||||||
|
g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
|
||||||
|
g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
|
||||||
|
g.P()
|
||||||
|
g.P("package ", pbFile.GoPackageName)
|
||||||
|
g.P()
|
||||||
|
g.P(`import (
|
||||||
|
"gitter.top/coco/coco"
|
||||||
|
)`)
|
||||||
|
g.P()
|
||||||
values, err := GenerateRouterMap(service)
|
values, err := GenerateRouterMap(service)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_, _ = fmt.Fprintf(os.Stderr, "generate router map failed: %v", err)
|
_, _ = fmt.Fprintf(os.Stderr, "generate router map failed: %v", err)
|
||||||
@ -47,12 +125,168 @@ func (c *Coco) Generate(plugin *protogen.Plugin) error {
|
|||||||
g.P(values)
|
g.P(values)
|
||||||
g.P()
|
g.P()
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
return nil
|
}
|
||||||
|
|
||||||
|
func (c *Coco) generateRouterImpl(plugin *protogen.Plugin) {
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
// service empty
|
||||||
|
if len(pbFile.Services) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, service := range pbFile.Services {
|
||||||
|
// is service router group
|
||||||
|
if !IsCommentRouterGroup(service.Comments) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
genTo := GetCommentGenerateTo(service.Comments)
|
||||||
|
if genTo == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// mkdir when Dir(genTo) not found
|
||||||
|
if err := mkdir(filepath.Dir(genTo)); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "mkdir genTo path failed: %s\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// new version, package name only support `controller`+`api_version`
|
||||||
|
// so old value `string(pbFile.GoPackageName)` Deprecated.
|
||||||
|
version := filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
|
||||||
|
generator := newRouterImpl(genTo, "controller"+version, string(pbFile.GoPackageName), service.GoName, service)
|
||||||
|
generator.ProjectName = c.ProjectName
|
||||||
|
if !generator.IsFileExist() {
|
||||||
|
if err := generator.generateNewFile(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate all router failed: %v\n", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := generator.generateNewRouters(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate router impl failed: %v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) generateGrpcImpl(plugin *protogen.Plugin) {
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
// service empty
|
||||||
|
if len(pbFile.Services) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, service := range pbFile.Services {
|
||||||
|
genTo := GetCommentRpcGenerateTo(service.Comments)
|
||||||
|
if genTo == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// mkdir when Dir(genTo) not found
|
||||||
|
if err := mkdir(filepath.Dir(genTo)); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "mkdir genTo path failed: %s\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
version := filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
|
||||||
|
generator := newRouterImpl(genTo, "rpc"+version, string(pbFile.GoPackageName), service.GoName, service)
|
||||||
|
generator.ProjectName = c.ProjectName
|
||||||
|
if !generator.IsFileExist() {
|
||||||
|
if err := generator.generateNewGrpcFile(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate all grpc failed: %v\n", err)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := generator.generateNewGrpcServers(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate rpc impl failed: %v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) generateMongoModel(plugin *protogen.Plugin) {
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
if len(pbFile.Messages) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var needMessages []*protogen.Message
|
||||||
|
for _, message := range pbFile.Messages {
|
||||||
|
// 非model
|
||||||
|
if !IsCommentModel(message) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
needMessages = append(needMessages, message)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(needMessages) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
filename := fmt.Sprintf("%s/autogen_model_%s.go",
|
||||||
|
separator(filepath.Dir(pbFile.GeneratedFilenamePrefix)), filepath.Base(pbFile.GeneratedFilenamePrefix))
|
||||||
|
g := plugin.NewGeneratedFile(filename, pbFile.GoImportPath)
|
||||||
|
g.P("// Code generated by protoc-gen-coco. DO NOT EDIT.")
|
||||||
|
g.P("// source: ", pbFile.GeneratedFilenamePrefix, ".proto")
|
||||||
|
g.P("// generate at: ", time.Now().Format("2006-01-02 15:04:05"))
|
||||||
|
g.P()
|
||||||
|
g.P("package ", pbFile.GoPackageName)
|
||||||
|
g.P()
|
||||||
|
for _, message := range needMessages {
|
||||||
|
value, err := GenerateModel(message, string(pbFile.GoPackageName))
|
||||||
|
if err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate model failed: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
g.P(value)
|
||||||
|
g.P()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Coco) generateRouterWire(plugin *protogen.Plugin) {
|
||||||
|
if c.DisableGenerateRouterWire {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, pbFile := range plugin.Files {
|
||||||
|
if len(pbFile.Services) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
for _, service := range pbFile.Services {
|
||||||
|
// is service router group
|
||||||
|
if !IsCommentRouterGroup(service.Comments) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := mkdir("gen/wire"); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "mkdir gen/wire path failed: %s\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
wireDir := "gen/wire/wire.go"
|
||||||
|
generator := newRouterImplWire(wireDir, "wire", string(pbFile.GoPackageName), service.GoName)
|
||||||
|
generator.Version = filepath.Base(filepath.Dir(pbFile.GeneratedFilenamePrefix))
|
||||||
|
generator.ProjectName = c.ProjectName
|
||||||
|
if !generator.IsFileExist() {
|
||||||
|
if err := generator.generateNewWireFile(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate router wire failed: %v\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := generator.generateServicesWire(); err != nil {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "generate router wire failed: %v\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
var coco = new(Coco)
|
var coco = new(Coco)
|
||||||
protogen.Options{}.Run(coco.Generate)
|
var flags flag.FlagSet
|
||||||
|
|
||||||
|
flags.BoolVar(&coco.DisableGenerateErrorCode, "disable_error_code", false, "disable generate error code")
|
||||||
|
flags.BoolVar(&coco.DisableGenerateMongoModel, "disable_mongodb_model", false, "disable generate mongodb model")
|
||||||
|
flags.BoolVar(&coco.DisableGenerateRouter, "disable_router", false, "disable generate router code")
|
||||||
|
flags.BoolVar(&coco.DisableGenerateRouterWire, "disable_router_wire", false, "disable generate router wire")
|
||||||
|
flags.StringVar(&coco.Prefix, "prefix", "", "proto prefix")
|
||||||
|
flags.StringVar(&coco.ProjectName, "project_name", "", "project name")
|
||||||
|
|
||||||
|
protogen.Options{
|
||||||
|
ParamFunc: flags.Set,
|
||||||
|
}.Run(coco.Generate)
|
||||||
}
|
}
|
||||||
|
|||||||
63
model_generate.go
Normal file
63
model_generate.go
Normal file
@ -0,0 +1,63 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
)
|
||||||
|
|
||||||
|
const ModelTpl = `const TableName{{.ModelName}} = "{{.TableName}}"
|
||||||
|
func (t *{{.ModelName}}) TableName() string {
|
||||||
|
return "{{.TableName}}"
|
||||||
|
}
|
||||||
|
|
||||||
|
{{range $structs := .DbFields}}
|
||||||
|
func (m *{{$.ModelName}}) Get{{$structs.FieldName}}Field() string {
|
||||||
|
return "{{$structs.BsonField}}"
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
`
|
||||||
|
|
||||||
|
type DbField struct {
|
||||||
|
FieldName string
|
||||||
|
BsonField string
|
||||||
|
}
|
||||||
|
|
||||||
|
type model struct {
|
||||||
|
ModelName string
|
||||||
|
TableName string
|
||||||
|
PackageName string
|
||||||
|
DbFields []*DbField
|
||||||
|
}
|
||||||
|
|
||||||
|
func GenerateModel(message *protogen.Message, packageName string) (string, error) {
|
||||||
|
var name = message.GoIdent.GoName
|
||||||
|
var entry = &model{
|
||||||
|
ModelName: name,
|
||||||
|
TableName: GetCommentTableName(message),
|
||||||
|
PackageName: packageName,
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, field := range message.Fields {
|
||||||
|
var dbField = &DbField{
|
||||||
|
FieldName: field.GoName,
|
||||||
|
BsonField: GetCommentBsonName(field),
|
||||||
|
}
|
||||||
|
entry.DbFields = append(entry.DbFields, dbField)
|
||||||
|
}
|
||||||
|
|
||||||
|
tpl, err := template.New("model_tpl").Parse(ModelTpl)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("parse %s template failed: %v", name, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tpl.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = tpl.Execute(&buf, entry); err != nil {
|
||||||
|
return "", fmt.Errorf("execute %s template failed: %v", name, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf.String(), nil
|
||||||
|
}
|
||||||
@ -2,9 +2,14 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"google.golang.org/protobuf/compiler/protogen"
|
"os"
|
||||||
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
|
"gitter.top/common/goast"
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
)
|
)
|
||||||
|
|
||||||
type generateRouterMethod struct {
|
type generateRouterMethod struct {
|
||||||
@ -25,12 +30,15 @@ type generateRouters struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const routerGenerateTpl = `type AutoGen{{.StructName}}Impl interface { ` + `{{range .Methods}}
|
const routerGenerateTpl = `type AutoGen{{.StructName}}Impl interface { ` + `{{range .Methods}}
|
||||||
{{.MethodName}}(ctx *core.Context, req *{{.InputName}}) (resp *{{.OutputName}}, err error){{end}}
|
{{.MethodName}}(ctx *coco.Context, req *{{.InputName}}) (resp *{{.OutputName}}, err error){{end}}
|
||||||
}
|
}
|
||||||
var (
|
|
||||||
AutoGen{{.StructName}} = &core.Routers{
|
type {{.StructName}}_RouterMap struct {}
|
||||||
BaseURL: "{{.BaseURL}}",
|
|
||||||
Apis: core.RouterMap{ {{range .Methods}}
|
func (receiver *{{.StructName}}_RouterMap) GetRouterMap() *coco.Routers {
|
||||||
|
return &coco.Routers {
|
||||||
|
BaseURL: "{{.BaseURL}}",
|
||||||
|
Apis: coco.RouterMap{ {{range .Methods}}
|
||||||
"{{.MethodName}}": {
|
"{{.MethodName}}": {
|
||||||
API: "{{.API}}",
|
API: "{{.API}}",
|
||||||
Method: "{{.Method}}",
|
Method: "{{.Method}}",
|
||||||
@ -38,8 +46,8 @@ var (
|
|||||||
Describe: "{{.Describe}}",
|
Describe: "{{.Describe}}",
|
||||||
},{{end}}
|
},{{end}}
|
||||||
},
|
},
|
||||||
|
}
|
||||||
}
|
}
|
||||||
)
|
|
||||||
`
|
`
|
||||||
|
|
||||||
func GenerateRouterMap(srv *protogen.Service) (string, error) {
|
func GenerateRouterMap(srv *protogen.Service) (string, error) {
|
||||||
@ -73,3 +81,425 @@ func GenerateRouterMap(srv *protogen.Service) (string, error) {
|
|||||||
|
|
||||||
return buf.String(), nil
|
return buf.String(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type rpcInfo struct {
|
||||||
|
FuncName string // 函数名
|
||||||
|
RouterPath string // 路由路径
|
||||||
|
Method string // 请求类型 POST/GET...
|
||||||
|
Author string // 接口作者
|
||||||
|
Describe string // 描述
|
||||||
|
ReqName string // request name
|
||||||
|
RespName string // response name
|
||||||
|
}
|
||||||
|
|
||||||
|
type routerImpl struct {
|
||||||
|
Filename string
|
||||||
|
PkgName string
|
||||||
|
SrvName string
|
||||||
|
GoPkgName string
|
||||||
|
ProjectName string
|
||||||
|
Version string
|
||||||
|
Routers []*rpcInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
// newRouterImpl 生成路由信息
|
||||||
|
// filename 文件生成地址
|
||||||
|
// pkgName 文件包名
|
||||||
|
// srvName 被控制的service名称
|
||||||
|
// service service信息
|
||||||
|
func newRouterImpl(filename, pkgName, goPkgName, srvName string, service *protogen.Service) *routerImpl {
|
||||||
|
if filename == "" {
|
||||||
|
filename = "./"
|
||||||
|
}
|
||||||
|
return &routerImpl{
|
||||||
|
Filename: filename,
|
||||||
|
SrvName: srvName,
|
||||||
|
PkgName: pkgName,
|
||||||
|
GoPkgName: goPkgName,
|
||||||
|
Routers: GetRPCInfoList(service),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func newRouterImplWire(filename, pkgName, goPkgName, srvName string) *routerImpl {
|
||||||
|
return &routerImpl{
|
||||||
|
Filename: filename,
|
||||||
|
SrvName: srvName,
|
||||||
|
PkgName: pkgName,
|
||||||
|
GoPkgName: goPkgName,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// fixPkgName 修正不规范的包名
|
||||||
|
func fixPkgName(name string) string {
|
||||||
|
name = strings.ReplaceAll(name, "-", "_")
|
||||||
|
name = strings.ReplaceAll(name, ".", "_")
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ri *routerImpl) IsFileExist() bool {
|
||||||
|
_, err := os.Stat(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, os.ErrNotExist) {
|
||||||
|
return false
|
||||||
|
} else {
|
||||||
|
_, _ = fmt.Fprintf(os.Stderr, "get file stat failed: %v", err)
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ri *routerImpl) GetFileInfo(filename string) (os.FileInfo, error) {
|
||||||
|
return os.Stat(filename)
|
||||||
|
}
|
||||||
|
|
||||||
|
const CompleteRouteGenerateAndPackageTpl = `package {{.PkgName}}
|
||||||
|
|
||||||
|
import (
|
||||||
|
"gitter.top/coco/coco"
|
||||||
|
|
||||||
|
{{$.GoPkgName}} "{{.ProjectName}}/gen/v1"
|
||||||
|
)
|
||||||
|
|
||||||
|
type {{.SrvName}} struct {}
|
||||||
|
|
||||||
|
// IDE: {{.SrvName}} implemented {{.GoPkgName}}.AutoGen{{.SrvName}}Impl interface
|
||||||
|
var _ {{.GoPkgName}}.AutoGen{{.SrvName}}Impl = (*{{.SrvName}})(nil)
|
||||||
|
|
||||||
|
{{range $route := .Routers}}
|
||||||
|
// {{$route.FuncName}} {{$route.Describe}}
|
||||||
|
func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx *coco.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
|
||||||
|
resp = new({{$.GoPkgName}}.{{$route.RespName}})
|
||||||
|
|
||||||
|
// TODO impl...
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
`
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateNewFile() error {
|
||||||
|
t, err := template.New("generate_new_file").Parse(CompleteRouteGenerateAndPackageTpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const CompleteGrpcGenerateAndPackageTpl = `package {{.PkgName}}
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
|
||||||
|
{{.GoPkgName}} "{{.ProjectName}}/gen/v1"
|
||||||
|
)
|
||||||
|
|
||||||
|
type {{.SrvName}} struct {}
|
||||||
|
|
||||||
|
// IDE: {{.SrvName}} implemented {{.GoPkgName}}.{{.SrvName}}Server interface
|
||||||
|
var _ {{.GoPkgName}}.{{.SrvName}}Server = (*{{.SrvName}})(nil)
|
||||||
|
|
||||||
|
{{range $route := .Routers}}
|
||||||
|
// {{$route.FuncName}} {{$route.Describe}}
|
||||||
|
func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx context.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
|
||||||
|
resp = new({{$.GoPkgName}}.{{$route.RespName}})
|
||||||
|
|
||||||
|
// TODO impl...
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
`
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateNewGrpcFile() error {
|
||||||
|
t, err := template.New("generate_new_grpc_file").Parse(CompleteGrpcGenerateAndPackageTpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateNewRouters() error {
|
||||||
|
info, err := ri.GetFileInfo(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
parser, err := goast.NewParser(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
definer, err := parser.Parse()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !definer.ExistType(ri.SrvName) {
|
||||||
|
tpl := `
|
||||||
|
type {{.SrvName}} struct {}
|
||||||
|
|
||||||
|
// IDE: {{.SrvName}} implemented {{.GoPkgName}}.AutoGen{{.SrvName}}Impl interface
|
||||||
|
var _ {{.GoPkgName}}.AutoGen{{.SrvName}}Impl = (*{{.SrvName}})(nil)
|
||||||
|
`
|
||||||
|
t, err := template.New("generate_type").Parse(tpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var needImpls []*rpcInfo
|
||||||
|
for _, router := range ri.Routers {
|
||||||
|
if !definer.ExistMethod(fmt.Sprintf("%s.%s", ri.SrvName, router.FuncName)) {
|
||||||
|
needImpls = append(needImpls, router)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(needImpls) != 0 {
|
||||||
|
tpl := `{{range $route := .Routers}}
|
||||||
|
// {{$route.FuncName}} {{$route.Describe}}
|
||||||
|
func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx *coco.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
|
||||||
|
resp = new({{$.GoPkgName}}.{{$route.RespName}})
|
||||||
|
|
||||||
|
// TODO impl...
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
||||||
|
{{end}}`
|
||||||
|
ri.Routers = needImpls
|
||||||
|
t, err := template.New("generate_impls").Parse(tpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateNewGrpcServers() error {
|
||||||
|
info, err := ri.GetFileInfo(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
parser, err := goast.NewParser(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
definer, err := parser.Parse()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !definer.ExistType(ri.SrvName) {
|
||||||
|
tpl := `
|
||||||
|
type {{.SrvName}} struct {}
|
||||||
|
|
||||||
|
// IDE: {{.SrvName}} implemented {{.GoPkgName}}.{{.SrvName}}Server interface
|
||||||
|
var _ {{.GoPkgName}}.{{.SrvName}}Server = (*{{.SrvName}})(nil)
|
||||||
|
`
|
||||||
|
t, err := template.New("generate_type").Parse(tpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var needImpls []*rpcInfo
|
||||||
|
for _, router := range ri.Routers {
|
||||||
|
if !definer.ExistMethod(fmt.Sprintf("%s.%s", ri.SrvName, router.FuncName)) {
|
||||||
|
needImpls = append(needImpls, router)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(needImpls) != 0 {
|
||||||
|
tpl := `{{range $route := .Routers}}
|
||||||
|
// {{$route.FuncName}} {{$route.Describe}}
|
||||||
|
func (receiver *{{$.SrvName}}) {{$route.FuncName}}(ctx context.Context, req *{{$.GoPkgName}}.{{$route.ReqName}}) (resp *{{$.GoPkgName}}.{{$route.RespName}}, err error) {
|
||||||
|
resp = new({{$.GoPkgName}}.{{$route.RespName}})
|
||||||
|
|
||||||
|
// TODO impl...
|
||||||
|
|
||||||
|
return resp, nil
|
||||||
|
}
|
||||||
|
{{end}}`
|
||||||
|
ri.Routers = needImpls
|
||||||
|
t, err := template.New("generate_impls").Parse(tpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const routeWireCompleteTpl = `//go:build wireinject
|
||||||
|
// +build wireinject
|
||||||
|
|
||||||
|
package wire
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/google/wire"
|
||||||
|
|
||||||
|
"{{.ProjectName}}/gen/{{.Version}}"
|
||||||
|
"{{.ProjectName}}/services/controller/{{.Version}}"
|
||||||
|
)
|
||||||
|
|
||||||
|
`
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateNewWireFile() error {
|
||||||
|
t, err := template.New("generate_new_wire").Parse(routeWireCompleteTpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const routeWireTpl = `type {{.SrvName}} struct {
|
||||||
|
*controller{{.Version}}.{{.SrvName}}
|
||||||
|
*gen{{.Version}}.{{.SrvName}}_RouterMap
|
||||||
|
}
|
||||||
|
|
||||||
|
func New{{.SrvName}}(c *controller{{.Version}}.{{.SrvName}}, r *gen{{.Version}}.{{.SrvName}}_RouterMap) *{{.SrvName}} {
|
||||||
|
return &{{.SrvName}}{c, r}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Init{{.SrvName}}() *{{.SrvName}} {
|
||||||
|
wire.Build(wire.Struct(new(controller{{.Version}}.{{.SrvName}})), wire.Struct(new(gen{{.Version}}.{{.SrvName}}_RouterMap)), New{{.SrvName}})
|
||||||
|
return &{{.SrvName}}{}
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
func (ri *routerImpl) generateServicesWire() error {
|
||||||
|
info, err := ri.GetFileInfo(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
parser, err := goast.NewParser(ri.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
definer, err := parser.Parse()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if definer.ExistType(ri.SrvName) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
t, err := template.New("generate_router_wire").Parse(routeWireTpl)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
t.DefinedTemplates()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err = t.Execute(&buf, ri); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
f, err := os.OpenFile(ri.Filename, os.O_APPEND|os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if _, err = f.Write(buf.Bytes()); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|||||||
41
test.proto
41
test.proto
@ -1,41 +0,0 @@
|
|||||||
syntax = "proto3";
|
|
||||||
|
|
||||||
package v1;
|
|
||||||
|
|
||||||
option go_package = "./;main";
|
|
||||||
|
|
||||||
// @route_group: true
|
|
||||||
// @base_url: /api/v1
|
|
||||||
service ExampleService { // tail
|
|
||||||
rpc ExampleCall1(ExampleMessage1) returns(ReturnType) {}
|
|
||||||
rpc ExampleCall2(ExampleMessage2) returns(ReturnType) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
// @route_group: true
|
|
||||||
// @base_url: /api/v2
|
|
||||||
service Example1Service { // tail
|
|
||||||
rpc ExampleCall1(ExampleMessage1) returns(ReturnType) {}
|
|
||||||
rpc ExampleCall2(ExampleMessage2) returns(ReturnType) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
// ExampleMessage1 - Example Leading Comment for ExampleMessage1
|
|
||||||
message ExampleMessage1 {
|
|
||||||
string MyString = 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
ExampleMessage2 - Example Leading Comment for ExampleMessage2
|
|
||||||
*/
|
|
||||||
message ExampleMessage2 {
|
|
||||||
int32 MyInt = 1;
|
|
||||||
// MyInt - Example trailing Comment
|
|
||||||
message ExampleNested {
|
|
||||||
bytes data = 1;
|
|
||||||
}
|
|
||||||
ExampleNested nested = 2;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
ReturnType - Empty Structure Placeholder
|
|
||||||
*/
|
|
||||||
message ReturnType {}
|
|
||||||
51
util.go
51
util.go
@ -1,6 +1,12 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import "unicode"
|
import (
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
|
"google.golang.org/protobuf/compiler/protogen"
|
||||||
|
)
|
||||||
|
|
||||||
// CamelCaseToUnderscore converts CamelCase to camel_case
|
// CamelCaseToUnderscore converts CamelCase to camel_case
|
||||||
func CamelCaseToUnderscore(str string) string {
|
func CamelCaseToUnderscore(str string) string {
|
||||||
@ -38,3 +44,46 @@ func CamelCaseToJavascriptCase(str string) string {
|
|||||||
}
|
}
|
||||||
return result
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// JavascriptCaseToCamelCase convert camelCase to CamelCase
|
||||||
|
func JavascriptCaseToCamelCase(str string) string {
|
||||||
|
var result string
|
||||||
|
for idx, ch := range str {
|
||||||
|
if idx == 0 {
|
||||||
|
result = string(unicode.ToUpper(ch))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
result += string(ch)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCommentFromEnumItem(comment protogen.CommentSet) string {
|
||||||
|
c := strings.Trim(string(comment.Trailing), " \n\t\r")
|
||||||
|
if c == "" {
|
||||||
|
c = "unknown error"
|
||||||
|
}
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
func separator(name string) string {
|
||||||
|
name = strings.ReplaceAll(name, "\\\\", "/")
|
||||||
|
return strings.ReplaceAll(name, "\\", "/")
|
||||||
|
}
|
||||||
|
|
||||||
|
func trim(s string) string {
|
||||||
|
return strings.Trim(s, "\r\t\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
func mkdir(p string) error {
|
||||||
|
_, err := os.Stat(p)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
err := os.MkdirAll(p, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user