feat: create select cursosr at SemistructuredTemplate

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse · Mar 31, 2024 · 19f331a · 19f331a
1 parent 6074ec1
commit 19f331a
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 1 deletion.
diff --git a/...c/main/java/org/eclipse/jnosql/mapping/semistructured/AbstractSemistructuredTemplate.java b/...c/main/java/org/eclipse/jnosql/mapping/semistructured/AbstractSemistructuredTemplate.java
@@ -16,6 +16,9 @@
 
 
 import jakarta.data.exceptions.NonUniqueResultException;
+import jakarta.data.page.CursoredPage;
+import jakarta.data.page.PageRequest;
+import jakarta.data.page.impl.CursoredPageRecord;
 import jakarta.nosql.QueryMapper;
 
 import org.eclipse.jnosql.communication.semistructured.CommunicationEntity;
@@ -34,6 +37,7 @@
 
 import java.time.Duration;
 import java.util.Iterator;
+import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.function.Consumer;
@@ -299,6 +303,19 @@ public <T> void deleteAll(Class<T> type) {
         manager().delete(DeleteQuery.delete().from(metadata.name()).build());
     }
 
+    @SuppressWarnings("unchecked")
+    @Override
+    public <T> CursoredPage<T> selectCursor(SelectQuery query, PageRequest<T> pageRequest){
+        Objects.requireNonNull(query, "query is required");
+        Objects.requireNonNull(pageRequest, "pageRequest is required");
+        CursoredPage<CommunicationEntity> cursoredPage = this.manager().selectCursor(query, pageRequest);
+        List<T> entities = cursoredPage.stream().<T>map(c -> converter().toEntity(c)).toList();
+        PageRequest<T> nextPageRequest = cursoredPage.hasNext()? (PageRequest<T>) cursoredPage.nextPageRequest(): null;
+        PageRequest<T> beforePageRequest = cursoredPage.hasPrevious()? (PageRequest<T>) cursoredPage.previousPageRequest(): null;
+        List<PageRequest.Cursor> cursors = ((CursoredPageRecord<CommunicationEntity>) cursoredPage).cursors();
+        return new CursoredPageRecord<>(entities, cursors, -1, pageRequest, nextPageRequest, beforePageRequest);
+    }
+
     protected <T> T persist(T entity, UnaryOperator<CommunicationEntity> persistAction) {
         return Stream.of(entity)
                 .map(toUnary(eventManager()::firePreEntity))

diff --git a/...tured/src/main/java/org/eclipse/jnosql/mapping/semistructured/SemistructuredTemplate.java b/...tured/src/main/java/org/eclipse/jnosql/mapping/semistructured/SemistructuredTemplate.java
@@ -11,7 +11,6 @@
 package org.eclipse.jnosql.mapping.semistructured;
 
 import jakarta.data.page.CursoredPage;
-import jakarta.data.page.Page;
 import jakarta.data.page.PageRequest;
 import org.eclipse.jnosql.mapping.PreparedStatement;
 import jakarta.nosql.Template;

diff --git a/...ed/src/test/java/org/eclipse/jnosql/mapping/semistructured/DefaultColumnTemplateTest.java b/...ed/src/test/java/org/eclipse/jnosql/mapping/semistructured/DefaultColumnTemplateTest.java
@@ -15,6 +15,9 @@
 package org.eclipse.jnosql.mapping.semistructured;
 
 import jakarta.data.exceptions.NonUniqueResultException;
+import jakarta.data.page.CursoredPage;
+import jakarta.data.page.PageRequest;
+import jakarta.data.page.impl.CursoredPageRecord;
 import jakarta.enterprise.inject.Instance;
 import jakarta.inject.Inject;
 import org.eclipse.jnosql.mapping.PreparedStatement;
@@ -43,6 +46,8 @@
 
 import java.time.Duration;
 import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
 import java.util.Optional;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
@@ -439,4 +444,38 @@ void shouldDeleteAll(){
         template.deleteAll(Person.class);
         verify(managerMock).delete(delete().from("Person").build());
     }
+
+    @Test
+    void shouldSelectCursor(){
+        PageRequest<CommunicationEntity> request = PageRequest.ofSize(2);
+        PageRequest<CommunicationEntity> afterKey = PageRequest.<CommunicationEntity>ofSize(2)
+                .afterKey("Ada");
+        SelectQuery query = select().from("Person").orderBy("name").asc().build();
+
+        Mockito.when(managerMock.selectCursor(query, request))
+                .thenReturn(new CursoredPageRecord<>(content(),
+                        Collections.emptyList(), -1, request, afterKey, null));
+
+        PageRequest<Person> personRequest = PageRequest.ofSize(2);
+        CursoredPage<Person> result = template.selectCursor(query, personRequest);
+
+        SoftAssertions.assertSoftly(soft ->{
+            soft.assertThat(result).isNotNull();
+            soft.assertThat(result.content()).hasSize(1);
+            soft.assertThat(result.hasNext()).isTrue();
+            Person person = result.stream().findFirst().orElseThrow();
+
+            soft.assertThat(person.getAge()).isEqualTo(10);
+            soft.assertThat(person.getName()).isEqualTo("Name");
+            soft.assertThat(person.getPhones()).containsExactly("234", "432");
+        });
+
+    }
+
+
+    private List<CommunicationEntity> content(){
+        CommunicationEntity columnEntity = CommunicationEntity.of("Person");
+        columnEntity.addAll(Stream.of(columns).collect(Collectors.toList()));
+        return List.of(columnEntity);
+    }
 }