Skip to content

Commit a325758

Browse files
committed
adds support to count in ravendb
1 parent 862e9c6 commit a325758

File tree

2 files changed

+19
-0
lines changed

2 files changed

+19
-0
lines changed

ravendb-driver/src/main/java/org/jnosql/diana/ravendb/document/RavenDBDocumentCollectionManager.java

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
package org.jnosql.diana.ravendb.document;
1717

1818
import net.ravendb.client.documents.DocumentStore;
19+
import net.ravendb.client.documents.queries.Query;
20+
import net.ravendb.client.documents.session.IDocumentQuery;
1921
import net.ravendb.client.documents.session.IDocumentSession;
2022
import net.ravendb.client.documents.session.IEnumerableQuery;
2123
import net.ravendb.client.documents.session.IMetadataDictionary;
@@ -120,6 +122,15 @@ public List<DocumentEntity> select(DocumentQuery query) {
120122

121123
}
122124

125+
@Override
126+
public long count(String documentCollection) {
127+
Objects.requireNonNull(documentCollection, "documentCollection is required");
128+
try (IDocumentSession session = store.openSession()) {
129+
IDocumentQuery<HashMap> ravenQuery = session.query(HashMap.class, Query.collection(documentCollection));
130+
return ravenQuery.count();
131+
}
132+
}
133+
123134
@Override
124135
public void close() {
125136
store.close();

ravendb-driver/src/test/java/org/jnosql/diana/ravendb/document/RavenDBDocumentCollectionManagerTest.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -352,6 +352,14 @@ private DocumentEntity createSubdocumentList() {
352352
}
353353

354354

355+
@Test
356+
public void shouldCount() {
357+
DocumentEntity entity = getEntity();
358+
entityManager.insert(entity);
359+
assertTrue(entityManager.count(COLLECTION_NAME) > 0);
360+
}
361+
362+
355363
private DocumentEntity getEntity() {
356364
DocumentEntity entity = DocumentEntity.of(COLLECTION_NAME);
357365
Map<String, Object> map = new HashMap<>();

0 commit comments

Comments
 (0)