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

extended api

Browse Source
This commit is contained in:
zhuyasen
2024-06-28 23:56:08 +08:00
parent 5888ec77d2
commit 994a90850e
52 changed files with 7909 additions and 1084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -499,7 +499,7 @@ func makeCode(stmt *ast.CreateTableStmt, opt options) (*codeText, error) {
return nil, err
}
protoFileCode, err := getProtoFileCode(data, opt.IsWebProto)
protoFileCode, err := getProtoFileCode(data, opt.IsWebProto, opt.IsExtendedApi)
if err != nil {
return nil, err
}
@@ -751,17 +751,26 @@ func getModelJSONCode(data tmplData) (string, error) {
return modelJSONCode, nil
}
func getProtoFileCode(data tmplData, isWebProto bool) (string, error) {
func getProtoFileCode(data tmplData, isWebProto bool, isExtendedApi bool) (string, error) {
data.Fields = goTypeToProto(data.Fields)
var err error
builder := strings.Builder{}
if isWebProto {
err := protoFileForWebTmpl.Execute(&builder, data)
if isExtendedApi {
err = protoFileForWebTmpl.Execute(&builder, data)
} else {
err = protoFileForSimpleWebTmpl.Execute(&builder, data)
}
if err != nil {
return "", err
}
} else {
err := protoFileTmpl.Execute(&builder, data)
if isExtendedApi {
err = protoFileTmpl.Execute(&builder, data)
} else {
err = protoFileSimpleTmpl.Execute(&builder, data)
}
if err != nil {
return "", err
}