added waiting (#13)

src/config.rs

@@ -25,6 +25,16 @@ pub fn database_token() -> Option<String> {
     None
 }
 
+pub fn wait_timeout() -> usize {
+    if let Ok(v) = env::var("DATABASE_WAIT_TIMEOUT") {
+        if let Ok(v) = v.parse::<usize>() {
+            return v;
+        }
+    }
+
+    0
+}
+
 pub enum Database {
     LibSQL,
     Postgres,

src/database_drivers/libsql.rs

@@ -21,6 +21,7 @@ impl<'a> LibSQLDriver {
             Ok(c) => c,
             Err(err) => bail!("{:?}", err),
         };
+
         Ok(LibSQLDriver { db: client })
     }
 }

src/database_drivers/maria.rs

@@ -1,3 +1,4 @@
+use crate::config;
 use crate::database_drivers::DatabaseDriver;
 use anyhow::Result;
 use sqlx::mysql::MySqlRow;
@@ -11,7 +12,24 @@ pub struct MariaDBDriver {
 
 impl<'a> MariaDBDriver {
     pub async fn new<'b>(db_url: &str) -> Result<MariaDBDriver> {
-        let client = MySqlConnection::connect(db_url).await?;
+        let mut client = MySqlConnection::connect(db_url).await?;
+
+        let wait_timeout = config::wait_timeout();
+
+        let mut count = 0;
+        loop {
+            if count > wait_timeout {
+                break;
+            }
+
+            if client.ping().await.is_ok() {
+                break;
+            }
+
+            count += 1;
+            tokio::time::sleep(tokio::time::Duration::from_secs(1)).await;
+        }
+
         Ok(MariaDBDriver { db: client })
     }
 }

src/database_drivers/mysql.rs

@@ -1,3 +1,4 @@
+use crate::config;
 use crate::database_drivers::DatabaseDriver;
 use anyhow::Result;
 use sqlx::mysql::MySqlRow;
@@ -11,7 +12,23 @@ pub struct MySQLDriver {
 
 impl<'a> MySQLDriver {
     pub async fn new<'b>(db_url: &str) -> Result<MySQLDriver> {
-        let client = MySqlConnection::connect(db_url).await?;
+        let mut client = MySqlConnection::connect(db_url).await?;
+
+        let wait_timeout = config::wait_timeout();
+        let mut count = 0;
+        loop {
+            if count > wait_timeout {
+                break;
+            }
+
+            if client.ping().await.is_ok() {
+                break;
+            }
+
+            count += 1;
+            tokio::time::sleep(tokio::time::Duration::from_secs(1)).await;
+        }
+
         Ok(MySQLDriver { db: client })
     }
 }

src/database_drivers/postgres.rs

@@ -1,3 +1,4 @@
+use crate::config;
 use crate::database_drivers::DatabaseDriver;
 use anyhow::Result;
 use sqlx::postgres::PgRow;
@@ -11,7 +12,23 @@ pub struct PostgresDriver {
 
 impl<'a> PostgresDriver {
     pub async fn new<'b>(db_url: &str) -> Result<PostgresDriver> {
-        let client = PgConnection::connect(db_url).await?;
+        let mut client = PgConnection::connect(db_url).await?;
+
+        let wait_timeout = config::wait_timeout();
+        let mut count = 0;
+        loop {
+            if count > wait_timeout {
+                break;
+            }
+
+            if client.ping().await.is_ok() {
+                break;
+            }
+
+            count += 1;
+            tokio::time::sleep(tokio::time::Duration::from_secs(1)).await;
+        }
+
         Ok(PostgresDriver { db: client })
     }
 }