zishuo/sponge
1
0
Fork 0
mirror of https://github.com/zhufuyi/sponge.git synced 2025-10-04 16:32:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

modify the generate proto file command

Browse Source
This commit is contained in:
zhuyasen
2023-08-18 23:27:06 +08:00
parent 6075364a04
commit b4fdb52390
8 changed files with 147 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/sql2code/parser/parser.go
View File

@@ -380,7 +380,7 @@ func makeCode(stmt *ast.CreateTableStmt, opt options) (*codeText, error) {
return nil, err
}
protoFileCode, err := getProtoFileCode(data)
protoFileCode, err := getProtoFileCode(data, opt.IsWebProto)
if err != nil {
return nil, err
}
@@ -565,13 +565,20 @@ func getModelJSONCode(data tmplData) (string, error) {
return modelJSONCode, nil
}
func getProtoFileCode(data tmplData) (string, error) {
func getProtoFileCode(data tmplData, isWebProto bool) (string, error) {
data.Fields = goTypeToProto(data.Fields)
builder := strings.Builder{}
err := protoFileTmpl.Execute(&builder, data)
if err != nil {
return "", err
if isWebProto {
err := protoFileForWebTmpl.Execute(&builder, data)
if err != nil {
return "", err
}
} else {
err := protoFileTmpl.Execute(&builder, data)
if err != nil {
return "", err
}
}
code := builder.String()
@@ -584,6 +591,9 @@ func getProtoFileCode(data tmplData) (string, error) {
if err != nil {
return "", fmt.Errorf("handlerCreateStructTmpl error: %v", err)
}
if !isWebProto {
protoMessageUpdateCode = strings.ReplaceAll(protoMessageUpdateCode, ` [(tagger.tags) = "uri:\"id\"" ]`, "")
}
protoMessageDetailCode, err := tmplExecuteWithFilter(data, protoMessageDetailTmpl, columnID, columnCreatedAt, columnUpdatedAt)
if err != nil {