🔀 #84 from boostcampwm-2022/improve/async

복잡한 callback suspend하게 변경, PagingDataSource 개선

Author: soopeach

data/src/main/java/com/whyranoid/data/group/GroupRepositoryImpl.kt

@@ -7,6 +7,7 @@ import com.whyranoid.domain.model.GroupNotification
 import com.whyranoid.domain.model.Rule
 import com.whyranoid.domain.model.RunningHistory
 import com.whyranoid.domain.repository.GroupRepository
+import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.flow.Flow
 import javax.inject.Inject
 
@@ -20,7 +21,7 @@ class GroupRepositoryImpl @Inject constructor(
         return userDataSource.getMyGroupList(uid)
     }
 
-    override fun getMyGroupListFlow(uid: String) = userDataSource.getMyGroupListFlow(uid)
+    override fun getMyGroupListFlow(uid: String, coroutineScope: CoroutineScope) = userDataSource.getMyGroupListFlow(uid, coroutineScope)
 
     override suspend fun updateGroupInfo(
         groupId: String,

data/src/main/java/com/whyranoid/data/post/MyPostPagingDataSource.kt

@@ -1,5 +1,8 @@
 package com.whyranoid.data.post
 
+import com.google.firebase.firestore.DocumentSnapshot
+import com.google.firebase.firestore.QueryDocumentSnapshot
+import com.google.firebase.firestore.QuerySnapshot
 import com.whyranoid.domain.model.Post
 import kotlinx.coroutines.flow.Flow
 
@@ -9,6 +12,16 @@ interface PostDataSource {
 
     fun getMyPostFlow(uid: String): Flow<List<Post>>
 
+    suspend fun getCurrentPagingPost(key: QuerySnapshot?): QuerySnapshot
+
+    suspend fun getNextPagingPost(lastDocumentSnapshot: DocumentSnapshot): QuerySnapshot
+
+    suspend fun getMyCurrentPagingPost(key: QuerySnapshot?, uid: String): QuerySnapshot
+
+    suspend fun getMyNextPagingPost(lastDocumentSnapshot: DocumentSnapshot, uid: String): QuerySnapshot
+
+    suspend fun convertPostType(document: QueryDocumentSnapshot): Post?
+
     suspend fun createRecruitPost(
         authorUid: String,
         groupUid: String

data/src/main/java/com/whyranoid/data/post/PostDataSource.kt

@@ -1,7 +1,10 @@
 package com.whyranoid.data.post
 
+import com.google.firebase.firestore.DocumentSnapshot
 import com.google.firebase.firestore.FirebaseFirestore
 import com.google.firebase.firestore.Query
+import com.google.firebase.firestore.QueryDocumentSnapshot
+import com.google.firebase.firestore.QuerySnapshot
 import com.whyranoid.data.constant.CollectionId
 import com.whyranoid.data.constant.FieldId.AUTHOR_ID
 import com.whyranoid.data.constant.FieldId.GROUP_ID
@@ -219,6 +222,90 @@ class PostDataSourceImpl @Inject constructor(
             awaitClose()
         }
 
+    override suspend fun getCurrentPagingPost(key: QuerySnapshot?): QuerySnapshot {
+        return key ?: db.collection(CollectionId.POST_COLLECTION)
+            .orderBy(UPDATED_AT, Query.Direction.DESCENDING)
+            .limit(DATA_COUNT_PER_PAGE)
+            .get()
+            .await()
+    }
+
+    override suspend fun getNextPagingPost(lastDocumentSnapshot: DocumentSnapshot): QuerySnapshot {
+        return db.collection(CollectionId.POST_COLLECTION)
+            .orderBy(UPDATED_AT, Query.Direction.DESCENDING)
+            .limit(DATA_COUNT_PER_PAGE).startAfter(lastDocumentSnapshot)
+            .get()
+            .await()
+    }
+
+    override suspend fun getMyCurrentPagingPost(key: QuerySnapshot?, uid: String): QuerySnapshot {
+        return key ?: db.collection(CollectionId.POST_COLLECTION)
+            .whereEqualTo(AUTHOR_ID, uid)
+            .limit(DATA_COUNT_PER_PAGE)
+            .get()
+            .await()
+    }
+
+    override suspend fun getMyNextPagingPost(
+        lastDocumentSnapshot: DocumentSnapshot,
+        uid: String
+    ): QuerySnapshot {
+        return db.collection(CollectionId.POST_COLLECTION)
+            .whereEqualTo(AUTHOR_ID, uid)
+            .limit(DATA_COUNT_PER_PAGE)
+            .startAfter(lastDocumentSnapshot)
+            .get()
+            .await()
+    }
+
+    // TODO : 예외 처리
+    override suspend fun convertPostType(document: QueryDocumentSnapshot): Post? {
+        return if (document[RUNNING_HISTORY_ID] != null) {
+            document.toObject(RunningPostResponse::class.java).let { postResponse ->
+                val authorResponse = getUserResponse(postResponse.authorId)
+
+                authorResponse?.let {
+                    val runningHistory = getRunningHistory(postResponse.runningHistoryId)
+
+                    runningHistory?.let {
+                        RunningPost(
+                            postId = postResponse.postId,
+                            author = authorResponse.toUser(),
+                            updatedAt = postResponse.updatedAt,
+                            runningHistory = it,
+                            likeCount = 0,
+                            content = postResponse.content
+                        )
+                    }
+                }
+            }
+        } else {
+            document.toObject(RecruitPostResponse::class.java).let { postResponse ->
+                val authorResponse = getUserResponse(postResponse.authorId)
+
+                authorResponse?.let {
+                    val groupInfoResponse = getGroupInfoResponse(postResponse.groupId)
+
+                    groupInfoResponse?.let {
+                        val author = authorResponse.toUser()
+                        RecruitPost(
+                            postId = postResponse.postId,
+                            author = author,
+                            updatedAt = postResponse.updatedAt,
+                            groupInfo = groupInfoResponse
+                                .toGroupInfo(
+                                    author,
+                                    rules = groupInfoResponse.rules.map {
+                                        it.toRule()
+                                    }
+                                )
+                        )
+                    }
+                }
+            }
+        }
+    }
+
     override suspend fun createRecruitPost(
         authorUid: String,
         groupUid: String
@@ -310,4 +397,35 @@ class PostDataSourceImpl @Inject constructor(
                 }
         }
     }
+
+    // TODO : 예외 처리
+    private suspend fun getUserResponse(userId: String): UserResponse? {
+        return db.collection(CollectionId.USERS_COLLECTION)
+            .document(userId)
+            .get()
+            .await()
+            .toObject(UserResponse::class.java)
+    }
+
+    // TODO : 예외 처리
+    private suspend fun getRunningHistory(runningHistoryId: String): RunningHistory? {
+        return db.collection(CollectionId.RUNNING_HISTORY_COLLECTION)
+            .document(runningHistoryId)
+            .get()
+            .await()
+            .toObject(RunningHistory::class.java)
+    }
+
+    // TODO : 예외 처리
+    private suspend fun getGroupInfoResponse(groupId: String): GroupInfoResponse? {
+        return db.collection(CollectionId.GROUPS_COLLECTION)
+            .document(groupId)
+            .get()
+            .await()
+            .toObject(GroupInfoResponse::class.java)
+    }
+
+    companion object {
+        private const val DATA_COUNT_PER_PAGE = 10L
+    }
 }