feat(aggregator): add database migrate clap command

Author: dlachaume

mithril-aggregator/src/commands/database_command.rs

@@ -0,0 +1,81 @@
+use std::path::PathBuf;
+
+use anyhow::Context;
+use clap::{Parser, Subcommand};
+use slog::{debug, Logger};
+
+use mithril_common::StdResult;
+
+use crate::{dependency_injection::DependenciesBuilder, Configuration, ExecutionEnvironment};
+
+/// Database tools
+#[derive(Parser, Debug, Clone)]
+pub struct DatabaseCommand {
+    /// commands
+    #[clap(subcommand)]
+    pub database_subcommand: DatabaseSubCommand,
+}
+
+impl DatabaseCommand {
+    pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
+        self.database_subcommand.execute(root_logger).await
+    }
+}
+
+#[derive(Debug, Clone, Subcommand)]
+pub enum DatabaseSubCommand {
+    /// Migrate databases located in the given stores directory
+    Migrate(MigrateCommand),
+}
+
+impl DatabaseSubCommand {
+    pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
+        match self {
+            Self::Migrate(cmd) => cmd.execute(root_logger).await,
+        }
+    }
+}
+
+#[derive(Parser, Debug, Clone)]
+pub struct MigrateCommand {
+    /// Stores directory
+    #[clap(long, env = "STORES_DIRECTORY")]
+    stores_directory: PathBuf,
+}
+
+impl MigrateCommand {
+    pub async fn execute(&self, root_logger: Logger) -> StdResult<()> {
+        let config = Configuration {
+            environment: ExecutionEnvironment::Production,
+            data_stores_directory: self.stores_directory.clone(),
+            // Temporary solution to avoid the need to provide a full configuration
+            ..Configuration::new_sample()
+        };
+        debug!(root_logger, "DATABASE MIGRATE command"; "config" => format!("{config:?}"));
+        println!(
+            "Migrating databases from stores directory: {}",
+            self.stores_directory.to_string_lossy()
+        );
+        let mut dependencies_builder =
+            DependenciesBuilder::new(root_logger.clone(), config.clone());
+
+        dependencies_builder
+            .get_sqlite_connection()
+            .await
+            .with_context(|| "Dependencies Builder can not get sqlite connection")?;
+
+        dependencies_builder
+            .get_event_store_sqlite_connection()
+            .await
+            .with_context(|| "Dependencies Builder can not get event store sqlite connection")?;
+
+        dependencies_builder
+            .get_sqlite_connection_cardano_transaction_pool()
+            .await
+            .with_context(|| {
+                "Dependencies Builder can not get cardano transaction pool sqlite connection"
+            })?;
+
+        Ok(())
+    }
+}

mithril-aggregator/src/commands/mod.rs

@@ -1,3 +1,4 @@
+mod database_command;
 mod era_command;
 mod genesis_command;
 mod serve_command;
@@ -21,6 +22,7 @@ pub enum MainCommand {
     Era(era_command::EraCommand),
     Serve(serve_command::ServeCommand),
     Tools(tools_command::ToolsCommand),
+    Database(database_command::DatabaseCommand),
     #[clap(alias("doc"), hide(true))]
     GenerateDoc(GenerateDocCommands),
 }
@@ -44,6 +46,7 @@ impl MainCommand {
             Self::Era(cmd) => cmd.execute(root_logger, config_builder).await,
             Self::Serve(cmd) => cmd.execute(root_logger, config_builder).await,
             Self::Tools(cmd) => cmd.execute(root_logger, config_builder).await,
+            Self::Database(cmd) => cmd.execute(root_logger).await,
             Self::GenerateDoc(cmd) => {
                 let config_infos = vec![Configuration::extract(), DefaultConfiguration::extract()];
                 cmd.execute_with_configurations(&mut MainOpts::command(), &config_infos)
@@ -58,6 +61,7 @@ impl MainCommand {
             MainCommand::Genesis(_) => CommandType::CommandLine,
             MainCommand::Era(_) => CommandType::CommandLine,
             MainCommand::Tools(_) => CommandType::CommandLine,
+            MainCommand::Database(_) => CommandType::CommandLine,
             MainCommand::GenerateDoc(_) => CommandType::CommandLine,
         }
     }