Add: Graphql queries for retrieving attendance

Author: Wreck-X

Cargo.lock

@@ -4,9 +4,10 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-async-graphql = "7.0.6"
+async-graphql = { version = "7.0.6", features = ["chrono"] }
 async-graphql-axum = "7.0.6"
 axum = "0.7.3"
+chrono = "0.4.38"
 serde = { version = "1.0.188", features = ["derive"] }
 shuttle-axum = "0.46.0"
 shuttle-runtime = "0.46.0"

Cargo.toml

@@ -0,0 +1,21 @@
+-- migrations/YYYYMMDDHHMMSS_create_member_table.sql
+
+CREATE TABLE Member (
+    id SERIAL PRIMARY KEY,
+    rollno VARCHAR(20) NOT NULL,
+    name VARCHAR(255) NOT NULL,
+    hostel VARCHAR(255) NOT NULL,
+    email VARCHAR(255) NOT NULL UNIQUE,
+    sex VARCHAR(10) NOT NULL,
+    year INT NOT NULL,
+    created_at TIMESTAMP DEFAULT NOW()
+);
+
+CREATE TABLE Attendance (
+    id INT REFERENCES Member(id),
+    date DATE NOT NULL,
+    timein TIME NOT NULL,
+    timeout TIME NOT NULL,
+    PRIMARY KEY (id, date)
+);
+

migrations/20240626151806_create_members.sql

@@ -0,0 +1,10 @@
+use sqlx::FromRow;
+use async_graphql::SimpleObject;
+
+#[derive(FromRow, SimpleObject)]
+pub struct Attendance {
+    pub id: i32,
+    pub date: String,
+    pub timein: String,
+    pub timeout: String,
+}

migrations/20240627145806_create_member_table.sql

@@ -0,0 +1,13 @@
+use sqlx::FromRow;
+use async_graphql::SimpleObject;
+
+#[derive(FromRow, SimpleObject)]
+pub struct Member {
+    pub id: i32,
+    pub rollno: String,
+    pub name: String,
+    pub hostel: String,
+    pub email: String,
+    pub sex: String,
+    pub year: i32,
+}

src/db/attendance.rs

@@ -0,0 +1,2 @@
+pub mod member;
+pub mod attendance;

src/db/member.rs

@@ -0,0 +1,2 @@
+pub mod mutations;
+pub mod query;

src/db/mod.rs

@@ -0,0 +1,37 @@
+use async_graphql::{Context, Object};
+use sqlx::PgPool;
+use std::sync::Arc;
+
+use crate::db::member::Member;
+
+pub struct MutationRoot;
+
+#[Object]
+impl MutationRoot {
+    async fn add_member(
+        &self, 
+        ctx: &Context<'_>, 
+        rollno: String, 
+        name: String, 
+        hostel: String, 
+        email: String, 
+        sex: String, 
+        year: i32
+    ) -> Result<Member, sqlx::Error> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool not found in context");
+
+        let member = sqlx::query_as::<_, Member>(
+            "INSERT INTO Member (rollno, name, hostel, email, sex, year) VALUES ($1, $2, $3, $4, $5, $6) RETURNING *"
+        )
+        .bind(rollno)
+        .bind(name)
+        .bind(hostel)
+        .bind(email)
+        .bind(sex)
+        .bind(year)
+        .fetch_one(pool.as_ref())
+        .await?;
+
+        Ok(member)
+    }
+}

src/graphql/mod.rs

@@ -0,0 +1,27 @@
+use async_graphql::{Context, Object};
+use sqlx::PgPool;
+use std::sync::Arc;
+
+use crate::db::{member::Member, attendance::Attendance};
+
+pub struct QueryRoot;
+
+#[Object]
+impl QueryRoot {
+    async fn get_users(&self, ctx: &Context<'_>) -> Result<Vec<Member>, sqlx::Error> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool not found in context");
+        let users = sqlx::query_as::<_, Member>("SELECT * FROM Member")
+            .fetch_all(pool.as_ref())
+            .await?;
+        Ok(users)
+    }
+
+    async fn get_attendance(&self, ctx: &Context<'_>, date: String) -> Result<Vec<Attendance>, sqlx::Error> {
+        let pool = ctx.data::<Arc<PgPool>>().expect("Pool not found in context");
+        let attendance_list = sqlx::query_as::<_, Attendance>("SELECT id, date, timein, timeout FROM Attendance WHERE date = $1")
+            .bind(date)
+            .fetch_all(pool.as_ref())
+            .await?;
+        Ok(attendance_list)
+    }
+}