Add timeout to mysql connection

This commit is contained in:
Oystein Kristoffer Tveit 2024-04-26 00:23:59 +02:00
parent 2f039c0b1d
commit 561241d589
Signed by: oysteikt
GPG Key ID: 9F2F7D8250F35146
1 changed files with 17 additions and 11 deletions

View File

@ -1,6 +1,6 @@
use std::{fs, path::PathBuf};
use std::{fs, path::PathBuf, time::Duration};
use anyhow::Context;
use anyhow::{anyhow, Context};
use clap::Parser;
use serde::{Deserialize, Serialize};
use sqlx::{mysql::MySqlConnectOptions, ConnectOptions, MySqlConnection};
@ -110,13 +110,19 @@ pub fn get_config(args: ConfigOverrideArgs) -> anyhow::Result<Config> {
/// TODO: Add timeout.
pub async fn mysql_connection_from_config(config: Config) -> anyhow::Result<MySqlConnection> {
match tokio::time::timeout(
Duration::from_secs(2),
MySqlConnectOptions::new()
.host(&config.mysql.host)
.username(&config.mysql.username)
.password(&config.mysql.password)
.port(config.mysql.port.unwrap_or(3306))
.database("mysql")
.connect()
.await
.context("Failed to connect to MySQL")
.connect(),
).await {
Ok(conn) => conn.context("Failed to connect to MySQL"),
Err(_) => {
Err(anyhow!("Timed out after 2 seconds")).context("Failed to connect to MySQL")
}
}
}