mirrors/mCaptcha
1
0
Fork 0
mirror of https://github.com/mCaptcha/mCaptcha.git synced 2025-03-14 13:08:27 +03:00
Code Issues Projects Releases Packages Wiki Activity

feat: load db_* db connection

Browse source
This commit is contained in:
realaravinth 2022-05-10 22:59:42 +05:30
parent 1d8554cb36
commit 79506a93b9
No known key found for this signature in database
GPG key ID: AD9F0F08E855ED88
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/data.rs
View file

@ -19,6 +19,8 @@ use std::thread;
use actix::prelude::*;
use argon2_creds::{Config, ConfigBuilder, PasswordPolicy};
use db_core::prelude::*;
use db_sqlx_postgres::{ConnectionOptions, Fresh};
use lettre::transport::smtp::authentication::Mechanism;
use lettre::{
transport::smtp::authentication::Credentials, AsyncSmtpTransport, Tokio1Executor,
@ -41,6 +43,8 @@ use libmcaptcha::{
use sqlx::postgres::PgPoolOptions;
use sqlx::PgPool;
use db_core::MCDatabase;
use crate::errors::ServiceResult;
use crate::SETTINGS;
@ -142,6 +146,8 @@ impl SystemGroup {
pub struct Data {
/// databse pool
pub db: PgPool,
/// database ops defined by db crates
pub dblib: Box<dyn MCDatabase>,
/// credential management configuration
pub creds: Config,
/// mCaptcha system: Redis cache, etc.
@ -179,9 +185,19 @@ impl Data {
.await
.expect("Unable to form database pool");
let settings = &SETTINGS;
let pool = settings.database.pool;
let pool_options = PgPoolOptions::new().max_connections(pool);
let connection_options = ConnectionOptions::Fresh(Fresh {
pool_options,
url: settings.database.url.clone(),
});
let dblib = Box::new(connection_options.connect().await.unwrap());
let data = Data {
creds,
db,
dblib,
captcha: SystemGroup::new().await,
mailer: Self::get_mailer(),
};