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

fix mongodb parse bug

Browse Source
This commit is contained in:
zhuyasen
2024-05-11 13:37:30 +08:00
parent b63c1addc4
commit 7761fcd7b7
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/sql2code/parser/mongodb.go
View File

@@ -204,10 +204,10 @@ func toLowerFirst(str string) string {
func embedTimeField(names []string, fields []*MgoField) []*MgoField {
isHaveCreatedAt, isHaveUpdatedAt := false, false
for _, name := range names {
if name == "created_at" {
if name == "created_at" || name == "createdAt" {
isHaveCreatedAt = true
}
if name == "updated_at" {
if name == "updated_at" || name == "updatedAt" {
isHaveUpdatedAt = true
}
names = append(names, name)

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

@@ -972,7 +972,7 @@ func goTypeToProto(fields []tmplField) []tmplField {
field.GoType = "repeated string"
}
if field.DBDriver == DBDriverMongodb {
if field.DBDriver == DBDriverMongodb && field.GoType != "" {
if field.GoType[0] == '*' {
field.GoType = field.GoType[1:]
} else if strings.Contains(field.GoType, "[]*") {