graphman: add graphman remove command to graphql api

Author: shiyasmohd

server/graphman/src/resolvers/deployment_mutation.rs

@@ -3,7 +3,6 @@ use std::sync::Arc;
 use async_graphql::Context;
 use async_graphql::Object;
 use async_graphql::Result;
-use graph::prelude::SubgraphName;
 use graph_store_postgres::graphman::GraphmanStore;
 
 use crate::entities::DeploymentSelector;
@@ -13,6 +12,7 @@ use crate::resolvers::context::GraphmanContext;
 
 mod create;
 mod pause;
+mod remove;
 mod restart;
 mod resume;
 
@@ -71,8 +71,14 @@ impl DeploymentMutation {
     /// Create a subgraph
     pub async fn create(&self, ctx: &Context<'_>, name: String) -> Result<EmptyResponse> {
         let ctx = GraphmanContext::new(ctx)?;
+        create::run(&ctx, &name)?;
+        Ok(EmptyResponse::new())
+    }
 
-        create::run(&ctx, &name);
+    /// Remove a subgraph
+    pub async fn remove(&self, ctx: &Context<'_>, name: String) -> Result<EmptyResponse> {
+        let ctx = GraphmanContext::new(ctx)?;
+        remove::run(&ctx, &name)?;
         Ok(EmptyResponse::new())
     }
 }

server/graphman/src/resolvers/deployment_mutation/create.rs

@@ -1,11 +1,11 @@
+use crate::resolvers::context::GraphmanContext;
 use anyhow::anyhow;
+use async_graphql::Result;
 use graph::prelude::SubgraphName;
 use graph_store_postgres::command_support::catalog;
 use graphman::GraphmanError;
 
-use crate::resolvers::context::GraphmanContext;
-
-pub fn run(ctx: &GraphmanContext, name: &String) -> Result<(), GraphmanError> {
+pub fn run(ctx: &GraphmanContext, name: &String) -> Result<()> {
     let primary_pool = ctx.primary_pool.get().map_err(GraphmanError::from)?;
     let mut catalog_conn = catalog::Connection::new(primary_pool);
 
@@ -14,11 +14,12 @@ pub fn run(ctx: &GraphmanContext, name: &String) -> Result<(), GraphmanError> {
         Err(_) => {
             return Err(GraphmanError::Store(anyhow!(
                 "Subgraph name must contain only a-z, A-Z, 0-9, '-' and '_'"
-            )));
+            ))
+            .try_into()?)
         }
     };
 
-    catalog_conn.create_subgraph(&name);
+    catalog_conn.create_subgraph(&name)?;
 
     Ok(())
 }

server/graphman/src/resolvers/deployment_mutation/remove.rs

@@ -0,0 +1,26 @@
+use crate::resolvers::context::GraphmanContext;
+use anyhow::anyhow;
+use async_graphql::Result;
+use graph::prelude::{StoreEvent, SubgraphName};
+use graph_store_postgres::command_support::catalog;
+use graphman::GraphmanError;
+
+pub fn run(ctx: &GraphmanContext, name: &String) -> Result<()> {
+    let primary_pool = ctx.primary_pool.get().map_err(GraphmanError::from)?;
+    let mut catalog_conn = catalog::Connection::new(primary_pool);
+
+    let name = match SubgraphName::new(name) {
+        Ok(name) => name,
+        Err(_) => {
+            return Err(GraphmanError::Store(anyhow!(
+                "Subgraph name must contain only a-z, A-Z, 0-9, '-' and '_'"
+            ))
+            .try_into()?)
+        }
+    };
+
+    let changes = catalog_conn.remove_subgraph(name)?;
+    catalog_conn.send_store_event(&ctx.notification_sender, &StoreEvent::new(changes))?;
+
+    Ok(())
+}