refactor: 100개 단위로 페이지네이션 형태를 통해 모든 데이터 받아오기

Author: opficdev

DevLog/Domain/Entity/TodoQuery.swift

@@ -14,7 +14,8 @@ struct TodoQuery {
     let createdAtFrom: Date?
     let createdAtTo: Date?
     let createdAtDescending: Bool
-    let pageSize: Int?
+    let pageSize: Int
+    let fetchAllPages: Bool
 
     init(
         kind: TodoKind? = nil,
@@ -23,7 +24,8 @@ struct TodoQuery {
         createdAtFrom: Date? = nil,
         createdAtTo: Date? = nil,
         createdAtDescending: Bool = true,
-        pageSize: Int? = 20
+        pageSize: Int = 20,
+        fetchAllPages: Bool = false
     ) {
         self.kind = kind
         self.keyword = keyword
@@ -32,5 +34,6 @@ struct TodoQuery {
         self.createdAtTo = createdAtTo
         self.createdAtDescending = createdAtDescending
         self.pageSize = pageSize
+        self.fetchAllPages = fetchAllPages
     }
 }

DevLog/Domain/UseCase/UserData/Fetch/Todo/FetchTodosByDateRangeUseCaseImpl.swift

@@ -18,7 +18,8 @@ final class FetchTodosByDateRangeUseCaseImpl: FetchTodosByDateRangeUseCase {
         let query = TodoQuery(
             createdAtFrom: startDate,
             createdAtTo: endDate,
-            pageSize: nil
+            pageSize: 100,
+            fetchAllPages: true
         )
         let page = try await repository.fetchTodos(query, cursor: nil)
         return page.items

DevLog/Infra/Service/TodoService.swift

@@ -27,7 +27,8 @@ final class TodoService {
             query.isPinned != nil ? "pinned=\(query.isPinned!)" : nil,
             query.createdAtFrom != nil ? "createdAtFrom=\(query.createdAtFrom!)" : nil,
             query.createdAtTo != nil ? "createdAtTo=\(query.createdAtTo!)" : nil,
-            query.pageSize != nil ? "pageSize=\(query.pageSize!)" : nil,
+            "pageSize=\(query.pageSize)",
+            query.fetchAllPages ? "fetchAllPages=true" : nil,
             cursor != nil ? "cursor=\(cursor!)" : nil
         ]
         logger.info("Fetching todo page: \(logComponents.compactMap { $0 }.joined(separator: ", "))")
@@ -60,36 +61,49 @@ final class TodoService {
         }
 
         if trimmedKeyword.isEmpty {
+            if query.fetchAllPages {
+                var allItems: [TodoResponse] = []
+                var pageCursor = cursor
+
+                while true {
+                    var pageQuery = firestoreQuery
+                    if let pageCursor {
+                        pageQuery = pageQuery.start(after: [
+                            Timestamp(date: pageCursor.createdAt),
+                            pageCursor.documentID
+                        ])
+                    }
+
+                    pageQuery = pageQuery.limit(to: query.pageSize)
+                    let snapshot = try await pageQuery.getDocuments()
+                    allItems.append(contentsOf: snapshot.documents.compactMap { makeResponse(from: $0) })
+
+                    guard snapshot.documents.count == query.pageSize else {
+                        break
+                    }
+
+                    guard let lastDocument = snapshot.documents.last,
+                          let nextCursor = makeCursor(from: lastDocument) else {
+                        break
+                    }
+
+                    pageCursor = nextCursor
+                }
+
+                return TodoPageResponse(items: allItems, nextCursor: nil)
+            }
+
             if let cursor {
                 firestoreQuery = firestoreQuery.start(after: [
                     Timestamp(date: cursor.createdAt),
                     cursor.documentID
                 ])
             }
 
-            if let pageSize = query.pageSize {
-                firestoreQuery = firestoreQuery.limit(to: pageSize)
-            }
-
+            firestoreQuery = firestoreQuery.limit(to: query.pageSize)
             let snapshot = try await firestoreQuery.getDocuments()
-
             let items = snapshot.documents.compactMap { makeResponse(from: $0) }
-
-            let nextCursor: TodoCursorDTO?
-            if query.pageSize == nil {
-                nextCursor = nil
-            } else {
-                nextCursor = snapshot.documents.last.flatMap { document in
-                    guard let createdAt = document.data()[TodoFieldKey.createdAt.rawValue] as? Timestamp else {
-                        return nil
-                    }
-
-                    return TodoCursorDTO(
-                        createdAt: createdAt.dateValue(),
-                        documentID: document.documentID
-                    )
-                }
-            }
+            let nextCursor = snapshot.documents.last.flatMap { makeCursor(from: $0) }
 
             return TodoPageResponse(items: items, nextCursor: nextCursor)
         }
@@ -167,6 +181,17 @@ final class TodoService {
 }
 
 private extension TodoService {
+    func makeCursor(from document: QueryDocumentSnapshot) -> TodoCursorDTO? {
+        guard let createdAt = document.data()[TodoFieldKey.createdAt.rawValue] as? Timestamp else {
+            return nil
+        }
+
+        return TodoCursorDTO(
+            createdAt: createdAt.dateValue(),
+            documentID: document.documentID
+        )
+    }
+
     func makeResponse(from snapshot: QueryDocumentSnapshot) -> TodoResponse? {
         makeResponse(documentID: snapshot.documentID, data: snapshot.data())
     }