🔀 :: (#203) 담당 반 조회 API 수정 (#121)

DSM-PICK · Jul 12, 2023 · 5ac22f5 · 5ac22f5
1 parent bedea87
commit 5ac22f5
Showing 1 changed file with 6 additions and 2 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
@@ -119,9 +119,13 @@ class TeacherUseCase(
     override fun queryMyBuckGradeAndClassNum(): QueryMyBuckGradeAndClassNumResponse {
         val userInfo = userSpi.queryUserInfoByUserId(userSpi.getCurrentUserId())
 
-        val userGrade = if (userInfo.grade == 0) userInfo.grade else 1
-        val userClassNum = if (userInfo.classNum == 0) userInfo.classNum else 1
+        var userGrade = 1
+        var userClassNum = 1
 
+        if (!(userInfo.grade == 0 && userInfo.classNum == 0)) {
+            userGrade = userInfo.grade
+            userClassNum = userInfo.classNum
+        }
         return QueryMyBuckGradeAndClassNumResponse(userGrade, userClassNum)
     }