refactor: filter for distinct handles in PushService instead of in th…

…e repository
nmshd · Jan 28, 2025 · 2e69771 · 2e69771
1 parent e04ee3d
commit 2e69771
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/.../Devices.Application/Infrastructure/Persistence/Repository/IPnsRegistrationsRepository.cs b/.../Devices.Application/Infrastructure/Persistence/Repository/IPnsRegistrationsRepository.cs
@@ -9,7 +9,7 @@ public interface IPnsRegistrationsRepository
     Task Add(PnsRegistration registration, CancellationToken cancellationToken);
     Task Update(PnsRegistration registration, CancellationToken cancellationToken);
     Task<PnsRegistration?> FindByDeviceId(DeviceId deviceId, CancellationToken cancellationToken, bool track = false);
-    Task<List<PnsRegistration>> FindDistinctByDeviceIds(DeviceId[] deviceIds, CancellationToken cancellationToken, bool track = false);
+    Task<PnsRegistration[]> FindByDeviceIds(DeviceId[] deviceIds, CancellationToken cancellationToken, bool track = false);
     Task<int> Delete(List<DeviceId> deviceIds, CancellationToken cancellationToken);
     Task Delete(Expression<Func<PnsRegistration, bool>> filter, CancellationToken cancellationToken);
 }
diff --git a/...s/Devices/src/Devices.Infrastructure/Persistence/Repository/PnsRegistrationsRepository.cs b/...s/Devices/src/Devices.Infrastructure/Persistence/Repository/PnsRegistrationsRepository.cs
@@ -42,13 +42,11 @@ public async Task Add(PnsRegistration registration, CancellationToken cancellati
             .FirstOrDefaultAsync(registration => registration.DeviceId == deviceId, cancellationToken);
     }
 
-    public async Task<List<PnsRegistration>> FindDistinctByDeviceIds(DeviceId[] deviceIds, CancellationToken cancellationToken, bool track = false)
+    public async Task<PnsRegistration[]> FindByDeviceIds(DeviceId[] deviceIds, CancellationToken cancellationToken, bool track = false)
     {
         return await (track ? _registrations : _readonlyRegistrations)
             .Where(r => deviceIds.Contains(r.DeviceId))
-            .GroupBy(r => r.Handle)
-            .Select(g => g.First())
-            .ToListAsync(cancellationToken);
+            .ToArrayAsync(cancellationToken);
     }
 
     public async Task<int> Delete(List<DeviceId> deviceIds, CancellationToken cancellationToken)

diff --git a/Modules/Devices/src/Devices.Infrastructure/PushNotifications/PushService.cs b/Modules/Devices/src/Devices.Infrastructure/PushNotifications/PushService.cs
@@ -69,9 +69,11 @@ private async Task SendNotificationInternal(IPushNotification notification, Devi
     {
         var deviceIds = devices.Select(d => d.Id).ToArray();
 
-        var registrations = await _pnsRegistrationsRepository.FindDistinctByDeviceIds(deviceIds, cancellationToken);
+        var registrations = await _pnsRegistrationsRepository.FindByDeviceIds(deviceIds, cancellationToken);
 
-        var groups = registrations.GroupBy(registration => registration.Handle.Platform);
+        var groups = registrations
+            .DistinctBy(r => r.Handle) // Since there can be multiple registrations with the same handle, we should make sure we send the same push notification only once per handle
+            .GroupBy(r => r.Handle.Platform);
 
         foreach (var group in groups)
         {