Add parameter for db pool size

This commit is contained in:
Tobias Reisinger 2024-05-10 17:43:29 +02:00
parent b14049b3f6
commit f26e66d687
Signed by: serguzim
GPG key ID: 13AD60C237A28DFE
2 changed files with 3 additions and 3 deletions

View file

@ -33,14 +33,14 @@ pub async fn run_migrations(pool: &Pool<Sqlite>) -> Result<(), EmgauwaError> {
Ok(()) Ok(())
} }
pub async fn init(db: &str) -> Result<Pool<Sqlite>, EmgauwaError> { pub async fn init(db: &str, pool_size: u32) -> Result<Pool<Sqlite>, EmgauwaError> {
let options = SqliteConnectOptions::from_str(db)? let options = SqliteConnectOptions::from_str(db)?
.create_if_missing(true) .create_if_missing(true)
.log_statements(log::LevelFilter::Trace); .log_statements(log::LevelFilter::Trace);
let pool: Pool<Sqlite> = SqlitePoolOptions::new() let pool: Pool<Sqlite> = SqlitePoolOptions::new()
.acquire_timeout(std::time::Duration::from_secs(1)) .acquire_timeout(std::time::Duration::from_secs(1))
.max_connections(5) .max_connections(pool_size)
.connect_with(options) .connect_with(options)
.await?; .await?;

View file

@ -47,7 +47,7 @@ impl From<&EmgauwaError> for String {
EmgauwaError::Database(err) => String::from(err), EmgauwaError::Database(err) => String::from(err),
EmgauwaError::Uid(_) => String::from("the uid is in a bad format"), EmgauwaError::Uid(_) => String::from("the uid is in a bad format"),
EmgauwaError::Internal(_) => String::from("internal error"), EmgauwaError::Internal(_) => String::from("internal error"),
EmgauwaError::Connection(_) => String::from("the target controller is not connected"), EmgauwaError::Connection(uid) => format!("unable to connect to controller with uid: {}", uid),
EmgauwaError::Other(err) => format!("other error: {}", err), EmgauwaError::Other(err) => format!("other error: {}", err),
EmgauwaError::Hardware(err) => format!("hardware error: {}", err), EmgauwaError::Hardware(err) => format!("hardware error: {}", err),
} }