fix: 평일에 매일 전송 설정한 구독이 보내지지 않는 문제 해결

YAPP-Github · Sep 20, 2024 · fa5935b · fa5935b
1 parent 8d8b2ea
commit fa5935b
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/batch/src/main/kotlin/com/few/batch/service/article/reader/WorkBookSubscriberReader.kt b/batch/src/main/kotlin/com/few/batch/service/article/reader/WorkBookSubscriberReader.kt
@@ -45,10 +45,19 @@ class WorkBookSubscriberReader(
     }
 
     private fun sendDayCondition(sendDayField: TableField<SubscriptionRecord, String>, sendDayCode: BatchDayCode): Condition {
-        return if (sendDayCode == BatchDayCode.MON_TUE_WED_THU_FRI_SAT_SUN) {
-            sendDayField.eq(BatchDayCode.MON_TUE_WED_THU_FRI.code).or(sendDayField.eq(BatchDayCode.MON_TUE_WED_THU_FRI_SAT_SUN.code))
-        } else {
-            sendDayField.eq(sendDayCode.code)
+        return when (sendDayCode) {
+            /** 평일인 경우 매일을 포함하여 전송한다 */
+            BatchDayCode.MON_TUE_WED_THU_FRI -> {
+                sendDayField.eq(BatchDayCode.MON_TUE_WED_THU_FRI.code)
+                    .or(sendDayField.eq(BatchDayCode.MON_TUE_WED_THU_FRI_SAT_SUN.code))
+            }
+            /** 매일의 경우 매일만 전송한다 */
+            BatchDayCode.MON_TUE_WED_THU_FRI_SAT_SUN -> {
+                sendDayField.eq(BatchDayCode.MON_TUE_WED_THU_FRI_SAT_SUN.code)
+            }
+            else -> {
+                throw IllegalArgumentException("Invalid sendDayCode: $sendDayCode")
+            }
         }
     }
 }