mirror of
https://github.com/juspay/hyperswitch.git
synced 2025-10-30 09:38:33 +08:00
64 lines
2.0 KiB
Rust
64 lines
2.0 KiB
Rust
use bb8::PooledConnection;
|
|
use diesel::PgConnection;
|
|
use error_stack::ResultExt;
|
|
use storage_impl::errors as storage_errors;
|
|
|
|
use crate::errors;
|
|
|
|
pub type PgPool = bb8::Pool<async_bb8_diesel::ConnectionManager<PgConnection>>;
|
|
|
|
pub type PgPooledConn = async_bb8_diesel::Connection<PgConnection>;
|
|
|
|
/// Creates a Redis connection pool for the specified Redis settings
|
|
/// # Panics
|
|
///
|
|
/// Panics if failed to create a redis pool
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn redis_connection(
|
|
conf: &crate::configs::Settings,
|
|
) -> redis_interface::RedisConnectionPool {
|
|
redis_interface::RedisConnectionPool::new(&conf.redis)
|
|
.await
|
|
.expect("Failed to create Redis Connection Pool")
|
|
}
|
|
|
|
pub async fn pg_connection_read<T: storage_impl::DatabaseStore>(
|
|
store: &T,
|
|
) -> errors::CustomResult<
|
|
PooledConnection<'_, async_bb8_diesel::ConnectionManager<PgConnection>>,
|
|
storage_errors::StorageError,
|
|
> {
|
|
// If only OLAP is enabled get replica pool.
|
|
#[cfg(all(feature = "olap", not(feature = "oltp")))]
|
|
let pool = store.get_replica_pool();
|
|
|
|
// If either one of these are true we need to get master pool.
|
|
// 1. Only OLTP is enabled.
|
|
// 2. Both OLAP and OLTP is enabled.
|
|
// 3. Both OLAP and OLTP is disabled.
|
|
#[cfg(any(
|
|
all(not(feature = "olap"), feature = "oltp"),
|
|
all(feature = "olap", feature = "oltp"),
|
|
all(not(feature = "olap"), not(feature = "oltp"))
|
|
))]
|
|
let pool = store.get_master_pool();
|
|
|
|
pool.get()
|
|
.await
|
|
.change_context(storage_errors::StorageError::DatabaseConnectionError)
|
|
}
|
|
|
|
pub async fn pg_connection_write<T: storage_impl::DatabaseStore>(
|
|
store: &T,
|
|
) -> errors::CustomResult<
|
|
PooledConnection<'_, async_bb8_diesel::ConnectionManager<PgConnection>>,
|
|
storage_errors::StorageError,
|
|
> {
|
|
// Since all writes should happen to master DB only choose master DB.
|
|
let pool = store.get_master_pool();
|
|
|
|
pool.get()
|
|
.await
|
|
.change_context(storage_errors::StorageError::DatabaseConnectionError)
|
|
}
|