Added multiple sending notifications (#36)

vapor · May 18, 2021 · a880646 · a880646
1 parent 1a74041
commit a880646
Showing 1 changed file with 46 additions and 1 deletion.
diff --git a/Sources/APNS/Application+APNS.swift b/Sources/APNS/Application+APNS.swift
@@ -19,7 +19,6 @@ extension Application {
             }
         }
 
-
         struct PoolKey: StorageKey, LockKey {
             typealias Value = EventLoopGroupConnectionPool<APNSConnectionSource>
         }
@@ -60,6 +59,52 @@ extension Application.APNS: APNSwiftClient {
         self.application.eventLoopGroup.next()
     }
 
+    public func batchSend(
+        rawBytes payload: ByteBuffer,
+        pushType: APNSwiftConnection.PushType,
+        to deviceToken: String...,
+        expiration: Date?,
+        priority: Int?,
+        collapseIdentifier: String?,
+        topic: String?,
+        logger: Logger?,
+        apnsID: UUID? = nil)
+    -> EventLoopFuture<Void> {
+        batchSend(rawBytes: payload,
+                  pushType: pushType,
+                  to: deviceToken,
+                  expiration: expiration,
+                  priority: priority,
+                  collapseIdentifier: collapseIdentifier,
+                  topic: topic,
+                  logger: logger,
+                  apnsID: apnsID)
+    }
+
+    public func batchSend(
+        rawBytes payload: ByteBuffer,
+        pushType: APNSwiftConnection.PushType,
+        to deviceToken: [String],
+        expiration: Date?,
+        priority: Int?,
+        collapseIdentifier: String?,
+        topic: String?,
+        logger: Logger?,
+        apnsID: UUID? = nil)
+    -> EventLoopFuture<Void> {
+        deviceToken.map {
+            send(rawBytes: payload,
+                 pushType: pushType,
+                 to: $0,
+                 expiration: expiration,
+                 priority: priority,
+                 collapseIdentifier: collapseIdentifier,
+                 topic: topic,
+                 logger: logger,
+                 apnsID: apnsID)
+        }.flatten(on: self.eventLoop)
+    }
+
     public func send(
         rawBytes payload: ByteBuffer,
         pushType: APNSwiftConnection.PushType,