feat(count): support count feature

Ouest-France · Jan 29, 2024 · 31a12f5 · 31a12f5
1 parent 44783cc
commit 31a12f5
Show file tree

Hide file tree

Showing 12 changed files with 180 additions and 12 deletions.
diff --git a/...mplate-adapter/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestRestTemplate.java b/...mplate-adapter/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestRestTemplate.java
@@ -1,5 +1,6 @@
 package fr.ouestfrance.querydsl.postgrest;
 
+import fr.ouestfrance.querydsl.postgrest.model.CountItem;
 import fr.ouestfrance.querydsl.postgrest.model.Page;
 import fr.ouestfrance.querydsl.postgrest.model.PageImpl;
 import fr.ouestfrance.querydsl.postgrest.model.Range;
@@ -85,6 +86,12 @@ public <T> List<T> delete(String resource, Map<String, List<String>> params, Map
                 .queryParams(queryParams).build().toString(), new HashMap<>()), HttpMethod.DELETE, new HttpEntity<>(null, toHeaders(headers)), listRef(clazz)).getBody();
     }
 
+    @Override
+    public List<CountItem> count(String resource, Map<String, List<String>> map) {
+        return restTemplate.exchange(restTemplate.getUriTemplateHandler().expand(UriComponentsBuilder.fromPath(resource)
+                .queryParams(toMultiMap(map)).build().toString(), new HashMap<>()), HttpMethod.GET, new HttpEntity<>(null, new HttpHeaders()), listRef(CountItem.class)).getBody();
+    }
+
     private static <T> ParameterizedTypeReference<List<T>> listRef(Class<T> clazz) {
         return ParameterizedTypeReference.forType(TypeUtils.parameterize(List.class, clazz));
     }

diff --git a/...sl/postgrest/PostgrestRepositoryTest.java → .../PostgrestRestTemplateRepositoryTest.java b/...sl/postgrest/PostgrestRepositoryTest.java → .../PostgrestRestTemplateRepositoryTest.java
@@ -11,6 +11,7 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.mockserver.client.MockServerClient;
+import org.mockserver.integration.ClientAndServer;
 import org.mockserver.junit.jupiter.MockServerSettings;
 import org.mockserver.model.HttpRequest;
 import org.mockserver.model.HttpResponse;
@@ -27,7 +28,7 @@
 import static org.junit.jupiter.api.Assertions.*;
 
 @MockServerSettings(ports = 8007)
-class PostgrestRepositoryTest {
+class PostgrestRestTemplateRepositoryTest {
 
     private PostgrestRepository<Post> repository;
 
@@ -52,6 +53,14 @@ void shouldSearchPosts(MockServerClient client) {
         search.getData().stream().map(Object::getClass).forEach(x -> assertEquals(Post.class, x));
     }
 
+    @Test
+    void shouldCountPosts(ClientAndServer client) {
+        client.when(HttpRequest.request().withPath("/posts").withQueryStringParameter("select", "count()"))
+                .respond(jsonFileResponse("count_response.json"));
+        long count = repository.count(new PostRequest());
+        assertEquals(300, count);
+    }
+
     @Test
     void shouldSearchPostsWithoutContentRange(MockServerClient client) {
         client.when(HttpRequest.request().withPath("/posts").withQueryStringParameter("select", "*,authors(*)"))

diff --git a/querydsl-postgrest-resttemplate-adapter/src/test/resources/count_response.json b/querydsl-postgrest-resttemplate-adapter/src/test/resources/count_response.json
@@ -0,0 +1,5 @@
+[
+  {
+    "count": 300
+  }
+]
diff --git a/...webclient-adapter/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestWebClient.java b/...webclient-adapter/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestWebClient.java
@@ -1,5 +1,6 @@
 package fr.ouestfrance.querydsl.postgrest;
 
+import fr.ouestfrance.querydsl.postgrest.model.CountItem;
 import fr.ouestfrance.querydsl.postgrest.model.Page;
 import fr.ouestfrance.querydsl.postgrest.model.PageImpl;
 import fr.ouestfrance.querydsl.postgrest.model.Range;
@@ -73,6 +74,19 @@ public <T> Page<T> search(String resource, Map<String, List<String>> params,
                 }).orElse(Page.empty());
     }
 
+    @Override
+    public List<CountItem> count(String resource, Map<String, List<String>> params) {
+        ResponseEntity<List<CountItem>> response = webClient.get().uri(uriBuilder -> {
+                    uriBuilder.path(resource);
+                    uriBuilder.queryParams(toMultiMap(params));
+                    return uriBuilder.build();
+                })
+                .retrieve()
+                .toEntity(listRef(CountItem.class))
+                .block();
+        return Optional.ofNullable(response).map(HttpEntity::getBody).orElse(List.of());
+    }
+
     private static void safeAdd(Map<String, List<String>> headers, HttpHeaders httpHeaders) {
         Optional.ofNullable(headers)
                 .map(PostgrestWebClient::toMultiMap).ifPresent(httpHeaders::addAll);

diff --git a/...l/postgrest/PostrgrestRepositoryTest.java → ...st/PostrgrestWebClientRepositoryTest.java b/...l/postgrest/PostrgrestRepositoryTest.java → ...st/PostrgrestWebClientRepositoryTest.java
@@ -8,9 +8,7 @@
 import fr.ouestfrance.querydsl.postgrest.model.Pageable;
 import lombok.SneakyThrows;
 import org.junit.jupiter.api.Test;
-import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockserver.integration.ClientAndServer;
-import org.mockserver.junit.jupiter.MockServerExtension;
 import org.mockserver.junit.jupiter.MockServerSettings;
 import org.mockserver.model.HttpRequest;
 import org.mockserver.model.HttpResponse;
@@ -25,7 +23,7 @@
 import static org.junit.jupiter.api.Assertions.*;
 
 @MockServerSettings(ports = 8007)
-class PostrgrestRepositoryTest {
+class PostrgrestWebClientRepositoryTest {
 
     private final PostgrestRepository<Post> repository = new PostRepository(PostgrestWebClient.of(WebClient.builder()
             .baseUrl("http://localhost:8007/")
@@ -46,6 +44,14 @@ void shouldSearchPosts(ClientAndServer client) {
         search.getData().stream().map(Object::getClass).forEach(x -> assertEquals(Post.class, x));
     }
 
+    @Test
+    void shouldCountPosts(ClientAndServer client) {
+        client.when(HttpRequest.request().withPath("/posts").withQueryStringParameter("select", "count()"))
+                .respond(jsonFileResponse("count_response.json"));
+        long count = repository.count(new PostRequest());
+        assertEquals(300, count);
+    }
+
     @Test
     void shouldFindById(ClientAndServer client) {
         client.when(HttpRequest.request().withPath("/posts")

diff --git a/querydsl-postgrest-webclient-adapter/src/test/resources/count_response.json b/querydsl-postgrest-webclient-adapter/src/test/resources/count_response.json
@@ -0,0 +1,5 @@
+[
+  {
+    "count": 300
+  }
+]
diff --git a/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestClient.java b/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestClient.java
@@ -1,5 +1,6 @@
 package fr.ouestfrance.querydsl.postgrest;
 
+import fr.ouestfrance.querydsl.postgrest.model.CountItem;
 import fr.ouestfrance.querydsl.postgrest.model.Page;
 
 import java.util.List;
@@ -60,4 +61,12 @@ <T> Page<T> search(String resource, Map<String, List<String>> params,
      */
     <T> List<T> delete(String resource, Map<String, List<String>> params, Map<String, List<String>> headers, Class<T> clazz);
 
+    /**
+     * Count data
+     *
+     * @param resource resource name
+     * @param map      query params
+     * @return list of count items
+     */
+    List<CountItem> count(String resource, Map<String, List<String>> map);
 }
diff --git a/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestRepository.java b/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/PostgrestRepository.java
@@ -3,12 +3,10 @@
 import fr.ouestfrance.querydsl.postgrest.annotations.Header;
 import fr.ouestfrance.querydsl.postgrest.annotations.PostgrestConfiguration;
 import fr.ouestfrance.querydsl.postgrest.annotations.Select;
-import fr.ouestfrance.querydsl.postgrest.model.Filter;
-import fr.ouestfrance.querydsl.postgrest.model.Page;
-import fr.ouestfrance.querydsl.postgrest.model.PageImpl;
-import fr.ouestfrance.querydsl.postgrest.model.Pageable;
+import fr.ouestfrance.querydsl.postgrest.model.*;
 import fr.ouestfrance.querydsl.postgrest.model.exceptions.MissingConfigurationException;
 import fr.ouestfrance.querydsl.postgrest.model.exceptions.PostgrestRequestException;
+import fr.ouestfrance.querydsl.postgrest.model.impl.CountFilter;
 import fr.ouestfrance.querydsl.postgrest.model.impl.OrderFilter;
 import fr.ouestfrance.querydsl.postgrest.model.impl.SelectFilter;
 import fr.ouestfrance.querydsl.postgrest.services.ext.PostgrestQueryProcessorService;
@@ -84,6 +82,17 @@ public Page<T> search(Object criteria, Pageable pageable) {
         return response;
     }
 
+
+    @Override
+    public long count(Object criteria) {
+        List<Filter> queryParams = processorService.process(criteria);
+        queryParams.add(CountFilter.of());
+        List<CountItem> response = client.count(annotation.resource(), toMap(queryParams));
+        // Retrieve result headers
+        return response.stream().findFirst().map(x -> x.get("count")).map(Long::valueOf).orElse(0L);
+    }
+
+
     @Override
     public List<T> upsert(List<Object> values) {
         return client.post(annotation.resource(), values, headerMap(Header.Method.UPSERT), clazz);

diff --git a/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/Repository.java b/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/Repository.java
@@ -24,6 +24,15 @@ default Page<T> search(Object criteria) {
         return search(criteria, Pageable.unPaged());
     }
 
+
+    /**
+     * Count from criteria object
+     *
+     * @param criteria search criteria
+     * @return count result
+     */
+    long count(Object criteria);
+
     /**
      * Search from criteria object with pagination
      *

diff --git a/...ostgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/builders/QueryFilterVisitor.java b/...ostgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/builders/QueryFilterVisitor.java
@@ -2,10 +2,7 @@
 
 import fr.ouestfrance.querydsl.postgrest.model.Filter;
 import fr.ouestfrance.querydsl.postgrest.model.Sort;
-import fr.ouestfrance.querydsl.postgrest.model.impl.CompositeFilter;
-import fr.ouestfrance.querydsl.postgrest.model.impl.OrderFilter;
-import fr.ouestfrance.querydsl.postgrest.model.impl.QueryFilter;
-import fr.ouestfrance.querydsl.postgrest.model.impl.SelectFilter;
+import fr.ouestfrance.querydsl.postgrest.model.impl.*;
 
 import java.util.List;
 import java.util.stream.Collectors;
@@ -63,6 +60,19 @@ public void visit(SelectFilter filter) {
                 .collect(Collectors.joining(",")));
     }
 
+
+    /**
+     * Transform a Select filter to Query
+     *
+     * @param filter select filter
+     */
+    public void visit(CountFilter filter) {
+        builder.append("count()");
+        if(filter.getGroupBy() != null) {
+            builder.append(COMA).append(filter.getGroupBy());
+        }
+    }
+
     /**
      * Transform a Composite filter to Query
      *

diff --git a/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/model/CountItem.java b/querydsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/model/CountItem.java
@@ -0,0 +1,12 @@
+package fr.ouestfrance.querydsl.postgrest.model;
+
+import lombok.*;
+
+import java.util.HashMap;
+
+@Getter
+@Setter
+@NoArgsConstructor
+@EqualsAndHashCode(callSuper = false)
+public class CountItem extends HashMap<String, String> {
+}
diff --git a/...dsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/model/impl/CountFilter.java b/...dsl-postgrest/src/main/java/fr/ouestfrance/querydsl/postgrest/model/impl/CountFilter.java
@@ -0,0 +1,73 @@
+package fr.ouestfrance.querydsl.postgrest.model.impl;
+
+import fr.ouestfrance.querydsl.postgrest.builders.FilterVisitor;
+import fr.ouestfrance.querydsl.postgrest.builders.QueryFilterVisitor;
+import fr.ouestfrance.querydsl.postgrest.model.Filter;
+import lombok.AccessLevel;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+
+/**
+ * Select filter allow to describe a selection
+ */
+@Getter
+@RequiredArgsConstructor(access = AccessLevel.PRIVATE)
+@AllArgsConstructor(access = AccessLevel.PRIVATE)
+public class CountFilter implements Filter, FilterVisitor {
+
+    /**
+     * Default query param key for selection
+     */
+    private static final String KEY_PARAMETER = "select";
+    /**
+     * list of fields
+     */
+    private String groupBy;
+
+
+    /**
+     * Create select filter from embedded resources
+     * @return select filter
+     */
+    public static Filter groupBy(String field) {
+        return new CountFilter(field);
+    }
+
+    /**
+     * Create select filter from embedded resources
+     *
+     * @return select filter
+     */
+    public static Filter of() {
+        return new CountFilter();
+    }
+
+    @Override
+    public void accept(QueryFilterVisitor visitor) {
+        visitor.visit(this);
+    }
+
+    @Override
+    public String getKey() {
+        return KEY_PARAMETER;
+    }
+
+
+    /**
+     * Attribute name
+     */
+    @Getter
+    @RequiredArgsConstructor
+    public static class Attribute {
+        /**
+         * alias
+         */
+        private final String alias;
+        /**
+         * value selected
+         */
+        private final String value;
+
+    }
+}