mirror of
https://github.com/mainnika/mongox-go-driver.git
synced 2026-05-23 00:03:36 +00:00
Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| fd53c66690 | |||
| 6111341a3c | |||
| 9f647ca094 |
@@ -6,6 +6,7 @@ import (
|
|||||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
|
|
||||||
"github.com/mainnika/mongox-go-driver/v2/mongox"
|
"github.com/mainnika/mongox-go-driver/v2/mongox"
|
||||||
|
"github.com/mainnika/mongox-go-driver/v2/mongox/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
// GetID returns source document id
|
// GetID returns source document id
|
||||||
@@ -62,7 +63,7 @@ func getObjectOrPanic(source mongox.JSONBased) (id primitive.D) {
|
|||||||
func getInterfaceOrPanic(source mongox.InterfaceBased) (id interface{}) {
|
func getInterfaceOrPanic(source mongox.InterfaceBased) (id interface{}) {
|
||||||
|
|
||||||
id = source.GetID()
|
id = source.GetID()
|
||||||
if id != nil {
|
if !utils.IsNil(id) {
|
||||||
return id
|
return id
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ import (
|
|||||||
"github.com/mainnika/mongox-go-driver/v2/mongox"
|
"github.com/mainnika/mongox-go-driver/v2/mongox"
|
||||||
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
|
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
|
||||||
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
|
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
|
||||||
|
"github.com/mainnika/mongox-go-driver/v2/mongox/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
// DeleteOne removes a document from a database and then returns it into target
|
// DeleteOne removes a document from a database and then returns it into target
|
||||||
@@ -22,7 +23,7 @@ func (d *Database) DeleteOne(target interface{}, filters ...interface{}) error {
|
|||||||
|
|
||||||
opts.Sort = composed.Sorter()
|
opts.Sort = composed.Sorter()
|
||||||
|
|
||||||
if target != nil {
|
if !utils.IsNil(target) {
|
||||||
composed.And(primitive.M{"_id": base.GetID(target)})
|
composed.And(primitive.M{"_id": base.GetID(target)})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -64,15 +64,15 @@ func (d *Database) IndexEnsure(cfg interface{}, document interface{}) error {
|
|||||||
panic(fmt.Errorf("cannot evaluate index key"))
|
panic(fmt.Errorf("cannot evaluate index key"))
|
||||||
}
|
}
|
||||||
|
|
||||||
index := primitive.M{key: 1}
|
|
||||||
opts := &options.IndexOptions{
|
opts := &options.IndexOptions{
|
||||||
Background: &f,
|
Background: &f,
|
||||||
Unique: &f,
|
Unique: &f,
|
||||||
Name: &name,
|
Name: &name,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
index := primitive.D{{Key: key, Value: 1}}
|
||||||
if indexValues[0] == "-" {
|
if indexValues[0] == "-" {
|
||||||
index[key] = -1
|
index = primitive.D{{Key: key, Value: -1}}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, prop := range indexValues[1:] {
|
for _, prop := range indexValues[1:] {
|
||||||
@@ -114,9 +114,9 @@ func (d *Database) IndexEnsure(cfg interface{}, document interface{}) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if compoundValue[0] == '-' {
|
if compoundValue[0] == '-' {
|
||||||
index[compoundValue[1:]] = -1
|
index = append(index, primitive.E{compoundValue[1:], -1})
|
||||||
} else {
|
} else {
|
||||||
index[compoundValue] = 1
|
index = append(index, primitive.E{compoundValue, 1})
|
||||||
}
|
}
|
||||||
|
|
||||||
default:
|
default:
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import (
|
|||||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
|
|
||||||
"github.com/mainnika/mongox-go-driver/v2/mongox/base/protection"
|
"github.com/mainnika/mongox-go-driver/v2/mongox/base/protection"
|
||||||
|
"github.com/mainnika/mongox-go-driver/v2/mongox/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Compose is a function to compose filters into a single query
|
// Compose is a function to compose filters into a single query
|
||||||
@@ -26,6 +27,10 @@ func Compose(filters ...interface{}) *Query {
|
|||||||
// Push applies single filter to a query
|
// Push applies single filter to a query
|
||||||
func Push(q *Query, f interface{}) bool {
|
func Push(q *Query, f interface{}) bool {
|
||||||
|
|
||||||
|
if utils.IsNil(f) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
ok := false
|
ok := false
|
||||||
ok = ok || applyBson(q, f)
|
ok = ok || applyBson(q, f)
|
||||||
ok = ok || applyLimit(q, f)
|
ok = ok || applyLimit(q, f)
|
||||||
@@ -91,9 +96,6 @@ func applyProtection(q *Query, f interface{}) bool {
|
|||||||
x = &f.X
|
x = &f.X
|
||||||
v = &f.V
|
v = &f.V
|
||||||
case *protection.Key:
|
case *protection.Key:
|
||||||
if f == nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
x = &f.X
|
x = &f.X
|
||||||
v = &f.V
|
v = &f.V
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,21 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
// IsNil function evaluates the interface value to nil
|
||||||
|
func IsNil(i interface{}) bool {
|
||||||
|
|
||||||
|
type iface struct {
|
||||||
|
_ *interface{}
|
||||||
|
ptr unsafe.Pointer
|
||||||
|
}
|
||||||
|
|
||||||
|
unpacked := (*iface)(unsafe.Pointer(&i))
|
||||||
|
if unpacked.ptr == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return *(*unsafe.Pointer)(unpacked.ptr) == nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,32 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestIsNil(t *testing.T) {
|
||||||
|
|
||||||
|
testvalues := []struct {
|
||||||
|
i interface{}
|
||||||
|
isnil bool
|
||||||
|
}{
|
||||||
|
{nil, true},
|
||||||
|
{(*string)(nil), true},
|
||||||
|
{([]string)(nil), true},
|
||||||
|
{(map[string]string)(nil), true},
|
||||||
|
{(func() bool)(nil), true},
|
||||||
|
{(chan func() bool)(nil), true},
|
||||||
|
{"", true},
|
||||||
|
{0, true},
|
||||||
|
{append(([]string)(nil), ""), false},
|
||||||
|
{[]string{}, false},
|
||||||
|
{1, false},
|
||||||
|
{"1", false},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range testvalues {
|
||||||
|
assert.Equal(t, tt.isnil, IsNil(tt.i))
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user