Only return UserIds from db layer

src/Core/NotificationCenter/Repositories/INotificationRepository.cs

@@ -39,7 +39,7 @@ public interface INotificationRepository : IRepository<Notification, Guid>
     /// <param name="taskId">The unique identifier of the task.</param>
     /// <param name="userId">User Id</param>
     /// <returns>
-    /// A collection of notifications associated with the task that are now marked as deleted.
+    /// A collection of users ids for the notifications that are now marked as deleted.
     /// </returns>
-    Task<IEnumerable<Notification>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId);
+    Task<IEnumerable<Guid>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId);
 }

src/Core/Vault/Commands/MarkNotificationsForTaskAsDeletedCommand.cs

@@ -20,13 +20,13 @@ public class MarkNotificationsForTaskAsDeletedCommand : IMarkNotificationsForTas
 
     public async Task MarkAsDeletedAsync(Guid taskId, Guid userId)
     {
-        var notifications = await _notificationRepository.MarkNotificationsAsDeletedByTask(taskId, userId);
+        var userIds = await _notificationRepository.MarkNotificationsAsDeletedByTask(taskId, userId);
 
         // For each user associated with the notifications, send a push notification so local tasks can be updated.
-        var uniqueUserIds = notifications.Select(n => n.UserId).Where(u => u.HasValue).Distinct();
+        var uniqueUserIds = userIds.Distinct();
         foreach (var id in uniqueUserIds)
         {
-            await _pushNotificationService.PushPendingSecurityTasksAsync((Guid)id);
+            await _pushNotificationService.PushPendingSecurityTasksAsync(id);
         }
     }
 }

src/Infrastructure.Dapper/NotificationCenter/Repositories/NotificationRepository.cs

@@ -57,11 +57,11 @@ public class NotificationRepository : Repository<Notification, Guid>, INotificat
         };
     }
 
-    public async Task<IEnumerable<Notification>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId)
+    public async Task<IEnumerable<Guid>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId)
     {
         await using var connection = new SqlConnection(ConnectionString);
 
-        var results = await connection.QueryAsync<Notification>(
+        var results = await connection.QueryAsync<Guid>(
             "[dbo].[Notification_MarkAsDeletedByTask]",
             new
             {

src/Infrastructure.EntityFramework/NotificationCenter/Repositories/NotificationRepository.cs

@@ -75,7 +75,7 @@ public class NotificationRepository : Repository<Core.NotificationCenter.Entitie
         };
     }
 
-    public async Task<IEnumerable<Core.NotificationCenter.Entities.Notification>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId)
+    public async Task<IEnumerable<Guid>> MarkNotificationsAsDeletedByTask(Guid taskId, Guid userId)
     {
         await using var scope = ServiceScopeFactory.CreateAsyncScope();
         var dbContext = GetDatabaseContext(scope);
@@ -116,6 +116,11 @@ public class NotificationRepository : Repository<Core.NotificationCenter.Entitie
 
         await dbContext.SaveChangesAsync();
 
-        return notifications;
+        var userIds = notifications
+            .Select(n => n.UserId)
+            .Where(u => u.HasValue)
+            .ToList();
+
+        return (IEnumerable<Guid>)userIds;
     }
 }

src/Sql/NotificationCenter/dbo/Stored Procedures/Notification_MarkAsDeletedByTask.sql

@@ -6,14 +6,14 @@ BEGIN
     SET NOCOUNT ON;
 
     -- Collect NotificationIds as they are altered
-    DECLARE @AlteredNotifications TABLE (
-        NotificationId UNIQUEIDENTIFIER
+    DECLARE @UserIdsForAlteredNotifications TABLE (
+        UserId UNIQUEIDENTIFIER
     );
 
     -- Update existing NotificationStatus as deleted
     UPDATE ns
     SET ns.DeletedDate = GETUTCDATE()
-    OUTPUT inserted.NotificationId INTO @AlteredNotifications
+    OUTPUT inserted.UserId INTO @UserIdsForAlteredNotifications
     FROM NotificationStatus ns
     INNER JOIN Notification n ON ns.NotificationId = n.Id
     WHERE n.TaskId = @TaskId
@@ -22,7 +22,7 @@ BEGIN
 
     -- Insert NotificationStatus records for notifications that don't have one yet
     INSERT INTO NotificationStatus (NotificationId, UserId, DeletedDate)
-    OUTPUT inserted.NotificationId INTO @AlteredNotifications
+    OUTPUT inserted.UserId INTO @UserIdsForAlteredNotifications
     SELECT n.Id, @UserId, GETUTCDATE()
     FROM Notification n
     LEFT JOIN NotificationStatus ns
@@ -30,9 +30,8 @@ BEGIN
     WHERE n.TaskId = @TaskId
       AND ns.NotificationId IS NULL;
 
-    -- Return all notifications that have been altered
-    SELECT n.*
-    FROM Notification n
-    INNER JOIN @AlteredNotifications a ON n.Id = a.NotificationId;
+    -- Return the user ids associated with the altered notifications
+    SELECT u.UserId
+    FROM @UserIdsForAlteredNotifications u;
 END
 GO

util/Migrator/DbScripts/2025-05-30_00_Notification_MarkAsDeletedByTask.sql

@@ -6,14 +6,14 @@ BEGIN
     SET NOCOUNT ON;
 
     -- Collect NotificationIds as they are altered
-    DECLARE @AlteredNotifications TABLE (
-        NotificationId UNIQUEIDENTIFIER
+    DECLARE @UserIdsForAlteredNotifications TABLE (
+        UserId UNIQUEIDENTIFIER
     );
 
     -- Update existing NotificationStatus as deleted
     UPDATE ns
     SET ns.DeletedDate = GETUTCDATE()
-    OUTPUT inserted.NotificationId INTO @AlteredNotifications
+    OUTPUT inserted.UserId INTO @UserIdsForAlteredNotifications
     FROM NotificationStatus ns
     INNER JOIN Notification n ON ns.NotificationId = n.Id
     WHERE n.TaskId = @TaskId
@@ -22,7 +22,7 @@ BEGIN
 
     -- Insert NotificationStatus records for notifications that don't have one yet
     INSERT INTO NotificationStatus (NotificationId, UserId, DeletedDate)
-    OUTPUT inserted.NotificationId INTO @AlteredNotifications
+    OUTPUT inserted.UserId INTO @UserIdsForAlteredNotifications
     SELECT n.Id, @UserId, GETUTCDATE()
     FROM Notification n
     LEFT JOIN NotificationStatus ns
@@ -30,9 +30,8 @@ BEGIN
     WHERE n.TaskId = @TaskId
       AND ns.NotificationId IS NULL;
 
-    -- Return all notifications that have been altered
-    SELECT n.*
-    FROM Notification n
-    INNER JOIN @AlteredNotifications a ON n.Id = a.NotificationId;
+    -- Return the user ids associated with the altered notifications
+    SELECT u.UserId
+    FROM @UserIdsForAlteredNotifications u;
 END
 GO