mirror of
https://github.com/juspay/hyperswitch.git
synced 2025-11-03 13:30:39 +08:00
91519d8462
Co-authored-by: hyperswitch-bot[bot] <148525504+hyperswitch-bot[bot]@users.noreply.github.com>
72 lines
2.3 KiB
Rust
72 lines
2.3 KiB
Rust
use bb8::PooledConnection;
|
|
use diesel::PgConnection;
|
|
#[cfg(feature = "aws_kms")]
|
|
use external_services::aws_kms::{self, decrypt::AwsKmsDecrypt};
|
|
#[cfg(feature = "hashicorp-vault")]
|
|
use external_services::hashicorp_vault::{self, decrypt::VaultFetch, Kv2};
|
|
#[cfg(not(feature = "aws_kms"))]
|
|
use masking::PeekInterface;
|
|
|
|
use crate::settings::Database;
|
|
|
|
pub type PgPool = bb8::Pool<async_bb8_diesel::ConnectionManager<PgConnection>>;
|
|
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn redis_connection(
|
|
conf: &crate::settings::Settings,
|
|
) -> redis_interface::RedisConnectionPool {
|
|
redis_interface::RedisConnectionPool::new(&conf.redis)
|
|
.await
|
|
.expect("Failed to create Redis connection Pool")
|
|
}
|
|
|
|
// TODO: use stores defined in storage_impl instead
|
|
/// # Panics
|
|
///
|
|
/// Will panic if could not create a db pool
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn diesel_make_pg_pool(
|
|
database: &Database,
|
|
_test_transaction: bool,
|
|
#[cfg(feature = "aws_kms")] aws_kms_client: &'static aws_kms::AwsKmsClient,
|
|
#[cfg(feature = "hashicorp-vault")] hashicorp_client: &'static hashicorp_vault::HashiCorpVault,
|
|
) -> PgPool {
|
|
let password = database.password.clone();
|
|
#[cfg(feature = "hashicorp-vault")]
|
|
let password = password
|
|
.fetch_inner::<Kv2>(hashicorp_client)
|
|
.await
|
|
.expect("Failed while fetching db password");
|
|
|
|
#[cfg(feature = "aws_kms")]
|
|
let password = password
|
|
.decrypt_inner(aws_kms_client)
|
|
.await
|
|
.expect("Failed to decrypt password");
|
|
|
|
#[cfg(not(feature = "aws_kms"))]
|
|
let password = &password.peek();
|
|
|
|
let database_url = format!(
|
|
"postgres://{}:{}@{}:{}/{}",
|
|
database.username, password, database.host, database.port, database.dbname
|
|
);
|
|
let manager = async_bb8_diesel::ConnectionManager::<PgConnection>::new(database_url);
|
|
let pool = bb8::Pool::builder()
|
|
.max_size(database.pool_size)
|
|
.connection_timeout(std::time::Duration::from_secs(database.connection_timeout));
|
|
|
|
pool.build(manager)
|
|
.await
|
|
.expect("Failed to create PostgreSQL connection pool")
|
|
}
|
|
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn pg_connection(
|
|
pool: &PgPool,
|
|
) -> PooledConnection<'_, async_bb8_diesel::ConnectionManager<PgConnection>> {
|
|
pool.get()
|
|
.await
|
|
.expect("Couldn't retrieve PostgreSQL connection")
|
|
}
|