mongo-driver-3/src/mongo_driver_3/collection.clj

666 lines
26 KiB
Clojure
Raw Normal View History

2019-11-14 02:01:38 +00:00
(ns mongo-driver-3.collection
(:refer-clojure :exclude [find empty? drop])
(:import (clojure.lang Ratio Keyword Named IPersistentMap)
(com.mongodb ReadConcern ReadPreference WriteConcern MongoNamespace)
2019-11-22 09:55:03 +00:00
(com.mongodb.client MongoDatabase MongoCollection ClientSession)
2019-11-14 02:01:38 +00:00
(com.mongodb.client.model InsertOneOptions InsertManyOptions DeleteOptions FindOneAndUpdateOptions ReturnDocument FindOneAndReplaceOptions CountOptions CreateCollectionOptions RenameCollectionOptions IndexOptions IndexModel UpdateOptions ReplaceOptions)
(java.util List Collection)
(java.util.concurrent TimeUnit)
(org.bson Document)
(org.bson.types Decimal128)))
2019-11-22 09:55:03 +00:00
(set! *warn-on-reflection* true)
2019-11-14 02:01:38 +00:00
;;; Conversions
(defprotocol ConvertToDocument
2019-11-17 06:00:30 +00:00
(^Document document [input] "Convert from clojure to Mongo Document"))
2019-11-14 02:01:38 +00:00
(extend-protocol ConvertToDocument
nil
(document [_]
nil)
Ratio
(document [^Ratio input]
(double input))
Keyword
(document [^Keyword input]
(.getName input))
Named
(document [^Named input]
(.getName input))
IPersistentMap
(document [^IPersistentMap input]
(let [o (Document.)]
(doseq [[k v] input]
(.append o (document k) (document v)))
o))
Collection
(document [^Collection input]
(map document input))
Object
(document [input]
input))
(defprotocol ConvertFromDocument
2019-11-17 06:00:30 +00:00
(from-document [input keywordize?] "Converts Mongo Document to Clojure"))
2019-11-14 02:01:38 +00:00
(extend-protocol ConvertFromDocument
nil
(from-document [input _]
input)
Object
(from-document [input _] input)
Decimal128
(from-document [^Decimal128 input _]
(.bigDecimalValue input))
List
2019-11-14 12:09:01 +00:00
(from-document [^List input keywordize?]
(vec (map #(from-document % keywordize?) input)))
2019-11-14 02:01:38 +00:00
Document
2019-11-14 12:09:01 +00:00
(from-document [^Document input keywordize?]
(reduce (if keywordize?
2019-11-14 02:01:38 +00:00
(fn [m ^String k]
(assoc m (keyword k) (from-document (.get input k) true)))
(fn [m ^String k]
(assoc m k (from-document (.get input k) false))))
{} (.keySet input))))
;;; Collection
(def kw->ReadConcern
{:available (ReadConcern/AVAILABLE)
:default (ReadConcern/DEFAULT)
:linearizable (ReadConcern/LINEARIZABLE)
:local (ReadConcern/LOCAL)
:majority (ReadConcern/MAJORITY)
:snapshot (ReadConcern/SNAPSHOT)})
(defn ->ReadConcern
2019-11-17 06:00:30 +00:00
"Coerce `rc` into a ReadConcern if not nil. See `collection` for usage."
2019-11-19 08:01:21 +00:00
[{:keys [read-concern]}]
(when read-concern
(if (instance? ReadConcern read-concern)
read-concern
(or (kw->ReadConcern read-concern) (throw (IllegalArgumentException.
(str "No match for read concern of " (name read-concern))))))))
2019-11-14 02:01:38 +00:00
(defn ->ReadPreference
2019-11-17 06:00:30 +00:00
"Coerce `rp` into a ReadPreference if not nil. See `collection` for usage."
2019-11-19 08:01:21 +00:00
[{:keys [read-preference]}]
(when read-preference
(if (instance? ReadPreference read-preference)
read-preference
(ReadPreference/valueOf (name read-preference)))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^WriteConcern ->WriteConcern
2019-11-17 06:00:30 +00:00
"Coerces write-concern related options to a WriteConcern. See `collection` for usage."
2019-11-19 08:01:21 +00:00
[{:keys [write-concern ^Integer write-concern/w ^Long write-concern/w-timeout-ms ^Boolean write-concern/journal?]}]
2019-11-14 02:01:38 +00:00
(when (some some? [write-concern w w-timeout-ms journal?])
2019-11-19 08:01:21 +00:00
(let [^WriteConcern wc (when write-concern
(if (instance? WriteConcern write-concern)
write-concern
(WriteConcern/valueOf (name write-concern))))]
(cond-> (or wc (WriteConcern/ACKNOWLEDGED))
w (.withW w)
w-timeout-ms (.withWTimeout w-timeout-ms (TimeUnit/MILLISECONDS))
(some? journal?) (.withJournal journal?)))))
(defn ^MongoCollection collection
2019-11-14 02:01:38 +00:00
"Coerces `coll` to a MongoCollection with some options.
2019-11-17 06:00:30 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-17 06:00:30 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name or a MongoCollection. This is to provide flexibility, either in reuse of
instances or in some more complex configuration we do not directly support.
- `opts` (optional), a map of:
- `:read-preference` Accepts a ReadPreference or a kw corresponding to one:
[:primary, :primaryPreferred, :secondary, :secondaryPreferred, :nearest]
Invalid values will throw an exception.
- `:read-concern` Accepts a ReadConcern or kw corresponding to one:
[:available, :default, :linearizable, :local, :majority, :snapshot]
Invalid values will throw an exception.
- `:write-concern` A WriteConcern or kw corresponding to one:
[:acknowledged, :journaled, :majority, :unacknowledged, :w1, :w2, :w3],
defaulting to :acknowledged, if some invalid option is provided.
- `:write-concern/w` an int >= 0, controlling the number of replicas to acknowledge
- `:write-concern/w-timeout-ms` How long to wait for secondaries to acknowledge before failing,
in milliseconds (0 means indefinite).
- `:write-concern/journal?` If true, block until write operations have been committed to the journal."
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll]
(collection db coll {}))
([^MongoDatabase db coll opts]
2019-11-19 08:01:21 +00:00
(let [^MongoCollection coll' (if (instance? MongoCollection coll) coll (.getCollection db coll))
rp (->ReadPreference opts)
rc (->ReadConcern opts)
wc (->WriteConcern opts)]
(cond-> ^MongoCollection coll'
rp (.withReadPreference rp)
rc (.withReadConcern rc)
wc (.withWriteConcern wc)))))
2019-11-14 02:01:38 +00:00
;;; CRUD functions
2019-11-14 12:09:01 +00:00
(defn aggregate
2019-11-17 06:00:30 +00:00
"Aggregates documents according to the specified aggregation pipeline and return a seq of maps,
unless configured otherwise..
2019-11-14 12:09:01 +00:00
Arguments:
- `db` is a MongoDatabase
- `coll` is a collection namee
- `q` is a map representing a query.
- `opts` (optional), a map of:
- `:allow-disk-use?` whether to allow writing temporary files
- `:batch-size` Documents to return per batch, e.g. 1
- `:bypass-document-validation?` Boolean
- `:keywordize?` keywordize the keys of return results, default: true
- `:raw?` return the mongo AggregateIterable directly instead of processing into a seq, default: false
- `:session` a ClientSession"
([^MongoDatabase db coll pipeline]
(aggregate db coll pipeline {}))
([^MongoDatabase db coll pipeline opts]
2019-11-19 08:01:21 +00:00
(let [{:keys [^ClientSession session allow-disk-use? ^Integer batch-size bypass-document-validation? keywordize? raw?] :or {keywordize? true raw? false}} opts
it (cond-> (if session
(.aggregate (collection db coll opts) session ^List (map document pipeline))
(.aggregate (collection db coll opts) ^List (map document pipeline)))
(some? allow-disk-use?) (.allowDiskUse allow-disk-use?)
(some? bypass-document-validation?) (.bypassDocumentValidation bypass-document-validation?)
batch-size (.batchSize batch-size))]
2019-11-14 12:09:01 +00:00
(if-not raw?
(map (fn [x] (from-document x keywordize?)) (seq it))
it))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^CountOptions ->CountOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into CountOptions. See `count-documents` for usage."
[{:keys [count-options hint limit max-time-ms skip]}]
2019-11-19 08:01:21 +00:00
(let [^CountOptions opts (or count-options (CountOptions.))]
(cond-> opts
hint (.hint (document hint))
limit (.limit limit)
max-time-ms (.maxTime max-time-ms (TimeUnit/MILLISECONDS))
skip (.skip skip))))
2019-11-14 02:01:38 +00:00
(defn count-documents
"Count documents in a collection, optionally matching a filter query `q`.
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `opts` (optional), a map of:
- `:hint` an index name (string) hint or specification (map)
- `:max-time-ms` max amount of time to allow the query to run, in milliseconds
- `:skip` number of documents to skip before counting
- `:limit` max number of documents to count
- `:count-options` a CountOptions, for configuring directly. If specified, any
other [preceding] query options will be applied to it.
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`."
([^MongoDatabase db coll]
(.countDocuments (collection db coll {})))
([^MongoDatabase db coll q]
(count-documents db coll q {}))
([^MongoDatabase db coll q opts]
(let [opts' (->CountOptions opts)]
(if-let [session (:session opts)]
(.countDocuments (collection db coll opts) session (document q) opts')
(.countDocuments (collection db coll opts) (document q) opts')))))
2019-11-19 08:01:21 +00:00
(defn ^DeleteOptions ->DeleteOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into DeleteOptions. See `delete-one` and `delete-many` for usage."
[{:keys [delete-options]}]
2019-11-19 08:01:21 +00:00
(let [^DeleteOptions opts (or delete-options (DeleteOptions.))]
2019-11-14 02:01:38 +00:00
opts))
(defn delete-one
2019-11-14 12:09:01 +00:00
"Deletes a single document from a collection and returns a DeleteResult.
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `opts` (optional), a map of:
- `:delete-options` A DeleteOptions for configuring directly.
- `:session` A ClientSession
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Additionally takes options specified in `collection`"
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q]
(delete-one db coll q {}))
([^MongoDatabase db coll q opts]
(if-let [session (:session opts)]
(.deleteOne (collection db coll opts) session (document q) (->DeleteOptions opts))
(.deleteOne (collection db coll opts) (document q) (->DeleteOptions opts)))))
(defn delete-many
"Deletes multiple documents from a collection and returns a DeleteResult.
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `opts` (optional), a map of:
- `:delete-options` A DeleteOptions for configuring directly.
- `:session` A ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`"
([^MongoDatabase db coll q]
(delete-many db coll q {}))
([^MongoDatabase db coll q opts]
(if-let [session (:session opts)]
(.deleteMany (collection db coll opts) session (document q) (->DeleteOptions opts))
(.deleteMany (collection db coll opts) (document q) (->DeleteOptions opts)))))
(defn find
2019-11-14 12:09:01 +00:00
"Finds documents and returns a seq of maps, unless configured otherwise.
Arguments:
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `opts` (optional), a map of:
- `:limit` Max number of documents to return, e.g. 1
- `:skip` Number of documents to skip, e.g. 1
- `:sort` document representing sort order, e.g. {:timestamp -1}
- `:projection` document representing fields to return, e.g. {:_id 0}
- `:keywordize?` keywordize the keys of return results, default: true
- `:raw?` return the mongo FindIterable directly instead of processing into a seq, default: false
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`."
([^MongoDatabase db coll q]
2019-11-14 12:09:01 +00:00
(find db coll q {}))
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q opts]
2019-11-19 08:01:21 +00:00
(let [{:keys [limit skip sort projection ^ClientSession session keywordize? raw?] :or {keywordize? true raw? false}} opts]
(let [it (cond-> (if session
(.find (collection db coll opts) session (document q))
(.find (collection db coll opts) (document q)))
limit (.limit limit)
skip (.skip skip)
sort (.sort (document sort))
projection (.projection (document projection)))]
2019-11-14 12:09:01 +00:00
(if-not raw?
(map (fn [x] (from-document x keywordize?)) (seq it))
it)))))
(defn find-one
2019-11-14 02:01:38 +00:00
"Finds a single document and returns it as a clojure map, or nil if not found.
2019-11-14 12:09:01 +00:00
Takes the same options as `find`."
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q]
2019-11-14 12:09:01 +00:00
(find-one db coll q {}))
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q opts]
2019-11-14 12:09:01 +00:00
(first (find db coll q (assoc opts :limit 1 :raw? false)))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^FindOneAndUpdateOptions ->FindOneAndUpdateOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into FindOneAndUpdateOptions. See `find-one-and-update` for usage."
[{:keys [find-one-and-update-options upsert? return-new? sort projection]}]
2019-11-19 08:01:21 +00:00
(let [^FindOneAndUpdateOptions opts (or find-one-and-update-options (FindOneAndUpdateOptions.))]
(cond-> opts
(some? upsert?) (.upsert upsert?)
return-new? (.returnDocument (ReturnDocument/AFTER))
sort (.sort (document sort))
projection (.projection (document projection)))))
2019-11-14 02:01:38 +00:00
(defn find-one-and-update
"Atomically find a document (at most one) and modify it.
2019-11-14 12:09:01 +00:00
Arguments:
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `update` is a map representing an update. The update to apply must include only update operators.
- `opts` (optional), a map of:
- `: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
- `:sort` map representing sort order, e.g. {:timestamp -1}
- `:projection` map representing fields to return, e.g. {:_id 0}
- `:find-one-and-update-options` A FindOneAndUpdateOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it.
- `:keywordize?` keywordize the keys of return results, default: true
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Additionally takes options specified in `collection`."
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q update]
(find-one-and-update db coll q update {}))
([^MongoDatabase db coll q update opts]
2019-11-19 08:01:21 +00:00
(let [{:keys [keywordize? ^ClientSession session] :or {keywordize? true}} opts
2019-11-14 02:01:38 +00:00
opts' (->FindOneAndUpdateOptions opts)]
(-> (if session
(.findOneAndUpdate (collection db coll opts) session (document q) (document update) opts')
(.findOneAndUpdate (collection db coll opts) (document q) (document update) opts'))
2019-11-14 12:09:01 +00:00
(from-document keywordize?)))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^FindOneAndReplaceOptions ->FindOneAndReplaceOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into FindOneAndReplaceOptions. See `find-one-and-replace` for usage."
[{:keys [find-one-and-replace-options upsert? return-new? sort projection]}]
2019-11-19 08:01:21 +00:00
(let [^FindOneAndReplaceOptions opts (or find-one-and-replace-options (FindOneAndReplaceOptions.))]
(cond-> opts
(some? upsert?) (.upsert upsert?)
return-new? (.returnDocument (ReturnDocument/AFTER))
sort (.sort (document sort))
projection (.projection (document projection)))))
2019-11-14 02:01:38 +00:00
(defn find-one-and-replace
"Atomically find a document (at most one) and replace it.
2019-11-14 12:09:01 +00:00
Arguments:
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `doc` is a new document to add.
- `opts` (optional), a map of:
- `: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
- `:sort` map representing sort order, e.g. {:timestamp -1}
- `:projection` map representing fields to return, e.g. {:_id 0}
- `:find-one-and-replace-options` A FindOneAndReplaceOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it.
- `:keywordize?` keywordize the keys of return results, default: true
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Additionally takes options specified in `collection`."
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q doc]
(find-one-and-replace db coll q doc {}))
([^MongoDatabase db coll q doc opts]
2019-11-14 12:09:01 +00:00
(let [{:keys [keywordize? session] :or {keywordize? true}} opts
2019-11-14 02:01:38 +00:00
opts' (->FindOneAndReplaceOptions opts)]
(-> (if session
(.findOneAndReplace (collection db coll opts) session (document q) (document doc) opts')
(.findOneAndReplace (collection db coll opts) (document q) (document doc) opts'))
2019-11-14 12:09:01 +00:00
(from-document keywordize?)))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^InsertOneOptions ->InsertOneOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into InsertOneOptions. See `insert-one` for usage."
[{:keys [insert-one-options bypass-document-validation?]}]
2019-11-19 08:01:21 +00:00
(let [^InsertOneOptions opts (or insert-one-options (InsertOneOptions.))]
2019-11-14 02:01:38 +00:00
(when (some? bypass-document-validation?) (.bypassDocumentValidation opts bypass-document-validation?))
opts))
(defn insert-one
2019-11-14 12:09:01 +00:00
"Inserts a single document into a collection, and returns nil.
2019-11-14 02:01:38 +00:00
If the document does not have an _id field, it will be auto-generated by the underlying driver.
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `doc` is a map to insert.
- `opts` (optional), a map of:
- `:bypass-document-validation?` Boolean
- `:insert-one-options` An InsertOneOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it.
- `:session` A ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`."
([^MongoDatabase db coll doc]
(insert-one db coll doc {}))
([^MongoDatabase db coll doc opts]
(let [opts' (->InsertOneOptions opts)]
(if-let [session (:session opts)]
(.insertOne (collection db coll opts) session (document doc) opts')
(.insertOne (collection db coll opts) (document doc) opts')))))
2019-11-19 08:01:21 +00:00
(defn ^InsertManyOptions ->InsertManyOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into InsertManyOptions. See `insert-many` for usage."
[{:keys [insert-many-options bypass-document-validation? ordered?]}]
2019-11-19 08:01:21 +00:00
(let [^InsertManyOptions opts (or insert-many-options (InsertManyOptions.))]
(cond-> opts
(some? bypass-document-validation?) (.bypassDocumentValidation bypass-document-validation?)
(some? ordered?) (.ordered ordered?))))
2019-11-14 02:01:38 +00:00
(defn insert-many
"Inserts multiple documents into a collection.
If a document does not have an _id field, it will be auto-generated by the underlying driver.
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `docs` is a collection of maps to insert
- `opts` (optional), a map of:
- `:bypass-document-validation?` Boolean
- `:ordered?` Boolean whether serve should insert documents in order provided (default true)
- `:insert-many-options` An InsertManyOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it.
- `:session` A ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`"
([^MongoDatabase db coll docs]
(insert-many db coll docs {}))
([^MongoDatabase db coll docs opts]
(let [opts' (->InsertManyOptions opts)]
2019-11-19 08:01:21 +00:00
(if-let [^ClientSession session (:session opts)]
(.insertMany (collection db coll opts) session ^List (map document docs) opts')
(.insertMany (collection db coll opts) ^List (map document docs) opts')))))
2019-11-14 02:01:38 +00:00
2019-11-19 08:01:21 +00:00
(defn ^ReplaceOptions ->ReplaceOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into ReplaceOptions. See `replace-one` and `replace-many` for usage."
[{:keys [replace-options upsert? bypass-document-validation?]}]
2019-11-19 08:01:21 +00:00
(let [^ReplaceOptions opts (or replace-options (ReplaceOptions.))]
(cond-> opts
(some? upsert?) (.upsert upsert?)
(some? bypass-document-validation?) (.bypassDocumentValidation bypass-document-validation?))))
2019-11-14 02:01:38 +00:00
(defn replace-one
2019-11-14 12:09:01 +00:00
"Replace a single document in a collection and returns an UpdateResult.
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `doc` is a new document to add.
- `opts` (optional), a map of:
- `:upsert?` whether to insert a new document if nothing is found, default: false
- `:bypass-document-validation?` Boolean
- `:replace-options` A ReplaceOptions for configuring directly. If specified,
any other [preceding[ query options will be applied to it.
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
Additionally takes options specified in `collection`"
([^MongoDatabase db coll q doc]
(find-one-and-replace db coll q doc {}))
([^MongoDatabase db coll q doc opts]
2019-11-19 08:01:21 +00:00
(if-let [^ClientSession session (:session opts)]
2019-11-14 02:01:38 +00:00
(.replaceOne (collection db coll opts) session (document q) (document doc) (->ReplaceOptions opts))
(.replaceOne (collection db coll opts) (document q) (document doc) (->ReplaceOptions opts)))))
2019-11-19 08:01:21 +00:00
(defn ^UpdateOptions ->UpdateOptions
2019-11-14 12:09:01 +00:00
"Coerce options map into UpdateOptions. See `update-one` and `update-many` for usage."
2019-11-14 02:01:38 +00:00
[{:keys [update-options upsert? bypass-document-validation?]}]
2019-11-19 08:01:21 +00:00
(let [^UpdateOptions opts (or update-options (UpdateOptions.))]
(cond-> opts
(some? upsert?) (.upsert upsert?)
(some? bypass-document-validation?) (.bypassDocumentValidation bypass-document-validation?))))
2019-11-14 02:01:38 +00:00
(defn update-one
2019-11-14 12:09:01 +00:00
"Updates a single document in a collection and returns an UpdateResult.
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `update` is a map representing an update. The update to apply must include only update operators.
- `opts` (optional), a map of:
- `:upsert?` whether to insert a new document if nothing is found, default: false
- `:bypass-document-validation?` Boolean
- `:update-options` An UpdateOptions for configuring directly. If specified,
any other [preceding[ query options will be applied to it.
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Additionally takes options specified in `collection`"
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q update]
(update-one db coll q update {}))
([^MongoDatabase db coll q update opts]
2019-11-19 08:01:21 +00:00
(if-let [^ClientSession session (:session opts)]
2019-11-14 02:01:38 +00:00
(.updateOne (collection db coll opts) session (document q) (document update) (->UpdateOptions opts))
(.updateOne (collection db coll opts) (document q) (document update) (->UpdateOptions opts)))))
(defn update-many
2019-11-14 12:09:01 +00:00
"Updates many documents in a collection and returns an UpdateResult.
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `q` is a map representing a query.
- `update` is a map representing an update. The update to apply must include only update operators.
- `opts` (optional), a map of:
- `:upsert?` whether to insert a new document if nothing is found, default: false
- `:bypass-document-validation?` Boolean
- `:update-options` An UpdateOptions for configuring directly. If specified,
any other [preceding[ query options will be applied to it.
- `:session` a ClientSession
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Additionally takes options specified in `collection`"
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q update]
2019-11-19 08:01:21 +00:00
(update-many db coll q update {}))
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll q update opts]
2019-11-19 08:01:21 +00:00
(if-let [^ClientSession session (:session opts)]
2019-11-14 02:01:38 +00:00
(.updateMany (collection db coll opts) session (document q) (document update) (->UpdateOptions opts))
(.updateMany (collection db coll opts) (document q) (document update) (->UpdateOptions opts)))))
;;; Admin functions
2019-11-19 08:01:21 +00:00
(defn ^CreateCollectionOptions ->CreateCollectionOptions
2019-11-14 02:01:38 +00:00
"Coerce options map into CreateCollectionOptions. See `create` usage."
[{:keys [create-collection-options capped? max-documents max-size-bytes]}]
2019-11-19 08:01:21 +00:00
(let [^CreateCollectionOptions opts (or create-collection-options (CreateCollectionOptions.))]
(cond-> opts
(some? capped?) (.capped capped?)
max-documents (.maxDocuments max-documents)
max-size-bytes (.sizeInBytes max-size-bytes))))
2019-11-14 02:01:38 +00:00
(defn create
"Creates a collection
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `opts` (optional), a map of:
- `:capped?` Boolean whether to create a capped collection
- `:max-documents` max documents for a capped collection
- `:max-size-bytes` max collection size in bytes for a capped collection
- `:create-collection-options` A CreateCollectionOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it"
2019-11-19 08:01:21 +00:00
([^MongoDatabase db ^String coll]
2019-11-14 02:01:38 +00:00
(create db coll {}))
2019-11-19 08:01:21 +00:00
([^MongoDatabase db ^String coll opts]
2019-11-14 02:01:38 +00:00
(let [opts' (->CreateCollectionOptions opts)]
(.createCollection db coll opts'))))
2019-11-19 08:01:21 +00:00
(defn ^RenameCollectionOptions ->RenameCollectionOptions
2019-11-14 03:27:59 +00:00
"Coerce options map into RenameCollectionOptions. See `rename` usage."
[{:keys [rename-collection-options drop-target?]}]
2019-11-19 08:01:21 +00:00
(let [^RenameCollectionOptions opts (or rename-collection-options (RenameCollectionOptions.))]
(cond-> opts
(some? drop-target?) (.dropTarget drop-target?))))
2019-11-14 03:27:59 +00:00
2019-11-14 02:01:38 +00:00
(defn rename
2019-11-14 03:27:59 +00:00
"Renames `coll` to `new-coll` in the same DB.
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 03:27:59 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `new-coll` is the target collection name
- `opts` (optional), a map of:
- `:drop-target?` Boolean drop tne target collection if it exists. Default: false
- `:rename-collection-options` A RenameCollectionOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it"
2019-11-14 03:27:59 +00:00
([^MongoDatabase db coll new-coll]
(rename db coll new-coll {}))
([^MongoDatabase db coll new-coll opts]
(let [opts' (->RenameCollectionOptions opts)]
2019-11-14 02:01:38 +00:00
2019-11-14 03:27:59 +00:00
(.renameCollection (collection db coll opts)
(MongoNamespace. (.getName db) new-coll)
opts'))))
2019-11-14 02:01:38 +00:00
(defn drop
"Drops a collection from a database."
[^MongoDatabase db coll]
(.drop (collection db coll)))
2019-11-19 08:01:21 +00:00
(defn ^IndexOptions ->IndexOptions
2019-11-14 02:01:38 +00:00
"Coerces an options map into an IndexOptions.
See `create-index` for usage"
[{:keys [index-options name sparse? unique?]}]
2019-11-19 08:01:21 +00:00
(let [^IndexOptions opts (or index-options (IndexOptions.))]
(cond-> opts
name (.name name)
(some? sparse?) (.sparse sparse?)
(some? unique?) (.unique unique?))))
2019-11-14 02:01:38 +00:00
(defn create-index
"Creates an index
2019-11-14 12:09:01 +00:00
Arguments:
- `db` is a MongoDatabase
- `coll` is a collection name
- `keys` is a document representing index keys, e.g. {:a 1}
- `opts` (optional), a map of:
- `:name`
- `:sparse?`
- `:unique?`
- `:index-options` An IndexOptions for configuring directly. If specified,
any other [preceding] query options will be applied to it"
2019-11-14 02:01:38 +00:00
([^MongoDatabase db coll keys]
(create-index db coll keys {}))
([^MongoDatabase db coll keys opts]
(.createIndex (collection db coll opts) (document keys) (->IndexOptions opts))))
(defn create-indexes
"Creates many indexes.
2019-11-14 12:09:01 +00:00
Arguments:
2019-11-14 02:01:38 +00:00
2019-11-14 12:09:01 +00:00
- `db` is a MongoDatabase
- `coll` is a collection name
- `indexes` is a collection of maps with the following keys:
- `:keys` (mandatory) a document representing index keys, e.g. {:a 1}
- `:name`
- `:sparse?`
- `:unique?`"
2019-11-14 03:27:59 +00:00
([^MongoDatabase db coll indexes]
(create-indexes db coll indexes {}))
([^MongoDatabase db coll indexes opts]
(->> indexes
(map (fn [x] (IndexModel. (document (:keys x)) (->IndexOptions x))))
(.createIndexes (collection db coll opts)))))
2019-11-14 02:01:38 +00:00
(defn list-indexes
"Lists indexes."
2019-11-14 03:27:59 +00:00
([^MongoDatabase db coll]
(list-indexes db coll {}))
([^MongoDatabase db coll opts]
(->> (.listIndexes (collection db coll opts))
2019-11-22 09:55:03 +00:00
(map #(from-document % true)))))