Use primitive import instead of bson alias

This commit is contained in:
2020-11-18 11:06:26 +01:00
parent f859828370
commit 0b798fe818
5 changed files with 22 additions and 24 deletions
+2 -3
View File
@@ -4,7 +4,6 @@ import (
"fmt"
"github.com/modern-go/reflect2"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"github.com/mainnika/mongox-go-driver/v2/mongox/base/protection"
@@ -44,10 +43,10 @@ func Push(q *Query, f interface{}) (ok bool) {
return ok
}
// applyBson is a fallback for a custom bson.M
// applyBson is a fallback for a custom primitive.M
func applyBson(q *Query, f interface{}) (ok bool) {
if f, ok := f.(bson.M); ok {
if f, ok := f.(primitive.M); ok {
q.And(f)
return true
}
+9 -9
View File
@@ -1,12 +1,12 @@
package query
import (
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
// Query is an enchanched bson.M map
// Query is an enchanched primitive.M map
type Query struct {
m bson.M
m primitive.M
limiter Limiter
sorter Sorter
skipper Skipper
@@ -17,16 +17,16 @@ type Query struct {
}
// And function pushes the elem query to the $and array of the query
func (q *Query) And(elem bson.M) (query *Query) {
func (q *Query) And(elem primitive.M) (query *Query) {
if q.m == nil {
q.m = bson.M{}
q.m = primitive.M{}
}
queries, exists := q.m["$and"].(bson.A)
queries, exists := q.m["$and"].(primitive.A)
if !exists {
q.m["$and"] = bson.A{elem}
q.m["$and"] = primitive.A{elem}
return q
}
@@ -66,7 +66,7 @@ func (q *Query) Skipper() (skip *int64) {
}
// Updater is an update command for a query
func (q *Query) Updater() (update bson.A) {
func (q *Query) Updater() (update primitive.A) {
if q.updater == nil {
return
@@ -103,6 +103,6 @@ func (q *Query) Empty() (isEmpty bool) {
}
// M returns underlying query map
func (q *Query) M() (m bson.M) {
func (q *Query) M() (m primitive.M) {
return q.m
}
+5 -5
View File
@@ -1,20 +1,20 @@
package query
import (
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
// Sorter is a filter to sort the data before query
type Sorter interface {
Sort() (sort bson.M)
Sort() (sort primitive.M)
}
// Sort is a simple implementations of the Sorter filter
type Sort bson.M
type Sort primitive.M
var _ Sorter = &Sort{}
// Sort returns a slice of fields which have to be sorted
func (f Sort) Sort() (sort bson.M) {
return bson.M(f)
func (f Sort) Sort() (sort primitive.M) {
return primitive.M(f)
}
+5 -5
View File
@@ -1,20 +1,20 @@
package query
import (
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
// Updater is a filter to update the data
type Updater interface {
Update() (update bson.A)
Update() (update primitive.A)
}
// Update is a simple implementations of the Updater filter
type Update bson.M
type Update primitive.M
var _ Updater = &Update{}
// Update returns an update command
func (u Update) Update() (update bson.A) {
return bson.A{bson.M(u)}
func (u Update) Update() (update primitive.A) {
return primitive.A{primitive.M(u)}
}