mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-07-03 17:22:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add filters support for an array deleting

Browse Source
This commit is contained in:
Nikita Tokarchuk
2020-07-23 01:43:39 +02:00
parent ee4bf46662
commit 08397650c6
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/database/deletearray.go
+6 -2
View File
@@ -8,10 +8,11 @@ import (
"go.mongodb.org/mongo-driver/mongo/options" "go.mongodb.org/mongo-driver/mongo/options"
"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"
) )
// DeleteArray removes documents list from a database by their ids // DeleteArray removes documents list from a database by their ids
func (d *Database) DeleteArray(target interface{}) (err error) { func (d *Database) DeleteArray(target interface{}, filters ...interface{}) (err error) {
targetV := reflect.ValueOf(target) targetV := reflect.ValueOf(target)
targetT := targetV.Type() targetT := targetV.Type()
@@ -37,6 +38,7 @@ func (d *Database) DeleteArray(target interface{}) (err error) {
collection := d.GetCollectionOf(zeroElem.Interface()) collection := d.GetCollectionOf(zeroElem.Interface())
opts := options.Delete() opts := options.Delete()
ids := primitive.A{} ids := primitive.A{}
composed := query.Compose(filters...)
for i := 0; i < targetLen; i++ { for i := 0; i < targetLen; i++ {
elem := targetSliceV.Index(i) elem := targetSliceV.Index(i)
@@ -47,7 +49,9 @@ func (d *Database) DeleteArray(target interface{}) (err error) {
return fmt.Errorf("can't delete zero elements") return fmt.Errorf("can't delete zero elements")
} }
result, err := collection.DeleteMany(d.Context(), primitive.M{"_id": primitive.M{"$in": ids}}, opts) composed.And(primitive.M{"_id": primitive.M{"$in": ids}})
result, err := collection.DeleteMany(d.Context(), composed.M(), opts)
if err != nil { if err != nil {
return return
} }
mongox/mongox.go
+1 -1
View File
@@ -22,7 +22,7 @@ type Database interface {
New(ctx context.Context) (db Database) New(ctx context.Context) (db Database)
GetCollectionOf(document interface{}) (collection *Collection) GetCollectionOf(document interface{}) (collection *Collection)
Count(target interface{}, filters ...interface{}) (count int64, err error) Count(target interface{}, filters ...interface{}) (count int64, err error)
DeleteArray(target interface{}) (err error) DeleteArray(target interface{}, filters ...interface{}) (err error)
DeleteOne(target interface{}, filters ...interface{}) (err error) DeleteOne(target interface{}, filters ...interface{}) (err error)
LoadArray(target interface{}, filters ...interface{}) (err error) LoadArray(target interface{}, filters ...interface{}) (err error)
LoadOne(target interface{}, filters ...interface{}) (err error) LoadOne(target interface{}, filters ...interface{}) (err error)