chore: optimize assignee notification handling with chunked processing

Author: Dobrunia

package.json

@@ -1,6 +1,6 @@
 {
   "name": "hawk.api",
-  "version": "1.5.3",
+  "version": "1.5.0",
   "main": "index.ts",
   "license": "BUSL-1.1",
   "scripts": {

src/resolvers/event.js

@@ -6,6 +6,7 @@ const {
 } = require('./helpers/bulkEvents');
 const { aiService } = require('../services/ai');
 const { UserInputError } = require('apollo-server-express');
+const { ObjectId } = require('mongodb');
 
 /**
  * See all types and fields here {@see ../typeDefs/event.graphql}
@@ -317,6 +318,10 @@ module.exports = {
       const factory = getEventsFactory(context, projectId);
 
       if (assignee) {
+        if (!ObjectId.isValid(String(assignee))) {
+          throw new UserInputError('assignee must be a valid id or null');
+        }
+
         const userExists = await factories.usersFactory.findById(assignee);
 
         if (!userExists) {

src/resolvers/helpers/bulkEvents.js

@@ -2,6 +2,8 @@ const sendPersonalNotification = require('../../utils/personalNotifications').de
 const { UserInputError } = require('apollo-server-express');
 const { ObjectId } = require('mongodb');
 
+const ASSIGNEE_NOTIFICATIONS_CHUNK_SIZE = 25;
+
 /**
  * Enqueue assignee notifications in background (do not block resolver response)
  *
@@ -26,20 +28,40 @@ function fireAndForgetAssigneeNotifications({
     return;
   }
 
-  Promise.allSettled(eventIds.map(eventId => sendPersonalNotification(assigneeData, {
-    type: 'assignee',
-    payload: {
-      assigneeId,
-      projectId,
-      whoAssignedId,
-      eventId,
-    },
-  })))
-    .then((results) => {
-      const failedResults = results.filter(result => result.status === 'rejected');
+  Promise.resolve()
+    .then(async () => {
+      const failedResults = [];
+
+      for (let i = 0; i < eventIds.length; i += ASSIGNEE_NOTIFICATIONS_CHUNK_SIZE) {
+        const chunk = eventIds.slice(i, i + ASSIGNEE_NOTIFICATIONS_CHUNK_SIZE);
+        const results = await Promise.allSettled(chunk.map(eventId => sendPersonalNotification(assigneeData, {
+          type: 'assignee',
+          payload: {
+            assigneeId,
+            projectId,
+            whoAssignedId,
+            eventId,
+          },
+        })));
+
+        failedResults.push(...results.filter(result => result.status === 'rejected'));
+      }
 
       if (failedResults.length > 0) {
-        console.error('Failed to enqueue assignee notifications', failedResults);
+        const failedMessages = failedResults.map((result) => {
+          const reason = result && result.reason;
+
+          if (reason && typeof reason.message === 'string') {
+            return reason.message;
+          }
+
+          return String(reason || 'Unknown error');
+        });
+
+        console.error('Failed to enqueue assignee notifications', {
+          failedCount: failedResults.length,
+          errors: failedMessages,
+        });
       }
     })
     .catch((error) => {

src/typeDefs/event.ts

@@ -602,7 +602,8 @@ extend type Mutation {
 
   """
   Toggle the same mark on many original events at once (resolved, ignored or starred).
-  Same toggle semantics as toggleEventMark per event.
+  Uses bulk semantics: if every selected event already has the mark, clear it for all;
+  otherwise set it on each selected event that does not have it yet.
   """
   bulkToggleEventMarks(
     """

test/resolvers/event-bulk-update-assignee.test.ts

@@ -26,16 +26,17 @@ const eventResolvers = require('../../src/resolvers/event') as {
 };
 
 const bulkUpdateAssignee = jest.fn();
+const ASSIGNEE_ID = '507f1f77bcf86cd799439012';
 
 describe('EventsMutations.bulkUpdateAssignee', () => {
   const ctx = {
     user: { id: 'u1' },
     factories: {
       usersFactory: {
-        findById: jest.fn().mockResolvedValue({ id: 'assignee-1' }),
+        findById: jest.fn().mockResolvedValue({ id: ASSIGNEE_ID }),
         dataLoaders: {
           userById: {
-            load: jest.fn().mockResolvedValue({ id: 'assignee-1', email: 'a@a.a' }),
+            load: jest.fn().mockResolvedValue({ id: ASSIGNEE_ID, email: 'a@a.a' }),
           },
         },
       },
@@ -44,7 +45,7 @@ describe('EventsMutations.bulkUpdateAssignee', () => {
       },
       workspacesFactory: {
         findById: jest.fn().mockResolvedValue({
-          getMemberInfo: jest.fn().mockResolvedValue({ userId: 'assignee-1' }),
+          getMemberInfo: jest.fn().mockResolvedValue({ userId: ASSIGNEE_ID }),
         }),
       },
     },
@@ -66,21 +67,39 @@ describe('EventsMutations.bulkUpdateAssignee', () => {
     await expect(
       eventResolvers.EventsMutations.bulkUpdateAssignee(
         {},
-        { input: { projectId: 'p1', eventIds: [], assignee: 'assignee-1' } },
+        { input: { projectId: 'p1', eventIds: [], assignee: ASSIGNEE_ID } },
         ctx
       )
     ).rejects.toThrow(UserInputError);
     expect(bulkUpdateAssignee).not.toHaveBeenCalled();
   });
 
+  it('should throw when assignee id is invalid', async () => {
+    await expect(
+      eventResolvers.EventsMutations.bulkUpdateAssignee(
+        {},
+        {
+          input: {
+            projectId: 'p1',
+            eventIds: [ '507f1f77bcf86cd799439011' ],
+            assignee: 'not-an-object-id',
+          },
+        },
+        ctx
+      )
+    ).rejects.toThrow(UserInputError);
+
+    expect(bulkUpdateAssignee).not.toHaveBeenCalled();
+  });
+
   it('should call factory for bulk assign', async () => {
     const result = await eventResolvers.EventsMutations.bulkUpdateAssignee(
       {},
       {
         input: {
           projectId: 'p1',
           eventIds: [ '507f1f77bcf86cd799439011' ],
-          assignee: 'assignee-1',
+          assignee: ASSIGNEE_ID,
         },
       },
       ctx
@@ -89,15 +108,15 @@ describe('EventsMutations.bulkUpdateAssignee', () => {
     expect(result.updatedCount).toBe(1);
     expect(bulkUpdateAssignee).toHaveBeenCalledWith(
       [ '507f1f77bcf86cd799439011' ],
-      'assignee-1'
+      ASSIGNEE_ID
     );
     expect(sendPersonalNotification).toHaveBeenCalledTimes(1);
     expect(sendPersonalNotification).toHaveBeenCalledWith(
-      expect.objectContaining({ id: 'assignee-1' }),
+      expect.objectContaining({ id: ASSIGNEE_ID }),
       expect.objectContaining({
         type: 'assignee',
         payload: expect.objectContaining({
-          assigneeId: 'assignee-1',
+          assigneeId: ASSIGNEE_ID,
           projectId: 'p1',
           whoAssignedId: 'u1',
           eventId: '507f1f77bcf86cd799439011',
@@ -119,15 +138,15 @@ describe('EventsMutations.bulkUpdateAssignee', () => {
         input: {
           projectId: 'p1',
           eventIds: [ '507f1f77bcf86cd799439011', 'invalid-id' ],
-          assignee: 'assignee-1',
+          assignee: ASSIGNEE_ID,
         },
       },
       ctx
     );
 
     expect(bulkUpdateAssignee).toHaveBeenCalledWith(
       [ '507f1f77bcf86cd799439011' ],
-      'assignee-1'
+      ASSIGNEE_ID
     );
     expect(result).toEqual({
       updatedCount: 1,
@@ -143,7 +162,7 @@ describe('EventsMutations.bulkUpdateAssignee', () => {
         input: {
           projectId: 'p1',
           eventIds: [ 'bad-1', 'bad-2' ],
-          assignee: 'assignee-1',
+          assignee: ASSIGNEE_ID,
         },
       },
       ctx