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

Initial implementation of monger.collection/find-one

Browse source
This commit is contained in:
Michael S. Klishin 2011-08-29 19:36:11 +04:00
parent b8c545257f
commit 1ee5436841
2 changed files with 66 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/monger/collection.clj
View file

@ -16,7 +16,9 @@
;; API ;; API
;; ;;
;;
;; monger.collection/insert ;; monger.collection/insert
;;
(defn ^WriteResult insert (defn ^WriteResult insert
([^String collection, ^DBObject document] ([^String collection, ^DBObject document]
@ -35,7 +37,9 @@
(let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.insert #^DBCollection coll #^List (to-db-object documents) #^WriteConcern concern)))) (.insert #^DBCollection coll #^List (to-db-object documents) #^WriteConcern concern))))
;;
;; monger.collection/find ;; monger.collection/find
;;
(declare fields-to-db-object) (declare fields-to-db-object)
(defn ^DBCursor find (defn ^DBCursor find
@ -48,10 +52,27 @@
([^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 #^DBCollection coll #^DBObject (to-db-object ref) #^DBObject (to-db-object map-of-fields)))) (.find #^DBCollection coll #^DBObject (to-db-object ref) #^DBObject (to-db-object map-of-fields)))))
)
;;
;; monger.collection/find-one
;;
(defn ^DBCursor find-one
([^String collection, ^Map ref]
(let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]
(.find #^DBCollection coll #^DBObject (to-db-object ref))))
([^String collection, ^Map ref, ^List fields]
(let [#^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)
map-of-fields (fields-to-db-object fields)]
(.find #^DBCollection coll #^DBObject (to-db-object ref) #^DBObject (to-db-object map-of-fields)))))
;;
;; monger.collection/find-by-id
;;
(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)]
@ -62,10 +83,16 @@
(.findOne #^DBCollection coll #^DBObject (to-db-object { :_id id }) #^DBObject (to-db-object map-of-fields))))) (.findOne #^DBCollection coll #^DBObject (to-db-object { :_id id }) #^DBObject (to-db-object map-of-fields)))))
;;
;; monger.collection/group ;; monger.collection/group
;;
;; TBD
;;
;; monger.collection/count ;; monger.collection/count
;;
(defn ^long count (defn ^long count
([^String collection] ([^String collection]
(let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)] (let [^DBCollection coll (.getCollection monger.core/*mongodb-database* collection)]

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

@ -1,7 +1,7 @@
(set! *warn-on-reflection* true) (set! *warn-on-reflection* true)
(ns monger.test.collection (ns monger.test.collection
(:import [com.mongodb WriteResult WriteConcern DBCursor]) (:import [com.mongodb WriteResult WriteConcern DBCursor DBObject])
(:require [monger core collection errors util] [clojure stacktrace]) (:require [monger core collection errors util] [clojure stacktrace])
(:use [clojure.test])) (:use [clojure.test]))
@ -107,6 +107,41 @@
(is (instance? DBCursor cursor)))) (is (instance? DBCursor cursor))))
;;
;; find-one
;;
(deftest find-one-full-document-when-collection-is-empty
(let [collection "docs"]
(monger.collection/remove collection)
(def cursor (monger.collection/find-one collection {}))
(is (instance? DBCursor cursor))
(is (empty? cursor))))
(deftest find-one-full-document-when-collection-has-matches
(let [collection "docs"
doc-id (monger.util/random-uuid)
doc { :data-store "MongoDB", :language "Clojure", :_id doc-id }]
(monger.collection/remove collection)
(monger.collection/insert collection doc)
(def cursor (monger.collection/find-one collection { :language "Clojure" }))
(is (= (:_id doc) (.get (.next #^DBCursor cursor) "_id")))))
(deftest find-one-full-document-when-collection-has-matches
(let [collection "docs"
doc-id (monger.util/random-uuid)
doc { :data-store "MongoDB", :language "Clojure", :_id doc-id }
fields [:language]]
(monger.collection/remove collection)
(monger.collection/insert collection doc)
(def cursor (monger.collection/find-one collection { :language "Clojure" } fields))
(def #^DBObject loaded (.next #^DBCursor cursor))
(is (nil? (.get #^DBObject loaded "data-stire")))
(is (= doc-id (.get #^DBObject loaded "_id")))
(is (= "Clojure" (.get #^DBObject loaded "language")))))
;; ;;
;; find-by-id ;; find-by-id