Migrations behind env variable & stats reset migration

Author: chatasma

src/database/migrations/mod.rs

@@ -1,7 +1,9 @@
 use crate::database::Database;
 use crate::database::migrations::denormalize_ip_identities::DenormalizeIpIdentitiesMigration;
+use crate::database::migrations::reset_stats::ResetStatsMigration;
 
 pub mod denormalize_ip_identities;
+mod reset_stats;
 
 #[async_trait]
 pub trait DatabaseMigration {
@@ -17,17 +19,26 @@ impl MigrationExecutor {
     pub fn new() -> Self {
         let denormalize_ip_identities_migration =
             Box::new(DenormalizeIpIdentitiesMigration {});
+        let reset_stats_migration =
+            Box::new(ResetStatsMigration {});
         Self {
-            migrations: vec![denormalize_ip_identities_migration]
+            migrations: vec![
+                denormalize_ip_identities_migration,
+                reset_stats_migration
+            ]
         }
     }
 
-    pub async fn execute_migration_by_name(&self, database: &Database, name: String) {
+    pub async fn execute_migration_by_name(&self, database: &Database, name: String) -> bool {
         match self.migrations.iter().find(|migration| migration.get_id() == name) {
             Some(migration) => {
-                migration.perform(database).await
+                migration.perform(database).await;
+                true
+            }
+            None => {
+                warn!("Could not find migration '{}'", name);
+                false
             }
-            None => {}
         }
     }
 }

src/database/migrations/reset_stats.rs

@@ -0,0 +1,34 @@
+use mongodb::bson::doc;
+use mongodb::options::UpdateOptions;
+use crate::database::Database;
+use crate::database::migrations::DatabaseMigration;
+
+pub struct ResetStatsMigration {}
+
+// resets stats for players, in specific, set stats objects to empty objects
+#[async_trait]
+impl DatabaseMigration for ResetStatsMigration {
+    fn get_id(&self) -> String {
+        String::from("reset_stats")
+    }
+
+    async fn perform(&self, database: &Database) {
+        info!("Resetting all player stats...");
+        let update_result = database.players.update_many(
+        doc! {},
+        doc! { "$set": {"stats": {}}},
+        None
+        ).await;
+        match update_result {
+            Ok(result) => {
+                info!(
+                    "Successfully reset player statistics, {} documents were modified",
+                    result.modified_count
+                );
+            }
+            Err(err) => {
+                warn!("Could not reset stats: {}", err);
+            }
+        };
+    }
+}

src/main.rs

@@ -168,8 +168,21 @@ async fn main() -> Result<(), String> {
         leaderboards
     };
 
-    // let migration_executor = MigrationExecutor::new();
-    // migration_executor.execute_migration_by_name(&*state.database, String::from("denormalize_ip_identities")).await;
+
+    if env::var("MARS_DATABASE_MIGRATION").is_ok() {
+        let migration = env::var("MARS_DATABASE_MIGRATION").unwrap_or("NONE".to_owned());
+        info!("API will not run, migration is set");
+        info!("Executing migration '{}'...", migration.to_owned());
+        let migration_executor = MigrationExecutor::new();
+        let migration_found = migration_executor.execute_migration_by_name(
+            &*state.database,
+            migration.to_owned()
+        ).await;
+        if !migration_found {
+            warn!("Did not find migration '{}'", &migration);
+        }
+        return Ok(());
+    };
 
     let ws_port = env::var("MARS_WS_PORT").unwrap_or("7000".to_owned()).parse::<u32>().unwrap_or(7000);
     let res = tokio::try_join!(