package query import ( "github.com/modern-go/reflect2" "github.com/valyala/bytebufferpool" "go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson/primitive" ) // Query is an enchanched primitive.M map type Query struct { m primitive.M limiter Limiter sorter Sorter skipper Skipper preloader Preloader updater Updater ondecode Callbacks onclose Callbacks oncreate Callbacks } // And function pushes the elem query to the $and array of the query func (q *Query) And(elem primitive.M) (query *Query) { if q.m == nil { q.m = primitive.M{} } queries, exists := q.m["$and"].(primitive.A) if !exists { q.m["$and"] = primitive.A{elem} return q } q.m["$and"] = append(queries, elem) return q } // Limiter returns limiter value or nil func (q *Query) Limiter() (limit *int64) { if q.limiter == nil { return } return q.limiter.Limit() } // Sorter is a sort rule for a query func (q *Query) Sorter() (sort interface{}) { if q.sorter == nil { return } return q.sorter.Sort() } // Skipper is a skipper for a query func (q *Query) Skipper() (skip *int64) { if q.skipper == nil { return } return q.skipper.Skip() } // Updater is an update command for a query func (q *Query) Updater() (update primitive.M, err error) { if q.updater == nil { update = primitive.M{} return } update = q.updater.Update() if reflect2.IsNil(update) { update = primitive.M{} return } buffer := bytebufferpool.Get() defer bytebufferpool.Put(buffer) // convert update document to bson map values bsonBytes, err := bson.MarshalAppend(buffer.B, update) if err != nil { return } update = primitive.M{} err = bson.Unmarshal(bsonBytes, update) if err != nil { return } return } // Preloader is a preloader list for a query func (q *Query) Preloader() (preloads []string, ok bool) { if q.preloader == nil { return nil, false } preloads = q.preloader.Preload() ok = len(preloads) > 0 return } // OnDecode callback is called after the mongo decode function func (q *Query) OnDecode() (callbacks Callbacks) { return q.ondecode } func (q *Query) OnClose() (callbacks Callbacks) { return q.onclose } // OnCreate callback is called if the mongox creates a new document instance during loading func (q *Query) OnCreate() (callbacks Callbacks) { return q.onclose } // Empty checks the query for any content func (q *Query) Empty() (isEmpty bool) { return len(q.m) == 0 } // M returns underlying query map func (q *Query) M() (m primitive.M) { return q.m }