Merge pull request #629 from coralproject/dec-inc-total-comment-count

Increase/Decrease totalCommentCount after mutation
coralproject · Jun 1, 2017 · c76e71c · c76e71c
2 parents 06e6730 + b801550
commit c76e71c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/client/coral-embed-stream/src/graphql/utils.js b/client/coral-embed-stream/src/graphql/utils.js
@@ -28,6 +28,14 @@ function findAndInsertComment(parent, id, comment) {
 }
 
 export function insertCommentIntoEmbedQuery(root, id, comment) {
+
+  // Increase total comment count by one.
+  root = update(root, {
+    asset: {
+      totalCommentCount: {$apply: (c) => c + 1},
+    },
+  });
+
   if (root.comment) {
     if (root.comment.parent) {
       return update(root, {
@@ -76,6 +84,14 @@ function findAndRemoveComment(parent, id) {
 }
 
 export function removeCommentFromEmbedQuery(root, id) {
+
+  // Decrease total comment by one.
+  root = update(root, {
+    asset: {
+      totalCommentCount: {$apply: (c) => c - 1},
+    },
+  });
+
   if (root.comment) {
     if (root.comment.parent) {
       return update(root, {