0046a02a4a
* dont migrate on start. just connect * it compiles * that log is loud * get transaction from a local clone * import the trait * put the test app into a thread instead of tokio::spawn * fix one test * try db before rpcs * db connection is too slow. need to wait for it * do db setup once while spawning
56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
use crate::{errors::Web3ProxyError, relational_db::DatabaseReplica};
|
|
use derivative::Derivative;
|
|
use migration::{
|
|
sea_orm::{DatabaseConnection, DatabaseTransaction, TransactionTrait},
|
|
DbErr,
|
|
};
|
|
use std::sync::{Arc, LazyLock};
|
|
use tokio::sync::RwLock as AsyncRwLock;
|
|
|
|
pub static DB_CONN: LazyLock<AsyncRwLock<Result<DatabaseConnection, DatabaseError>>> =
|
|
LazyLock::new(|| AsyncRwLock::new(Err(DatabaseError::NotConfigured)));
|
|
|
|
pub static DB_REPLICA: LazyLock<AsyncRwLock<Result<DatabaseReplica, DatabaseError>>> =
|
|
LazyLock::new(|| AsyncRwLock::new(Err(DatabaseError::NotConfigured)));
|
|
|
|
#[derive(Clone, Debug, Derivative)]
|
|
pub enum DatabaseError {
|
|
/// no database configured. depending on what you need, this may or may not be a problem
|
|
NotConfigured,
|
|
/// an error that happened when creating the connection pool
|
|
Connect(Arc<DbErr>),
|
|
/// an error that just happened
|
|
Begin(Arc<DbErr>),
|
|
}
|
|
|
|
impl From<DatabaseError> for Web3ProxyError {
|
|
fn from(value: DatabaseError) -> Self {
|
|
match value {
|
|
DatabaseError::NotConfigured => Self::NoDatabaseConfigured,
|
|
DatabaseError::Connect(err) | DatabaseError::Begin(err) => Self::DatabaseArc(err),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub async fn global_db_conn() -> Result<DatabaseConnection, DatabaseError> {
|
|
DB_CONN.read().await.clone()
|
|
}
|
|
|
|
#[inline]
|
|
pub async fn global_db_transaction() -> Result<DatabaseTransaction, DatabaseError> {
|
|
let x = global_db_conn().await?;
|
|
|
|
let x = x
|
|
.begin()
|
|
.await
|
|
.map_err(|x| DatabaseError::Begin(Arc::new(x)))?;
|
|
|
|
Ok(x)
|
|
}
|
|
|
|
#[inline]
|
|
pub async fn global_db_replica_conn() -> Result<DatabaseReplica, DatabaseError> {
|
|
DB_REPLICA.read().await.clone()
|
|
}
|