node: Add graphman dump command

Wire the dump subcommand into graphman CLI. Takes a deployment search
argument and an output directory, resolves the deployment, and delegates
to SubgraphStore::dump().

Author: lutter

node/src/bin/manager.rs

@@ -327,6 +327,21 @@ pub enum Command {
         #[clap(long, short, default_value = "http://localhost:8020")]
         url: String,
     },
+
+    /// Dump a subgraph deployment into a directory
+    ///
+    /// EXPERIMENTAL - NOT FOR PRODUCTION USE
+    ///
+    /// This will create a dump of the subgraph deployment in the specified
+    /// directory. The dump includes the subgraph manifest, the mapping, and
+    /// the data in the database as parquet files. The dump can be used to
+    /// restore the subgraph deployment later with the `restore` command.
+    Dump {
+        /// The deployment (see `help info`)
+        deployment: DeploymentSearch,
+        /// The name of the directory to dump to
+        directory: String,
+    },
 }
 
 impl Command {
@@ -1732,6 +1747,16 @@ async fn main() -> anyhow::Result<()> {
 
             commands::deploy::run(subgraph_store, deployment, name, url).await
         }
+
+        Dump {
+            deployment,
+            directory,
+        } => {
+            let (store, primary_pool) = ctx.store_and_primary().await;
+            let subgraph_store = store.subgraph_store();
+
+            commands::dump::run(subgraph_store, primary_pool, deployment, directory).await
+        }
     }
 }
 

node/src/manager/commands/dump.rs

@@ -0,0 +1,30 @@
+use std::fs;
+use std::sync::Arc;
+
+use graph::{bail, prelude::anyhow::Result};
+
+use graph_store_postgres::{ConnectionPool, SubgraphStore};
+
+use crate::manager::deployment::DeploymentSearch;
+
+pub async fn run(
+    subgraph_store: Arc<SubgraphStore>,
+    primary_pool: ConnectionPool,
+    deployment: DeploymentSearch,
+    directory: String,
+) -> Result<()> {
+    let directory = fs::canonicalize(&directory)?;
+    let stat = fs::metadata(&directory)?;
+
+    if !stat.is_dir() {
+        bail!(
+            "The path `{}` is not a directory",
+            directory.to_string_lossy()
+        );
+    }
+
+    let loc = deployment.locate_unique(&primary_pool).await?;
+
+    subgraph_store.dump(&loc, directory).await?;
+    Ok(())
+}

node/src/manager/commands/mod.rs

@@ -7,6 +7,7 @@ pub mod create;
 pub mod database;
 pub mod deploy;
 pub mod deployment;
+pub mod dump;
 pub mod index;
 pub mod listen;
 pub mod provider_checks;