From 590a1cb8ad6f10b855c4a51cc2f350892f5d6590 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 9 Mar 2023 21:46:33 +0800
Subject: [PATCH] Only publish labels to add to the pubsub queue

---
 packages/api/src/elastic/labels.ts         | 19 ++++++++++++-------
 packages/api/src/resolvers/labels/index.ts | 18 +++++++++++++-----
 2 files changed, 25 insertions(+), 12 deletions(-)

diff --git a/packages/api/src/elastic/labels.ts b/packages/api/src/elastic/labels.ts
index 11119d99b..79c2b6d97 100644
--- a/packages/api/src/elastic/labels.ts
+++ b/packages/api/src/elastic/labels.ts
@@ -57,7 +57,8 @@ export const addLabelInPage = async (
 export const updateLabelsInPage = async (
   pageId: string,
   labels: Label[],
-  ctx: PageContext
+  ctx: PageContext,
+  labelsToAdd?: Label[]
 ): Promise<boolean> => {
   try {
     const { body } = await client.update({
@@ -74,12 +75,16 @@ export const updateLabelsInPage = async (
     })
 
     if (body.result !== 'updated') return false
-
-    for (const label of labels) {
-      await ctx.pubsub.entityCreated<Label & { pageId: string }>(
-        EntityType.LABEL,
-        { pageId, ...label },
-        ctx.uid
+    if (labelsToAdd) {
+      // publish labels to be added
+      await Promise.all(
+        labelsToAdd.map((label) =>
+          ctx.pubsub.entityCreated<Label & { pageId: string }>(
+            EntityType.LABEL,
+            { pageId, ...label },
+            ctx.uid
+          )
+        )
       )
     }
 
diff --git a/packages/api/src/resolvers/labels/index.ts b/packages/api/src/resolvers/labels/index.ts
index 78cb75703..31bf420f8 100644
--- a/packages/api/src/resolvers/labels/index.ts
+++ b/packages/api/src/resolvers/labels/index.ts
@@ -243,12 +243,20 @@ export const setLabelsResolver = authorized<
         errorCodes: [SetLabelsErrorCode.NotFound],
       }
     }
-
+    // filter out labels that are already set
+    const labelsToAdd = labels.filter(
+      (label) => !page.labels?.some((pageLabel) => pageLabel.id === label.id)
+    )
     // update labels in the page
-    const updated = await updateLabelsInPage(pageId, labels, {
-      pubsub,
-      uid,
-    })
+    const updated = await updateLabelsInPage(
+      pageId,
+      labels,
+      {
+        pubsub,
+        uid,
+      },
+      labelsToAdd
+    )
     if (!updated) {
       return {
         errorCodes: [SetLabelsErrorCode.NotFound],