Added endpoint for getting list of objects for custom get

clintonbosch · clintonbosch · commit b8942b7170df · 2021-04-06T08:48:56.000+02:00
diff --git a/pom.xml b/pom.xml
@@ -4,7 +4,7 @@
 
     <groupId>com.afrozaar.wordpress</groupId>
     <artifactId>wp-api-v2-client-java</artifactId>
-    <version>4.15.2-SNAPSHOT</version>
+    <version>4.15.3-SNAPSHOT</version>
 
     <packaging>jar</packaging>
 
diff --git a/src/main/java/com/afrozaar/wordpress/wpapi/v2/Client.java b/src/main/java/com/afrozaar/wordpress/wpapi/v2/Client.java
@@ -864,6 +864,18 @@ public Object getCustom(String customPath, String context, Class clazz) throws N
         }
     }
 
+    @Override
+    public <T> List<T> getCustomObjects(String customPath, String context, Class clazz) throws NotFoundException {
+        List<T> collected = new ArrayList<>();
+        PagedResponse<T> response = this.getPagedResponse(customPath, clazz, context);
+        collected.addAll(response.getList());
+        while (response.hasNext()) {
+            response = traverse(response, PagedResponse.NEXT);
+            collected.addAll(response.getList());
+        }
+        return collected;
+    }
+
     @Override
     public void putCustom(String customPath) {
         doExchange1(customPath, HttpMethod.PUT, null, new Object[0], null, null).getBody();
diff --git a/src/main/java/com/afrozaar/wordpress/wpapi/v2/api/CustomCalls.java b/src/main/java/com/afrozaar/wordpress/wpapi/v2/api/CustomCalls.java
@@ -6,6 +6,7 @@
 import org.springframework.http.ResponseEntity;
 
 import javax.annotation.Nullable;
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -15,6 +16,8 @@ public interface CustomCalls {
 
     Object getCustom(String customPath, String context, Class clazz) throws NotFoundException;
 
+    <T> List<T> getCustomObjects(String customPath, String context, Class clazz) throws NotFoundException;
+
     void putCustom(String customPath);
 
     <T, B> ResponseEntity<T> doCustomExchange(String context, HttpMethod method, Class<T> typeRef, Object[] buildAndExpand,
