sumu
/
sponge
mirror of https://github.com/zhufuyi/sponge
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #108 from soluty/feat/mongoisnull 

add is null and is not null feature like gorm in mongodb
This commit is contained in:
zhuyasen 2025-06-23 19:20:52 +08:00 committed by GitHub
parent 32b1142d38 d2f4a82b16
commit 8487101398
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
pkg/mgo/query/query_condition.go
View File

@ -37,6 +37,10 @@ const (
In = "in"
// NotIn exclude
NotIn = "nin"
// IsNull is null
IsNull = "isnull"
// IsNotNull is not null
IsNotNull = "isnotnull"
// AND logic and
AND string = "and" //nolint
@ -49,23 +53,27 @@ const (
)
var expMap = map[string]string{
Eq: eqSymbol,
eqSymbol: eqSymbol,
Neq: neqSymbol,
neqSymbol: neqSymbol,
Gt: gtSymbol,
gtSymbol: gtSymbol,
Gte: gteSymbol,
gteSymbol: gteSymbol,
Lt: ltSymbol,
ltSymbol: ltSymbol,
Lte: lteSymbol,
lteSymbol: lteSymbol,
Like: Like,
In: In,
NotIn: NotIn,
"notin": NotIn,
"not in": NotIn,
Eq: eqSymbol,
eqSymbol: eqSymbol,
Neq: neqSymbol,
neqSymbol: neqSymbol,
Gt: gtSymbol,
gtSymbol: gtSymbol,
Gte: gteSymbol,
gteSymbol: gteSymbol,
Lt: ltSymbol,
ltSymbol: ltSymbol,
Lte: lteSymbol,
lteSymbol: lteSymbol,
Like: Like,
In: In,
NotIn: NotIn,
"notin": NotIn,
"not in": NotIn,
IsNull: IsNull,
IsNotNull: IsNotNull,
"is null": IsNull,
"is not null": IsNotNull,
}
var logicMap = map[string]string{
@ -207,7 +215,7 @@ func (c *Column) convertValue() error {
if v, ok := expMap[strings.ToLower(c.Exp)]; ok { //nolint
c.Exp = v
switch c.Exp {
//case eqSymbol:
// case eqSymbol:
case neqSymbol:
c.Value = bson.M{"$ne": c.Value}
case gtSymbol:
@ -218,6 +226,10 @@ func (c *Column) convertValue() error {
c.Value = bson.M{"$lt": c.Value}
case lteSymbol:
c.Value = bson.M{"$lte": c.Value}
case IsNull:
c.Value = bson.M{"$exist": false}
case IsNotNull:
c.Value = bson.M{"$exist": true}
case Like:
escapedValue := regexp.QuoteMeta(fmt.Sprintf("%v", c.Value))
c.Value = bson.M{"$regex": escapedValue, "$options": "i"}
@ -312,11 +324,13 @@ func (p *Params) ConvertToMongoFilter(opts ...RulerOption) (bson.M, error) {
if p.Columns[0].Logic == AND {
filter = bson.M{"$and": []bson.M{
{p.Columns[0].Name: p.Columns[0].Value},
{p.Columns[1].Name: p.Columns[1].Value}}}
{p.Columns[1].Name: p.Columns[1].Value},
}}
} else {
filter = bson.M{"$or": []bson.M{
{p.Columns[0].Name: p.Columns[0].Value},
{p.Columns[1].Name: p.Columns[1].Value}}}
{p.Columns[1].Name: p.Columns[1].Value},
}}
}
return filter, nil