feat(schema): cursor pagination

1 files changed, 8 insertions, 34 deletions

diff --git a/lib/api-config/src/schema/implementation.rs b/lib/api-config/src/schema/implementation.rs
index c414879..ca0757f 100644
--- a/lib/api-config/src/schema/implementation.rs
+++ b/lib/api-config/src/schema/implementation.rs
@@ -1,7 +1,10 @@
 use async_trait::async_trait;
 use tracing::debug;
+use warden_core::pagination::{Connection, PaginationArgs};
 
-use crate::schema::{self, SchemaDriver, SchemaService, TransactionSchema};
+use crate::schema::{
+    self, SchemaDriver, SchemaService, TransactionSchema, pagination::DecodedSchemaPagination,
+};
 
 #[async_trait]
 impl SchemaDriver for SchemaService {
@@ -44,41 +47,12 @@ impl SchemaDriver for SchemaService {
     }
 
     #[tracing::instrument(skip(self))]
-    async fn get_schemas(
+    async fn list_schemas(
         &self,
+        input: &PaginationArgs,
         limit: i64,
-        first: Option<i64>,
-        after: Option<&str>,
-    ) -> Result<Vec<TransactionSchema>, crate::ConfigurationError> {
+    ) -> Result<Connection<TransactionSchema>, crate::ConfigurationError> {
         debug!("getting transaction schemas");
-        let limit = first.unwrap_or(limit);
-        let mut last_type = String::default();
-        let mut last_version = String::default();
-
-        if let Some(s) = after {
-            let parts: Vec<&str> = s.split(',').collect();
-            if parts.len() == 2 {
-                last_type = parts[0].to_string();
-                last_version = parts[1].to_string();
-            }
-        }
-
-        let rows = sqlx::query_as!(
-            TransactionSchema,
-            "
-            select *
-            from transaction_schema
-            where ($1 = '' or (schema_type, schema_version) > ($1, $2))
-            order by schema_type asc, schema_version asc
-            limit $3
-        ",
-            &last_type,
-            &last_version,
-            limit + 1
-        )
-        .fetch_all(&self.database)
-        .await?;
-
-        Ok(rows)
+        schema::list_schemas::list_schemas(self, input, limit).await
     }
 }