Add restartPipeline to trustedsources and streams. (#44)

CatalystCode · Mar 1, 2018 · 0f80c01 · 0f80c01
1 parent 0de184f
commit 0f80c01
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/project-fortis-services/src/resolvers/Settings/mutations.js b/project-fortis-services/src/resolvers/Settings/mutations.js
@@ -213,6 +213,7 @@ function addTrustedSources(args, res) { // eslint-disable-line no-unused-vars
     });
 
     cassandraConnector.executeBatchMutations(mutations)
+      .then(() => streamingController.restartPipeline())
       .then(_ => { // eslint-disable-line no-unused-vars
         resolve({
           sources: args.input.sources
@@ -238,6 +239,7 @@ function removeTrustedSources(args, res) { // eslint-disable-line no-unused-vars
     }));
 
     cassandraConnector.executeBatchMutations(mutations)
+      .then(() => streamingController.restartPipeline())
       .then(_ => { // eslint-disable-line no-unused-vars
         resolve({
           sources: args.input.sources
@@ -415,6 +417,7 @@ function removeStreams(args, res) { // eslint-disable-line no-unused-vars
     }));
 
     cassandraConnector.executeBatchMutations(mutations)
+      .then(() => streamingController.restartPipeline())
       .then(_ => { // eslint-disable-line no-unused-vars
         resolve({
           streams