luciano/monger
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement fetching of partial documents

Browse source
This commit is contained in:
Michael S. Klishin 2011-08-15 01:09:25 +04:00
parent b8b255545a
commit f096c032cd
2 changed files with 28 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/monger/collection.clj
View file

@ -20,19 +20,19 @@
(defn ^WriteResult insert (defn ^WriteResult insert
([^String collection, ^DBObject doc] ([^String collection, ^DBObject doc]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.insert coll (to-db-object doc) monger.core/*mongodb-write-concern*))) (.insert coll (to-db-object doc) monger.core/*mongodb-write-concern*)))
([^String collection, ^DBObject doc, ^WriteConcern concern] ([^String collection, ^DBObject doc, ^WriteConcern concern]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.insert coll (to-db-object doc) concern)))) (.insert coll (to-db-object doc) concern))))
(defn ^WriteResult insert-batch (defn ^WriteResult insert-batch
([^String collection, ^List docs] ([^String collection, ^List docs]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.insert coll (to-db-object docs) WriteConcern/NORMAL))) (.insert coll (to-db-object docs) WriteConcern/NORMAL)))
([^String collection, ^List docs, ^WriteConcern concern] ([^String collection, ^List docs, ^WriteConcern concern]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.insert coll (to-db-object docs) concern)))) (.insert coll (to-db-object docs) concern))))
;; monger.collection/find ;; monger.collection/find
@ -40,13 +40,13 @@
(defn ^DBCursor find (defn ^DBCursor find
([^String collection] ([^String collection]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.find coll))) (.find coll)))
([^String collection, ^Map ref] ([^String collection, ^Map ref]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.find coll (to-db-object ref)))) (.find coll (to-db-object ref))))
([^String collection, ^Map ref, ^List fields] ([^String collection, ^Map ref, ^List fields]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection) (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)
map-of-fields (fields-to-db-object fields)] map-of-fields (fields-to-db-object fields)]
(.find coll (to-db-object ref) (to-db-object map-of-fields)))) (.find coll (to-db-object ref) (to-db-object map-of-fields))))
) )
@ -54,10 +54,10 @@
(defn ^DBObject find-by-id (defn ^DBObject find-by-id
([^String collection, ^String id] ([^String collection, ^String id]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.findOne coll (to-db-object { :_id id })))) (.findOne coll (to-db-object { :_id id }))))
([^String collection, ^String id, ^List fields] ([^String collection, ^String id, ^List fields]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection) (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)
map-of-fields (fields-to-db-object fields)] map-of-fields (fields-to-db-object fields)]
(.findOne coll (to-db-object { :_id id }) (to-db-object map-of-fields)))) (.findOne coll (to-db-object { :_id id }) (to-db-object map-of-fields))))
) )
@ -95,8 +95,5 @@
;; ;;
(defn- fields-to-db-object (defn- fields-to-db-object
[fields] [^List fields]
(let [n (clojure.core/count fields) (zipmap fields (repeat 1)))
ones (replicate n 1)
map-of-fields (zipmap fields ones)]
map-of-fields))

17
test/monger/test/collection.clj
View file

@ -141,3 +141,20 @@
(is (= 1 (.count (monger.collection/find collection { :language "Scala" })))) (is (= 1 (.count (monger.collection/find collection { :language "Scala" }))))
(is (= 3 (.count (monger.collection/find collection { :language "Clojure" })))) (is (= 3 (.count (monger.collection/find collection { :language "Clojure" }))))
(is (empty? (monger.collection/find collection { :language "Java" }))))) (is (empty? (monger.collection/find collection { :language "Java" })))))
(deftest find-multiple-partial-documents
(let [collection "libraries"]
(monger.collection/remove collection)
(monger.collection/insert collection { :language "Clojure", :name "monger" })
(monger.collection/insert collection { :language "Clojure", :name "langohr" })
(monger.collection/insert collection { :language "Clojure", :name "incanter" })
(monger.collection/insert collection { :language "Scala", :name "akka" })
(let [scala-libs (monger.collection/find collection { :language "Scala" } [:name])
clojure-libs (monger.collection/find collection { :language "Clojure"} [:language])]
(is (= 1 (.count scala-libs)))
(is (= 3 (.count clojure-libs)))
(doseq [i clojure-libs]
(let [doc (monger.convertion/from-db-object i true)]
(is (= (:language doc) "Clojure"))))
(is (empty? (monger.collection/find collection { :language "Erlang" } [:name]))))))