feat: user relation

This commit is contained in:
Maulana Sodiqin
2025-03-14 17:50:37 +07:00
parent 9ffe808797
commit 7ab036f5f7
22 changed files with 358 additions and 28 deletions
+41 -8
View File
@@ -1,7 +1,8 @@
use crate::{v1::UsersItemDto, AppState, ResourceEnum};
use std::error::Error;
use crate::{v1::UsersItemDto, AppState, RedisKeyEnum, ResourceEnum};
use anyhow::{bail, Result};
use redis::Commands;
use super::AuthRegisterRequestDto;
use super::{AuthQueryByEmailResponse, AuthRegisterRequestDto};
pub struct AuthRepository<'a> {
state: &'a AppState,
@@ -12,24 +13,56 @@ impl<'a> AuthRepository<'a> {
Self { state }
}
pub fn query_store_user_data(
&self,
user: AuthRegisterRequestDto,
) -> Result<String> {
let redis_key = format!("{}:{}", RedisKeyEnum::User, user.email.clone());
match &self.state.redisdb.get_connection().and_then(|mut conn| {
conn.set_ex::<_, String, ()>(
&redis_key,
serde_json::to_string(&user).unwrap_or_default(),
86400,
)
}) {
Ok(_) => Ok("Success store user data".to_string()),
Err(err) => Ok(format!("Redis storage failed: {}", err)),
}
}
pub fn query_get_stored_user(&self, email: String) -> Result<UsersItemDto> {
let redis_key = format!("{}:{}", RedisKeyEnum::User, email);
let mut conn = self.state.redisdb.get_connection()?;
let data: Option<String> = conn.get(&redis_key)?;
match data {
Some(user_json) => {
let user: UsersItemDto = serde_json::from_str(&user_json)?;
Ok(user)
}
None => bail!("No stored user data found for email"),
}
}
pub async fn query_user_by_email(
&self,
email: String,
) -> Result<AuthRegisterRequestDto, Box<dyn Error>> {
) -> Result<AuthQueryByEmailResponse> {
let db = &self.state.surrealdb;
let result = db.select((ResourceEnum::Users.to_string(), email)).await?;
match result {
Some(user) => Ok(user),
None => Err("User not found for email".into()),
Some(response) => Ok(response),
None => bail!("User not found"),
}
}
pub async fn query_create_user(
&self,
data: AuthRegisterRequestDto,
) -> Result<String, Box<dyn Error>> {
) -> Result<String> {
let db = &self.state.surrealdb;
let record: Option<UsersItemDto> = db
@@ -39,7 +72,7 @@ impl<'a> AuthRepository<'a> {
match record {
Some(_) => Ok("Success create user".into()),
None => Err("Failed to create user".into()),
None => bail!("Failed to create user"),
}
}
}