fix: array append/remove

Signed-off-by: Matt Krick <[email protected]>
ParabolInc · Jul 3, 2024 · f4f5ede · f4f5ede
1 parent 81d68b7
commit f4f5ede
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/packages/server/graphql/private/mutations/updateOrgFeatureFlag.ts b/packages/server/graphql/private/mutations/updateOrgFeatureFlag.ts
@@ -23,13 +23,10 @@ const updateOrgFeatureFlag: MutationResolvers['updateOrgFeatureFlag'] = async (
   // RESOLUTION
   await getKysely()
     .updateTable('Organization')
-    .$if(addFlag, (db) => db.set({featureFlags: sql`ARRAY_APPEND("featureFlags",${flag})`}))
+    .$if(addFlag, (db) => db.set({featureFlags: sql`arr_append_uniq("featureFlags",${flag})`}))
     .$if(!addFlag, (db) =>
       db.set({
-        featureFlags: sql`array_cat(
-          "featureFlags"[1:array_position("featureFlags",${flag})-1],
-          "featureFlags"[array_position("featureFlags",${flag})+1:]
-        )`
+        featureFlags: sql`ARRAY_REMOVE("featureFlags",${flag})`
       })
     )
     .where('id', 'in', orgIds)