Projects/mysqladm-rs
Projects
/
mysqladm-rs
12
2
Fork 0
Code Issues 36 Pull Requests Releases Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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