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 87 88 89 90 91 92 93 94 95
use crate::model::auth_account::{AuthAccountData, AuthAccountModel}; 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_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_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>; }