feat: support db migration

fan-tastic-z · May 30, 2024 · d2c3433 · d2c3433
1 parent 7993b84
commit d2c3433
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/app.rs b/src/app.rs
@@ -1,4 +1,5 @@
 use lazy_static::lazy_static;
+use migration::MigratorTrait;
 use sea_orm::DatabaseConnection;
 use std::{collections::HashMap, sync::Arc, time::Duration};
 use tokio::{task::JoinSet, time};
@@ -179,6 +180,11 @@ impl WatchVulnApp {
         }
         is_push
     }
+
+    pub async fn run_migration(&self) -> Result<()> {
+        migration::Migrator::up(&self.app_context.db, None).await?;
+        Ok(())
+    }
 }
 
 #[derive(Clone)]

diff --git a/src/main.rs b/src/main.rs
@@ -11,6 +11,7 @@ async fn main() -> Result<()> {
     let app_context = Arc::new(create_context(&environment).await?);
     logger::init(&app_context.config.logger);
     let app = WatchVulnApp::new(app_context);
+    app.run_migration().await?;
     app.run().await?;
     Ok(())
 }