feat: 상태값으로 조회하는 메서드 추가

Author: rnqhstmd

apps/commerce-streamer/src/main/java/com/loopers/domain/dlq/DlqMessageRepository.java

@@ -15,5 +15,5 @@ public interface DlqMessageRepository {
 
     List<DlqMessage> findAll();
 
-    long count();
+    long countByStatus(DlqMessage.DlqStatus status);
 }

apps/commerce-streamer/src/main/java/com/loopers/domain/dlq/DlqMessageService.java

@@ -76,6 +76,6 @@ public void incrementRetryCount(String id) {
 
     @Transactional(readOnly = true)
     public long countPendingMessages() {
-        return dlqMessageRepository.findByStatus(DlqMessage.DlqStatus.PENDING).size();
+        return dlqMessageRepository.countByStatus(DlqMessage.DlqStatus.PENDING);
     }
 }

apps/commerce-streamer/src/main/java/com/loopers/infrastructure/dlq/DlqMessageJpaRepository.java

@@ -1,6 +1,7 @@
 package com.loopers.infrastructure.dlq;
 
 import com.loopers.domain.dlq.DlqMessage;
+import org.springframework.data.domain.Pageable;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.Query;
 import org.springframework.data.repository.query.Param;
@@ -11,9 +12,10 @@ public interface DlqMessageJpaRepository extends JpaRepository<DlqMessage, Strin
 
     List<DlqMessage> findByStatus(DlqMessage.DlqStatus status);
 
-    @Query("SELECT d FROM DlqMessage d WHERE d.status = 'PENDING' AND d.retryCount < :maxRetryCount ORDER BY d.createdAt ASC LIMIT :limit")
-    List<DlqMessage> findPendingMessagesForRetry(
-            @Param("maxRetryCount") int maxRetryCount,
-            @Param("limit") int limit
-    );
+    @Query("SELECT d FROM DlqMessage d WHERE d.status = :status AND d.retryCount < :maxRetryCount ORDER BY d.createdAt ASC")
+    List<DlqMessage> findPendingMessagesForRetry(@Param("status") DlqMessage.DlqStatus status,
+                                                 @Param("maxRetryCount") int maxRetryCount,
+                                                 Pageable pageable);
+
+    long countByStatus(DlqMessage.DlqStatus status);
 }

apps/commerce-streamer/src/main/java/com/loopers/infrastructure/dlq/DlqMessageRepositoryImpl.java

@@ -3,10 +3,13 @@
 import com.loopers.domain.dlq.DlqMessage;
 import com.loopers.domain.dlq.DlqMessageRepository;
 import lombok.RequiredArgsConstructor;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.stereotype.Repository;
 
 import java.util.List;
 import java.util.Optional;
 
+@Repository
 @RequiredArgsConstructor
 public class DlqMessageRepositoryImpl implements DlqMessageRepository {
 
@@ -29,7 +32,7 @@ public List<DlqMessage> findByStatus(DlqMessage.DlqStatus status) {
 
     @Override
     public List<DlqMessage> findPendingMessagesForRetry(int maxRetryCount, int limit) {
-        return dlqMessageJpaRepository.findPendingMessagesForRetry(maxRetryCount, limit);
+        return dlqMessageJpaRepository.findPendingMessagesForRetry(DlqMessage.DlqStatus.PENDING, maxRetryCount, PageRequest.of(0, 100));
     }
 
     @Override
@@ -38,7 +41,7 @@ public List<DlqMessage> findAll() {
     }
 
     @Override
-    public long count() {
-        return dlqMessageJpaRepository.count();
+    public long countByStatus(DlqMessage.DlqStatus status) {
+        return dlqMessageJpaRepository.countByStatus(status);
     }
 }