feat: 시간표 점수 응답에 보여주기

yourssu · Feb 7, 2025 · 59fbefe · 59fbefe
1 parent a549540
commit 59fbefe
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/main/kotlin/com/yourssu/soongpt/domain/timetable/business/dto/TimetableResponse.kt b/src/main/kotlin/com/yourssu/soongpt/domain/timetable/business/dto/TimetableResponse.kt
@@ -5,13 +5,15 @@ import com.yourssu.soongpt.domain.timetable.implement.Timetable
 data class TimetableResponse(
     val timetableId: Long,
     val tag: String,
+    var score: Int?,
     val courses: List<TimetableCourseResponse>,
 ) {
     companion object {
-        fun from(timetable: Timetable, courses: List<TimetableCourseResponse>): TimetableResponse {
+        fun from(timetable: Timetable, courses: List<TimetableCourseResponse>, score: Int? = null): TimetableResponse {
             return TimetableResponse(
                 timetableId = timetable.id!!,
                 tag = timetable.tag.name,
+                score = score,
                 courses = courses,
             )
         }

diff --git a/src/main/kotlin/com/yourssu/soongpt/domain/timetable/implement/TimeTableFactory.kt b/src/main/kotlin/com/yourssu/soongpt/domain/timetable/implement/TimeTableFactory.kt
@@ -77,14 +77,15 @@ class TimeTableFactory(
     @Transactional
     fun issueTimetables(timetableCandidates: TimetableCandidates): List<TimetableResponse> {
         val responses = ArrayList<TimetableResponse>()
-        for (timetableCandidate in timetableCandidates.values) {
-            val timetable = timetableWriter.save(Timetable(tag = timetableCandidate.tag))
-            saveTimetableCourses(timetableCandidate.courses, timetable)
+        for (step in timetableCandidates.values) {
+            val timetable = timetableWriter.save(Timetable(tag = step.tag))
+            saveTimetableCourses(step.courses, timetable)
             responses.add(
                 TimetableResponse(
-                    timetable.id!!,
-                    timetable.tag.name,
-                    toTimetableCourseResponses(timetableCandidate.courses)
+                    timetableId = timetable.id!!,
+                    tag = timetable.tag.name,
+                    score = step.score,
+                    courses = toTimetableCourseResponses(step.courses)
                 )
             )
         }