Merge pull request #56 from TeamDMU/feat/sqs

[Feat] SQS 모듈 생성 후, FCM 모듈에서 변경

Author: GiJungPark

.github/workflows/ci.yml

@@ -41,6 +41,16 @@ jobs:
           echo "${{ secrets.MONGO }}" | base64 --decode > ./mongo.yml
         working-directory: ./dmforu-infrastructure/storage/mongo/src/main/resources
 
+      - name: Create SQS resources directory
+        run: |
+          mkdir -p ./dmforu-infrastructure/sqs/src/main/resources
+
+      - name: make sqs.yml
+        run: |
+          touch ./sqs.yml
+          echo "${{ secrets.SQS }}" | base64 --decode > ./sqs.yml
+        working-directory: ./dmforu-infrastructure/sqs/src/main/resources
+
       - name: Create Firebase key resources directory
         run: |
           mkdir -p ./dmforu-infrastructure/fcm/src/main/resources/key

.gitignore

@@ -42,4 +42,5 @@ out/
 ### Configuration Settings ###
 **/mongo.yml
 **/mysql.yml
+**/sqs.yml
 **/fire-base-key.json

dmforu-admin/build.gradle.kts

@@ -13,7 +13,7 @@ dependencies {
 
     implementation("org.springframework.boot:spring-boot-starter-web")
 
-    runtimeOnly(project(":dmforu-infrastructure:fcm"))
+    runtimeOnly(project(":dmforu-infrastructure:sqs"))
     runtimeOnly(project(":dmforu-infrastructure:storage:mysql"))
     runtimeOnly(project(":dmforu-infrastructure:storage:mongo"))
 

dmforu-admin/src/main/kotlin/com/dmforu/admin/AdminApplication.kt

@@ -8,7 +8,7 @@ import java.util.*
 @SpringBootApplication(
     scanBasePackages = [
         "com.dmforu.admin",
-        "com.dmforu.fcm",
+        "com.dmforu.sqs",
         "com.dmforu.storage.db.mongo",
         "com.dmforu.storage.db.mysql"
     ]

dmforu-admin/src/main/resources/application.yml

@@ -7,6 +7,7 @@ spring:
       - mysql.yml
       - mongo.yml
       - monitoring.yml
+      - sqs.yml
   web.resources.add-mappings: false
 
 spring.lifecycle.timeout-per-shutdown-phase: 30s

dmforu-infrastructure/sqs/build.gradle.kts

@@ -0,0 +1,9 @@
+dependencies {
+    compileOnly(project(":dmforu-domain"))
+
+    implementation(enforcedPlatform("software.amazon.awssdk:bom:2.21.20"))
+    implementation("software.amazon.awssdk:sqs")
+
+    testImplementation(project(":dmforu-domain"))
+    testImplementation("org.mockito.kotlin:mockito-kotlin:4.1.0")
+}

dmforu-infrastructure/sqs/src/main/kotlin/com/dmforu/sqs/AWSConfig.kt

@@ -0,0 +1,39 @@
+package com.dmforu.sqs
+
+import org.springframework.beans.factory.annotation.Value
+import org.springframework.context.annotation.Bean
+import org.springframework.context.annotation.Configuration
+import software.amazon.awssdk.auth.credentials.AwsBasicCredentials
+import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider
+import software.amazon.awssdk.regions.Region
+import software.amazon.awssdk.services.sqs.SqsAsyncClient
+
+@Configuration
+class AWSConfig(
+    @Value("\${cloud.aws.credentials.access-key}")
+    val accessKey: String,
+
+    @Value("\${cloud.aws.credentials.secret-key}")
+    val secretKey: String,
+
+    @Value("\${cloud.aws.sqs.queue.url}")
+    val queueUrl: String,
+
+    @Value("\${cloud.aws.sqs.queue.message-delay-seconds}")
+    val messageDelaySecs: Int,
+) {
+
+    @Bean
+    fun sqsAsyncClient(): SqsAsyncClient {
+        return SqsAsyncClient.builder()
+            .region(Region.AP_NORTHEAST_2)
+            .credentialsProvider(createAwsCredentialsProvider())
+            .build()
+    }
+
+    private fun createAwsCredentialsProvider(): StaticCredentialsProvider {
+        val basicAwsCredentials = AwsBasicCredentials.create(accessKey, secretKey)
+
+        return StaticCredentialsProvider.create(basicAwsCredentials)
+    }
+}

dmforu-infrastructure/sqs/src/main/kotlin/com/dmforu/sqs/SqsMessageMapper.kt

@@ -0,0 +1,32 @@
+package com.dmforu.sqs
+
+import software.amazon.awssdk.services.sqs.model.MessageAttributeValue
+import software.amazon.awssdk.services.sqs.model.SendMessageRequest
+
+object SqsMessageMapper {
+
+    fun createRequest(content: String, attributes: Map<String, MessageAttributeValue>, queueUrl: String, delaySeconds: Int): SendMessageRequest {
+        return SendMessageRequest.builder()
+            .queueUrl(queueUrl)
+            .delaySeconds(delaySeconds)
+            .messageAttributes(attributes)
+            .messageBody(content)
+            .build()
+    }
+
+    fun createAttributes(
+        tokens: List<String>,
+        title: String,
+        type: String,
+        url: String,
+    ): Map<String, MessageAttributeValue> {
+        val attributes = mutableMapOf<String, MessageAttributeValue>()
+
+        attributes["tokens"] = MessageAttributeValue.builder().stringValue(tokens.toString()).dataType("String").build()
+        attributes["title"] = MessageAttributeValue.builder().stringValue(title).dataType("String").build()
+        attributes["type"] = MessageAttributeValue.builder().stringValue(type).dataType("String").build()
+        attributes["url"] = MessageAttributeValue.builder().stringValue(url).dataType("String").build()
+
+        return attributes
+    }
+}

dmforu-infrastructure/sqs/src/main/kotlin/com/dmforu/sqs/SqsMessageSendException.kt

@@ -0,0 +1,3 @@
+package com.dmforu.sqs
+
+class SqsMessageSendException (message: String, cause: Throwable? = null) : RuntimeException(message, cause)

dmforu-infrastructure/sqs/src/main/kotlin/com/dmforu/sqs/SqsMessageSender.kt

@@ -0,0 +1,47 @@
+package com.dmforu.sqs
+
+import com.dmforu.domain.message.MessageSender
+import com.dmforu.domain.message.NoticeMessage
+import com.dmforu.sqs.SqsMessageMapper.createAttributes
+import com.dmforu.sqs.SqsMessageMapper.createRequest
+import org.springframework.stereotype.Component
+import software.amazon.awssdk.services.sqs.SqsAsyncClient
+import software.amazon.awssdk.services.sqs.model.SendMessageResponse
+import software.amazon.awssdk.services.sqs.model.SqsException
+
+@Component
+class SqsMessageSender(
+    private val sqsAsyncClient: SqsAsyncClient,
+    private val awsConfig: AWSConfig,
+) : MessageSender {
+
+    override fun sendNoticeMessage(message: NoticeMessage, tokens: List<String>) {
+        val attributes = createAttributes(
+            tokens = tokens,
+            title = message.title,
+            type = message.type,
+            url = message.url
+        )
+
+        val request = createRequest(
+            content = message.body,
+            attributes = attributes,
+            queueUrl = awsConfig.queueUrl,
+            delaySeconds = awsConfig.messageDelaySecs
+        )
+
+        try {
+            val future = sqsAsyncClient.sendMessage(request)
+
+            future.whenComplete { sendMessageResponse: SendMessageResponse, throwable: Throwable? ->
+                if (throwable != null) {
+                    throw SqsMessageSendException("[SQS Async Client] 메세지 전송에 실패하였습니다.", throwable)
+                }
+            }
+        } catch (sqsException: SqsException) {
+            throw SqsMessageSendException("[SQS] 메세지 전송에 실패하였습니다.", sqsException)
+        }
+    }
+
+
+}