feat: (#523) tokenSubscribe 저장 로직 추가

team-aliens · Jun 16, 2023 · ff8b4d4 · ff8b4d4
1 parent 493d9d1
commit ff8b4d4
Showing 1 changed file with 21 additions and 11 deletions.
diff --git a/...re/src/main/kotlin/team/aliens/dms/domain/notification/service/NotificationServiceImpl.kt b/...re/src/main/kotlin/team/aliens/dms/domain/notification/service/NotificationServiceImpl.kt
@@ -1,10 +1,12 @@
 package team.aliens.dms.domain.notification.service
 
 import org.springframework.stereotype.Component
+import team.aliens.dms.domain.notification.dto.TopicSubscribeResponse
 import team.aliens.dms.domain.notification.exception.DeviceTokenNotFoundException
 import team.aliens.dms.domain.notification.model.DeviceToken
 import team.aliens.dms.domain.notification.model.Notification
 import team.aliens.dms.domain.notification.model.Topic
+import team.aliens.dms.domain.notification.model.TopicSubscribe
 import team.aliens.dms.domain.notification.spi.DeviceTokenPort
 import team.aliens.dms.domain.notification.spi.NotificationPort
 import team.aliens.dms.domain.notification.spi.TopicSubscribePort
@@ -13,6 +15,7 @@ import java.util.UUID
 @Component
 class NotificationServiceImpl(
     private val notificationPort: NotificationPort,
+    private val topicSubscribePort: TopicSubscribePort,
     private val deviceTokenPort: DeviceTokenPort
 ) : NotificationService {
 
@@ -38,35 +41,42 @@ class NotificationServiceImpl(
     }
 
     override fun unsubscribeTopic(deviceToken: String, topic: Topic) {
-        val savedToken = this.getDeviceTokenByDeviceToken(deviceToken)
-        topicSubscribePort.saveTopicSubscribe(
-            TopicSubscribe(
-                deviceTokenId = savedToken.id,
-                topic = topic
-            )
-        )
         notificationPort.unsubscribeTopic(
-            deviceToken = this.getDeviceTokenByDeviceToken(deviceToken).deviceToken,
+            deviceToken = deviceToken,
             topic = topic
         )
     }
 
     override fun updateSubscribes(deviceToken: String, topicsToSubscribe: List<Pair<Topic, Boolean>>) {
 
-        val deviceToken = this.getDeviceTokenByDeviceToken(deviceToken)
+        val savedToken = this.getDeviceTokenByDeviceToken(deviceToken)
+
+        val subscribes = mutableListOf<Topic>()
+        val unsubscribes = mutableListOf<Topic>()
+
         topicsToSubscribe.forEach { (topic, isSubscribe) ->
             if (isSubscribe) {
                 notificationPort.subscribeTopic(
-                    deviceToken = deviceToken.deviceToken,
+                    deviceToken = deviceToken,
                     topic = topic
                 )
+                subscribes.add(topic)
             } else {
                 notificationPort.unsubscribeTopic(
-                    deviceToken = deviceToken.deviceToken,
+                    deviceToken = deviceToken,
                     topic = topic
                 )
+                unsubscribes.add(topic)
             }
         }
+
+        topicSubscribePort.saveAllTopicSubscribes(
+            subscribes.map { TopicSubscribe(savedToken.id, it) }
+        )
+        topicSubscribePort.deleteByUserIdAndTopics(
+            userId = savedToken.id,
+            topics = unsubscribes
+        )
     }
 
     private fun getDeviceTokenByDeviceToken(deviceToken: String) =