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

mongodb json tag

Browse Source
This commit is contained in:
zhuyasen
2024-03-05 00:39:41 +08:00
parent 20bd7c060f
commit 5734b2d4be
11 changed files with 33 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -64,16 +64,16 @@ var mgoTypeToGo = map[bsontype.Type]string{
bson.TypeJavaScript: goTypeString,
}
var jsonTagFormat int32 // 0: camel case, 1: snake case
var jsonTagFormat int32 = 1 // 0: snake case, 1: camel case
// SetJSONTagSnakeCase set json tag format to snake case
func SetJSONTagSnakeCase() {
atomic.AddInt32(&jsonTagFormat, 1)
atomic.AddInt32(&jsonTagFormat, -jsonTagFormat)
}
// SetJSONTagCamelCase set json tag format to camel case
func SetJSONTagCamelCase() {
atomic.AddInt32(&jsonTagFormat, -jsonTagFormat)
atomic.AddInt32(&jsonTagFormat, 1)
}
// MgoField mongo field
@@ -172,9 +172,9 @@ func parseObject(name string, elements []bson.RawElement) (goTypeStr string, goO
var jsonTag string
if jsonTagFormat == 0 {
jsonTag = toLowerFirst(xstrings.ToCamelCase(k))
} else {
jsonTag = xstrings.ToSnakeCase(k)
} else {
jsonTag = toLowerFirst(xstrings.ToCamelCase(k))
}
goObjStr += fmt.Sprintf(" %s %s `bson:\"%s\" json:\"%s\"`\n", xstrings.ToCamelCase(k), t, k, jsonTag)