Implement UpdateOne query support

mongox/query/compose.go

@@ -38,6 +38,7 @@ func Push(q *Query, f interface{}) (ok bool) {
 	ok = ok || applySkip(q, f)
 	ok = ok || applyProtection(q, f)
 	ok = ok || applyPreloader(q, f)
+	ok = ok || applyUpdater(q, f)
 	ok = ok || applyCallbacks(q, f)
 
 	return ok
@@ -125,6 +126,16 @@ func applyPreloader(q *Query, f interface{}) (ok bool) {
 	return false
 }
 
+func applyUpdater(q *Query, f interface{}) (ok bool) {
+
+	if f, ok := f.(Updater); ok {
+		q.updater = f
+		return true
+	}
+
+	return false
+}
+
 func applyCallbacks(q *Query, f interface{}) (ok bool) {
 
 	switch cb := f.(type) {

mongox/query/query.go

@@ -11,6 +11,7 @@ type Query struct {
 	sorter    Sorter
 	skipper   Skipper
 	preloader Preloader
+	updater   Updater
 	ondecode  Callbacks
 	onclose   Callbacks
 }
@@ -64,6 +65,16 @@ func (q *Query) Skipper() (skip *int64) {
 	return q.skipper.Skip()
 }
 
+// Updater is an update command for a query
+func (q *Query) Updater() (update bson.A) {
+
+	if q.updater == nil {
+		return
+	}
+
+	return q.updater.Update()
+}
+
 // Preloader is a preloader list for a query
 func (q *Query) Preloader() (ok bool, preloads []string) {
 

mongox/query/update.go

@@ -0,0 +1,20 @@
+package query
+
+import (
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+// Updater is a filter to update the data
+type Updater interface {
+	Update() (update bson.A)
+}
+
+// Update is a simple implementations of the Updater filter
+type Update bson.M
+
+var _ Updater = &Update{}
+
+// Update returns an update command
+func (u Update) Update() (update bson.A) {
+	return bson.A{bson.M(u)}
+}