1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
use crate::model::auth_account::{AuthAccountData, AuthAccountModel, AuthAccountStatus}; use crate::model::token::{TokenData, TokenModel}; use c3p0::*; use lightspeed_core::error::LightSpeedError; pub mod pg; #[async_trait::async_trait] pub trait AuthRepositoryManager: Clone + Send + Sync { type Conn: SqlConnection; type C3P0: C3p0Pool<Conn = Self::Conn>; type AuthAccountRepo: AuthAccountRepository<Conn = Self::Conn>; type TokenRepo: TokenRepository<Conn = Self::Conn>; fn c3p0(&self) -> &Self::C3P0; async fn start(&self) -> Result<(), LightSpeedError>; fn auth_account_repo(&self) -> Self::AuthAccountRepo; fn token_repo(&self) -> Self::TokenRepo; } #[async_trait::async_trait] pub trait AuthAccountRepository: Clone + Send + Sync { type Conn: SqlConnection; async fn fetch_all_by_status( &self, conn: &mut Self::Conn, status: AuthAccountStatus, start_user_id: i64, limit: u32, ) -> Result<Vec<AuthAccountModel>, LightSpeedError>; async fn fetch_by_id(&self, conn: &mut Self::Conn, user_id: i64) -> Result<AuthAccountModel, LightSpeedError>; async fn fetch_by_username( &self, conn: &mut Self::Conn, username: &str, ) -> Result<AuthAccountModel, LightSpeedError>; async fn fetch_by_username_optional( &self, conn: &mut Self::Conn, username: &str, ) -> Result<Option<AuthAccountModel>, LightSpeedError>; async fn fetch_by_email_optional( &self, conn: &mut Self::Conn, email: &str, ) -> Result<Option<AuthAccountModel>, LightSpeedError>; async fn save( &self, conn: &mut Self::Conn, model: NewModel<AuthAccountData>, ) -> Result<AuthAccountModel, LightSpeedError>; async fn update( &self, conn: &mut Self::Conn, model: Model<AuthAccountData>, ) -> Result<AuthAccountModel, LightSpeedError>; async fn delete(&self, conn: &mut Self::Conn, model: AuthAccountModel) -> Result<AuthAccountModel, LightSpeedError>; async fn delete_by_id(&self, conn: &mut Self::Conn, user_id: i64) -> Result<u64, LightSpeedError>; } #[async_trait::async_trait] pub trait TokenRepository: Clone + Send + Sync { type Conn: SqlConnection; async fn fetch_by_token(&self, conn: &mut Self::Conn, token_string: &str) -> Result<TokenModel, LightSpeedError>; async fn fetch_by_username( &self, conn: &mut Self::Conn, username: &str, ) -> Result<Vec<TokenModel>, LightSpeedError>; async fn save(&self, conn: &mut Self::Conn, model: NewModel<TokenData>) -> Result<TokenModel, LightSpeedError>; async fn delete(&self, conn: &mut Self::Conn, model: TokenModel) -> Result<TokenModel, LightSpeedError>; }