Exposed query as DBObject

querydsl · Apr 17, 2016 · 627df49 · 627df49
1 parent 5cdb0e3
commit 627df49
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/querydsl-mongodb/src/main/java/com/querydsl/mongodb/AbstractMongodbQuery.java b/querydsl-mongodb/src/main/java/com/querydsl/mongodb/AbstractMongodbQuery.java
@@ -411,6 +411,15 @@ public void setReadPreference(ReadPreference readPreference) {
         this.readPreference = readPreference;
     }
 
+    /**
+     * Get the where definition as a DBObject instance
+     *
+     * @return
+     */
+    public DBObject asDBObject() {
+        return createQuery(queryMixin.getMetadata().getWhere());
+    }
+
     @Override
     public String toString() {
         return createQuery(queryMixin.getMetadata().getWhere()).toString();

diff --git a/querydsl-mongodb/src/test/java/com/querydsl/mongodb/MongodbQueryTest.java b/querydsl-mongodb/src/test/java/com/querydsl/mongodb/MongodbQueryTest.java
@@ -27,6 +27,7 @@
 import org.mongodb.morphia.Morphia;
 
 import com.google.common.collect.Lists;
+import com.mongodb.BasicDBObject;
 import com.mongodb.Mongo;
 import com.mongodb.MongoException;
 import com.mongodb.ReadPreference;
@@ -579,7 +580,15 @@ public void readPreference() {
         MorphiaQuery<User> query = query();
         query.setReadPreference(ReadPreference.primary());
         assertEquals(4, query.fetchCount());
+    }
 
+    @Test
+    public void asDBObject() {
+        MorphiaQuery<User> query = query();
+        query.where(user.firstName.eq("Bob"), user.lastName.eq("Wilson"));
+        assertEquals(
+                new BasicDBObject().append("firstName", "Bob").append("lastName", "Wilson"),
+                query.asDBObject());
     }
 
     //TODO