luciano/mongo-driver-3
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for update pipelines

Browse source 
Resolves: #7
This commit is contained in:
Jacob Emcken 2022-10-02 17:47:50 +02:00
parent 34ef1f44b6
commit 989a6d87a3
No known key found for this signature in database
GPG key ID: F74CC648318D04D1
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/mongo_driver_3/collection.clj
View file

@ -228,7 +228,10 @@
- `db` is a MongoDatabase - `db` is a MongoDatabase
- `coll` is a collection name - `coll` is a collection name
- `q` is a map representing a query. - `q` is a map representing a query.
- `update` is a map representing an update. The update to apply must include only update operators. - `update` is either a map representing a document update or a vector
representing an 'aggregation pipeline'. A document update must include only
update operators, while an 'aggregation pipeline' can contain multiple
stages of `$set`, `$unset` and `$replaceWith`.
- `opts` (optional), a map of: - `opts` (optional), a map of:
- `:upsert?` whether to insert a new document if nothing is found, default: false - `:upsert?` whether to insert a new document if nothing is found, default: false
- `:return-new?` whether to return the document after update (insead of its state before the update), default: false - `:return-new?` whether to return the document after update (insead of its state before the update), default: false
@ -246,9 +249,14 @@
(let [{:keys [keywordize? ^ClientSession session] :or {keywordize? true}} opts (let [{:keys [keywordize? ^ClientSession session] :or {keywordize? true}} opts
^ClientSession session (or session *session*) ^ClientSession session (or session *session*)
opts' (->FindOneAndUpdateOptions opts)] opts' (->FindOneAndUpdateOptions opts)]
(-> (if session (-> (if (instance? List update)
(let [pipeline ^List (map document update)]
(if session
(.findOneAndUpdate (collection db coll opts) session (document q) pipeline opts')
(.findOneAndUpdate (collection db coll opts) (document q) pipeline opts')))
(if session
(.findOneAndUpdate (collection db coll opts) session (document q) (document update) opts') (.findOneAndUpdate (collection db coll opts) session (document q) (document update) opts')
(.findOneAndUpdate (collection db coll opts) (document q) (document update) opts')) (.findOneAndUpdate (collection db coll opts) (document q) (document update) opts')))
(from-document keywordize?))))) (from-document keywordize?)))))
(defn find-one-and-replace (defn find-one-and-replace