Merge pull request #9 from API-Team-Organization/develop

Feat :: 이미지 모두 불러오기

Author: gwon11225

.DS_Store

@@ -28,7 +28,6 @@ class ChatController(
         return ResponseBody.ok("답변 생성 성공", data)
     }
 
-    @GetMapping("/{roomId}")
     @PageableAsQueryParam
     suspend fun getChatList(
         @PathVariable("roomId") roomId: Long,
@@ -40,4 +39,8 @@ class ChatController(
         )
         return ResponseBody.ok("채팅 조회 성공", data)
     }
+
+    @GetMapping("/my-image")
+    fun getMyImage(@RequestParam pageNumber: Int, @RequestParam pageSize: Int) = chatService
+        .getMyImage(pageNumber, pageSize)
 }

src/main/kotlin/com/teamapi/palette/controller/ChatController.kt

@@ -9,4 +9,5 @@ import org.springframework.stereotype.Repository
 @Repository
 interface ChatRepository : CoroutineCrudRepository<Chat, Long> {
     suspend fun findAllByRoomIdIsOrderByDatetimeDesc(roomId: Long, pageable: Pageable): Flow<Chat>
+    fun findByIsAiAndUserId(isAi: Boolean, userId: Long, pageable: Pageable): Flux<Chat>
 }

src/main/kotlin/com/teamapi/palette/repository/ChatRepository.kt

@@ -13,12 +13,13 @@ import com.teamapi.palette.repository.ChatRepository
 import com.teamapi.palette.repository.RoomRepository
 import com.teamapi.palette.response.ErrorCode
 import com.teamapi.palette.response.exception.CustomException
+import org.springframework.ai.image.ImageResponse
+import org.springframework.data.domain.*
 import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.map
 import kotlinx.coroutines.flow.toList
 import kotlinx.coroutines.reactor.awaitSingle
 import org.slf4j.LoggerFactory
-import org.springframework.data.domain.Pageable
 import org.springframework.stereotype.Service
 import reactor.core.publisher.Mono
 import reactor.kotlin.core.util.function.component1
@@ -131,6 +132,18 @@ class ChatService(
 //            )
 //        )
 //    )
+    fun getMyImage(pageNumber: Int, pageSize: Int): Mono<PageImpl<String>> {
+        val page = PageRequest.of(pageNumber, pageSize)
+        return sessionHolder
+            .me()
+            .flatMap { userId ->
+                chatRepository.findByIsAiAndUserId(true, userId, page)
+                    .map { it.resource }
+                    .collectList()
+                    .zipWith(chatRepository.count())
+                    .map { p -> PageImpl(p.t1, page, p.t2) }
+            }
+    }
 
     private fun chatCompletion(options: ChatCompletionsOptions) = azure.getChatCompletions(
         "PaletteGPT", options