add try catch

Signed-off-by: Qxisylolo <[email protected]>
opensearch-project · Nov 28, 2024 · d5fb789 · d5fb789
1 parent 357c833
commit d5fb789
Showing 1 changed file with 40 additions and 23 deletions.
diff --git a/src/plugins/workspace/server/saved_objects/workspace_saved_objects_client_wrapper.ts b/src/plugins/workspace/server/saved_objects/workspace_saved_objects_client_wrapper.ts
@@ -505,48 +505,65 @@ export class WorkspaceSavedObjectsClientWrapper {
       let flag = true;
       const processedObjects = await Promise.all(
         objectToBulkGet.saved_objects.map(async (object) => {
-          if (validateIsWorkspaceDataSourceAndConnectionObjectType(object.type)) {
-            const hasPermission = this.validateDataSourcePermissions(
-              object,
-              wrapperOptions.request
-            );
-            if (!hasPermission) {
+          try {
+            if (validateIsWorkspaceDataSourceAndConnectionObjectType(object.type)) {
+              const hasPermission = this.validateDataSourcePermissions(
+                object,
+                wrapperOptions.request
+              );
+              if (!hasPermission) {
+                flag = false;
+                ACLAuditor?.increment(ACLAuditorStateKey.VALIDATE_FAILURE, 1);
+                return {
+                  ...object,
+                  workspaces: [],
+                  attributes: {} as T,
+                  error: {
+                    ...generateDataSourcePermissionError().output.payload,
+                    statusCode: 403,
+                  },
+                };
+              }
+            }
+
+            if (
+              !(await this.validateWorkspacesAndSavedObjectsPermissions(
+                object,
+                wrapperOptions.request,
+                [WorkspacePermissionMode.LibraryRead, WorkspacePermissionMode.LibraryWrite],
+                [WorkspacePermissionMode.Write, WorkspacePermissionMode.Read],
+                false
+              ))
+            ) {
               flag = false;
               ACLAuditor?.increment(ACLAuditorStateKey.VALIDATE_FAILURE, 1);
               return {
                 ...object,
                 workspaces: [],
                 attributes: {} as T,
                 error: {
-                  ...generateDataSourcePermissionError().output.payload,
+                  ...generateSavedObjectsPermissionError().output.payload,
                   statusCode: 403,
                 },
               };
             }
-          }
-
-          if (
-            !(await this.validateWorkspacesAndSavedObjectsPermissions(
-              object,
-              wrapperOptions.request,
-              [WorkspacePermissionMode.LibraryRead, WorkspacePermissionMode.LibraryWrite],
-              [WorkspacePermissionMode.Write, WorkspacePermissionMode.Read],
-              false
-            ))
-          ) {
+            return object;
+          } catch (error) {
             flag = false;
             ACLAuditor?.increment(ACLAuditorStateKey.VALIDATE_FAILURE, 1);
+
+            const formattedError = {
+              message: error?.message || 'unexpected error',
+              error: error?.error,
+              statusCode: error.statusCode,
+            };
             return {
               ...object,
               workspaces: [],
               attributes: {} as T,
-              error: {
-                ...generateSavedObjectsPermissionError().output.payload,
-                statusCode: 403,
-              },
+              error: formattedError,
             };
           }
-          return object;
         })
       );