feature #8: Pagination for Query & Function

src/test/kotlin/fr/postgresjson/RequesterTest.kt

@@ -2,8 +2,10 @@ package fr.postgresjson
 
 import com.github.jasync.sql.db.QueryResult
 import com.github.jasync.sql.db.util.isCompleted
+import fr.postgresjson.connexion.Paginated
 import fr.postgresjson.connexion.Requester
 import fr.postgresjson.entity.IdEntity
+import org.junit.Assert
 import org.junit.jupiter.api.Assertions
 import org.junit.jupiter.api.Assertions.assertEquals
 import org.junit.jupiter.api.Test
@@ -93,4 +95,32 @@ class RequesterTest: TestAbstract() {
 
         assertEquals("myName", obj!![0].name)
     }
+
+    @Test
+    fun `call select paginated on query`() {
+        val resources = File(this::class.java.getResource("/sql/query").toURI())
+        val result: Paginated<ObjTest> = Requester(getConnextion())
+            .addQuery(resources)
+            .getQuery("Test/selectPaginated")
+            .select(1, 2, mapOf("name" to "ff"))
+        Assert.assertNotNull(result)
+        Assert.assertEquals(result.result[0].name, "ff")
+        Assert.assertEquals(result.result[1].name, "ff-2")
+        Assert.assertEquals(result.total, 10)
+        Assert.assertEquals(result.offset, 0)
+    }
+
+    @Test
+    fun `call select paginated on function`() {
+        val resources = File(this::class.java.getResource("/sql/function").toURI())
+        val result: Paginated<ObjTest> = Requester(getConnextion())
+            .addFunction(resources)
+            .getFunction("test_function_paginated")
+            .select(1, 2, mapOf("name" to "ff"))
+        Assert.assertNotNull(result)
+        Assert.assertEquals(result.result[0].name, "ff")
+        Assert.assertEquals(result.result[1].name, "ff-2")
+        Assert.assertEquals(result.total, 10)
+        Assert.assertEquals(result.offset, 0)
+    }
 }

src/test/resources/fixtures/init.sql

@@ -48,4 +48,19 @@ BEGIN
             json_build_object('id', 4, 'name', hi)
         );
 END;
-$$;
+$$;
+
+CREATE OR REPLACE FUNCTION test_function_paginated (name text default 'plop', IN "limit" int default 10, IN "offset" int default 0, out result json, out total int)
+    LANGUAGE plpgsql
+AS
+$$
+BEGIN
+    SELECT json_build_array(
+                   json_build_object('id', 3, 'name', name::text),
+                   json_build_object('id', 4, 'name', name::text || '-2')
+               ),
+           10
+    INTO result, total
+    LIMIT "limit" OFFSET "offset";
+END;
+$$

src/test/resources/sql/function/Test/test_function_paginated.sql

@@ -0,0 +1,14 @@
+CREATE OR REPLACE FUNCTION test_function_paginated (name text default 'plop', IN "limit" int default 10, IN "offset" int default 0, out result json, out total int)
+LANGUAGE plpgsql
+AS
+$$
+BEGIN
+    SELECT json_build_array(
+        json_build_object('id', 3, 'name', name::text),
+        json_build_object('id', 4, 'name', name::text || '-2')
+    ),
+    10
+    INTO result, total
+    LIMIT "limit" OFFSET "offset";
+END;
+$$

src/test/resources/sql/query/Test/selectPaginated.sql

@@ -0,0 +1,5 @@
+SELECT json_build_array(
+   json_build_object('id', 3, 'name', :name::text),
+   json_build_object('id', 4, 'name', :name::text || '-2')
+), 10 as total
+LIMIT :limit OFFSET :offset