Add pagination

Query is `users(limit: X offset Y)` with defaults X=20 Y=0

2 files changed, 16 insertions, 4 deletions

diff --git a/src/graphql/query.rs b/src/graphql/query.rs
index 254dab6..0e19771 100644
--- a/src/graphql/query.rs
+++ b/src/graphql/query.rs
@@ -10,7 +10,12 @@ impl Query {
     }
 
     /// Returns all the users
-    async fn users<'ctx>(&self, ctx: &Context<'ctx>) -> Result<Option<Vec<user::User>>, String> {
-        user::get_users(ctx).await
+    async fn users<'ctx>(
+        &self,
+        ctx: &Context<'ctx>,
+        #[graphql(desc = "Limit results")] limit: Option<i64>,
+        #[graphql(desc = "Offset results")] offset: Option<i64>,
+    ) -> Result<Option<Vec<user::User>>, String> {
+        user::get_users(ctx, limit, offset).await
     }
 }
diff --git a/src/graphql/types/user.rs b/src/graphql/types/user.rs
index b675f1f..21554b9 100644
--- a/src/graphql/types/user.rs
+++ b/src/graphql/types/user.rs
@@ -31,7 +31,11 @@ impl User {
     }
 }
 
-pub async fn get_users<'ctx>(ctx: &Context<'ctx>) -> Result<Option<Vec<User>>, String> {
+pub async fn get_users<'ctx>(
+    ctx: &Context<'ctx>,
+    limit: Option<i64>,
+    offset: Option<i64>,
+) -> Result<Option<Vec<User>>, String> {
     let state = ctx.data::<AppState>().expect("Can't connect to db");
     let client = &*state.client;
     let auth: &Authentication = ctx.data().unwrap();
@@ -39,7 +43,10 @@ pub async fn get_users<'ctx>(ctx: &Context<'ctx>) -> Result<Option<Vec<User>>, S
         Authentication::NotLogged => Err("Unauthorized".to_string()),
         Authentication::Logged(_claims) => {
             let rows = client
-                .query("SELECT id, email, password, is_admin FROM users", &[])
+                .query(
+                    "SELECT id, email, password, is_admin FROM users LIMIT $1 OFFSET $2",
+                    &[&limit.unwrap_or(20), &offset.unwrap_or(0)],
+                )
                 .await
                 .unwrap();