Increase limit (#562) (#563)

Signed-off-by: Tyler Ohlsen <ohltyler@amazon.com> (cherry picked from commit a334f08) Co-authored-by: Tyler Ohlsen <ohltyler@amazon.com>
opensearch-project · Jul 20, 2023 · 41983bd · 41983bd
1 parent 1a2f407
commit 41983bd
Showing 1 changed file with 36 additions and 31 deletions.
diff --git a/public/components/FeatureAnywhereContextMenu/CreateAnomalyDetector/AddAnomalyDetector.tsx b/public/components/FeatureAnywhereContextMenu/CreateAnomalyDetector/AddAnomalyDetector.tsx
@@ -204,28 +204,32 @@ function AddAnomalyDetector({
             const maxAssociatedCount = uiSettings.get(
               PLUGIN_AUGMENTATION_MAX_OBJECTS_SETTING
             );
-            await savedObjectLoader.findAll().then(async (resp) => {
-              if (resp !== undefined) {
-                const savedAugmentObjects = get(resp, 'hits', []);
-                // gets all the saved object for this visualization
-                const savedObjectsForThisVisualization =
-                  savedAugmentObjects.filter(
-                    (savedObj) =>
-                      get(savedObj, 'visId', '') === embeddable.vis.id
+            await savedObjectLoader
+              .findAll('', 100, [], {
+                type: 'visualization',
+                id: embeddable.vis.id as string,
+              })
+              .then(async (resp) => {
+                if (resp !== undefined) {
+                  const savedObjectsForThisVisualization = get(
+                    resp,
+                    'hits',
+                    []
                   );
-                if (
-                  maxAssociatedCount <= savedObjectsForThisVisualization.length
-                ) {
-                  notifications.toasts.addDanger(
-                    `Cannot create the detector and associate it to the visualization due to the limit of the max
+                  if (
+                    maxAssociatedCount <=
+                    savedObjectsForThisVisualization.length
+                  ) {
+                    notifications.toasts.addDanger(
+                      `Cannot create the detector and associate it to the visualization due to the limit of the max
                     amount of associated plugin resources (${maxAssociatedCount}) with
                     ${savedObjectsForThisVisualization.length} associated to the visualization`
-                  );
-                } else {
-                  handleSubmit(formikProps);
+                    );
+                  } else {
+                    handleSubmit(formikProps);
+                  }
                 }
-              }
-            });
+              });
           }
         }
       });
@@ -244,20 +248,21 @@ function AddAnomalyDetector({
 
   useEffect(async () => {
     // Gets all augmented saved objects
-    await savedObjectLoader.findAll().then(async (resp) => {
-      if (resp !== undefined) {
-        const savedAugmentObjects = get(resp, 'hits', []);
-        // gets all the saved object for this visualization
-        const savedObjectsForThisVisualization = savedAugmentObjects.filter(
-          (savedObj) => get(savedObj, 'visId', '') === embeddable.vis.id
-        );
-        if (maxAssociatedCount <= savedObjectsForThisVisualization.length) {
-          setAssociationLimitReached(true);
-        } else {
-          setAssociationLimitReached(false);
+    await savedObjectLoader
+      .findAll('', 100, [], {
+        type: 'visualization',
+        id: embeddable.vis.id as string,
+      })
+      .then(async (resp) => {
+        if (resp !== undefined) {
+          const savedObjectsForThisVisualization = get(resp, 'hits', []);
+          if (maxAssociatedCount <= savedObjectsForThisVisualization.length) {
+            setAssociationLimitReached(true);
+          } else {
+            setAssociationLimitReached(false);
+          }
         }
-      }
-    });
+      });
   }, []);
 
   const getEmbeddableSection = () => {