♻️ PagingDataSource를 내가 쓴 글 탭에서 활용할 수 있게 변경

Author: soopeach

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

@@ -16,6 +16,10 @@ interface PostDataSource {
 
     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(

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

@@ -238,6 +238,26 @@ class PostDataSourceImpl @Inject constructor(
             .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) {

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

@@ -6,6 +6,7 @@ import com.google.firebase.firestore.QuerySnapshot
 import com.whyranoid.domain.model.Post
 
 class PostPagingDataSource(
+    private val myUid: String = EMPTY_STRING,
     private val postDataSource: PostDataSource
 ) : PagingSource<QuerySnapshot, Post>() {
 
@@ -18,7 +19,12 @@ class PostPagingDataSource(
             val postList = mutableListOf<Post>()
 
             // 현재 페이지
-            val currentPage = postDataSource.getCurrentPagingPost(params.key)
+            val currentPage =
+                if (myUid == EMPTY_STRING) {
+                    postDataSource.getCurrentPagingPost(params.key)
+                } else {
+                    postDataSource.getMyCurrentPagingPost(params.key, myUid)
+                }
 
             // Post 타입 캐스팅
             // TODO 예외 처리
@@ -32,7 +38,12 @@ class PostPagingDataSource(
             val lastDocumentSnapshot = currentPage.documents[currentPage.size() - 1]
 
             // 마지막 스냅샷 이후 페이지 불러오기
-            val nextPage = postDataSource.getNextPagingPost(lastDocumentSnapshot)
+            val nextPage =
+                if (myUid == EMPTY_STRING) {
+                    postDataSource.getNextPagingPost(lastDocumentSnapshot)
+                } else {
+                    postDataSource.getMyNextPagingPost(lastDocumentSnapshot, myUid)
+                }
 
             LoadResult.Page(
                 data = postList,
@@ -43,4 +54,8 @@ class PostPagingDataSource(
             LoadResult.Error(e)
         }
     }
+
+    companion object {
+        private const val EMPTY_STRING = ""
+    }
 }

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

@@ -11,25 +11,23 @@ import kotlinx.coroutines.flow.Flow
 import javax.inject.Inject
 
 class PostRepositoryImpl @Inject constructor(
-    private val postDataSource: PostDataSource,
-    private val myPostPagingDataSource: MyPostPagingDataSource
+    private val postDataSource: PostDataSource
 ) : PostRepository {
 
     // TODO : 캐싱하기
     override fun getPagingPosts(coroutineScope: CoroutineScope): Flow<PagingData<Post>> {
         return Pager(
             PagingConfig(pageSize = 5)
         ) {
-            PostPagingDataSource(postDataSource)
+            PostPagingDataSource(postDataSource = postDataSource)
         }.flow.cachedIn(coroutineScope)
     }
 
     override fun getMyPagingPosts(uid: String): Flow<PagingData<Post>> {
-        myPostPagingDataSource.setMyUid(uid)
         return Pager(
             PagingConfig(pageSize = 5)
         ) {
-            myPostPagingDataSource
+            PostPagingDataSource(myUid = uid, postDataSource)
         }.flow
     }