🔨 Add chatCount in Item

wafflestudio · Jan 31, 2025 · a4e1bce · a4e1bce
1 parent 993cfe3
commit a4e1bce
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/main/kotlin/com/toyProject7/karrot/article/controller/ArticleController.kt b/src/main/kotlin/com/toyProject7/karrot/article/controller/ArticleController.kt
@@ -95,7 +95,7 @@ class ArticleController(
         val articles: List<ArticleEntity> = articleService.getPreviousArticles(articleId)
         val response: List<Item> =
             articles.map { article ->
-                Item.fromArticle(Article.fromEntity(article))
+                Item.fromArticle(Article.fromEntity(article), articleService.getChattingUsersByArticle(Article.fromEntity(article)).size)
             }
         return ResponseEntity.ok(response)
     }
@@ -108,7 +108,7 @@ class ArticleController(
         val articles: List<ArticleEntity> = articleService.getArticlesThatUserLikes(user.id, articleId)
         val response =
             articles.map { article ->
-                Item.fromArticle(Article.fromEntity(article))
+                Item.fromArticle(Article.fromEntity(article), articleService.getChattingUsersByArticle(Article.fromEntity(article)).size)
             }
         return ResponseEntity.ok(response)
     }
@@ -121,7 +121,7 @@ class ArticleController(
         val articles: List<ArticleEntity> = articleService.getArticlesBySeller(user.id, articleId)
         val response: List<Item> =
             articles.map { article ->
-                Item.fromArticle(Article.fromEntity(article))
+                Item.fromArticle(Article.fromEntity(article), articleService.getChattingUsersByArticle(Article.fromEntity(article)).size)
             }
         return ResponseEntity.ok(response)
     }
@@ -134,7 +134,7 @@ class ArticleController(
         val articles = articleService.getArticlesByBuyer(user.id, articleId)
         val response: List<Item> =
             articles.map { article ->
-                Item.fromArticle(Article.fromEntity(article))
+                Item.fromArticle(Article.fromEntity(article), articleService.getChattingUsersByArticle(Article.fromEntity(article)).size)
             }
         return ResponseEntity.ok(response)
     }

diff --git a/src/main/kotlin/com/toyProject7/karrot/article/controller/Item.kt b/src/main/kotlin/com/toyProject7/karrot/article/controller/Item.kt
@@ -12,9 +12,13 @@ data class Item(
     val imagePresignedUrl: String,
     val createdAt: Instant,
     val likeCount: Int,
+    val chatCount: Int,
 ) {
     companion object {
-        fun fromArticle(article: Article): Item {
+        fun fromArticle(
+            article: Article,
+            chatCount: Int,
+        ): Item {
             return Item(
                 id = article.id,
                 title = article.title,
@@ -24,6 +28,7 @@ data class Item(
                 imagePresignedUrl = if (article.imagePresignedUrl.isEmpty()) "" else article.imagePresignedUrl.first(),
                 createdAt = article.createdAt,
                 likeCount = article.likeCount,
+                chatCount = chatCount,
             )
         }
     }