🔀 :: (#199) 담당 반 조회 API 수정 (#119)

DSM-PICK · Jul 12, 2023 · bedea87 · bedea87
1 parent 03b5faa
commit bedea87
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...ion/src/main/kotlin/com/pickdsm/pickserverspring/domain/teacher/usecase/TeacherUseCase.kt b/...ion/src/main/kotlin/com/pickdsm/pickserverspring/domain/teacher/usecase/TeacherUseCase.kt
@@ -117,10 +117,12 @@ class TeacherUseCase(
     }
 
     override fun queryMyBuckGradeAndClassNum(): QueryMyBuckGradeAndClassNumResponse {
-        val userId = userSpi.getCurrentUserId()
-        val userInfo = userSpi.queryUserInfoByUserId(userId)
+        val userInfo = userSpi.queryUserInfoByUserId(userSpi.getCurrentUserId())
 
-        return QueryMyBuckGradeAndClassNumResponse(userInfo.grade, userInfo.classNum)
+        val userGrade = if (userInfo.grade == 0) userInfo.grade else 1
+        val userClassNum = if (userInfo.classNum == 0) userInfo.classNum else 1
+
+        return QueryMyBuckGradeAndClassNumResponse(userGrade, userClassNum)
     }
 
     private fun User.paddedUserNum(): String =