학급 open api batch로 가져오기

build.gradle.kts

@@ -52,7 +52,7 @@ dependencies {
     runtimeOnly("com.mysql:mysql-connector-j")
 
     // mongodb
-    implementation("org.springframework.boot:spring-boot-starter-data-mongodb")
+//    implementation("org.springframework.boot:spring-boot-starter-data-mongodb")
 
     // security
     implementation("org.springframework.boot:spring-boot-starter-security")
@@ -85,6 +85,7 @@ dependencies {
 
     // webclient
     implementation("org.springframework.boot:spring-boot-starter-webflux")
+
     //JSON
     implementation("org.json:json:20231013")
 

src/main/kotlin/com/asap/asapbackend/batch/classroom/ClassroomInfoProvider.kt

@@ -0,0 +1,31 @@
+package com.asap.asapbackend.batch.classroom
+
+import com.asap.asapbackend.domain.classroom.domain.model.Classroom
+import com.asap.asapbackend.domain.school.domain.model.School
+
+interface ClassroomInfoProvider {
+    fun retrieveClassroomInfo(batchSize: Int, startIndex: Int): ClassroomDataContainer
+
+    data class ClassroomDataContainer(
+        val classroomInfo: List<ClassroomInfo>,
+        val hasNext: Boolean
+    )
+
+    data class ClassroomInfo(
+        val school: School,
+        val grade: Int,
+        val classNumber: String
+    ) {
+        fun toClassroom(): Classroom {
+            return Classroom(
+                school = school,
+                grade = grade,
+                className = classNumber
+            )
+        }
+    }
+}
+
+fun List<ClassroomInfoProvider.ClassroomInfo>.toClassrooms(): List<Classroom> {
+    return this.map { it.toClassroom() }
+}

src/main/kotlin/com/asap/asapbackend/batch/classroom/ClassroomScheduler.kt

@@ -1,16 +1,27 @@
 package com.asap.asapbackend.batch.classroom
 
+import com.asap.asapbackend.domain.classroom.domain.service.ClassroomAppender
+import com.asap.asapbackend.global.util.TransactionUtils
 import org.springframework.scheduling.annotation.Scheduled
-import org.springframework.web.bind.annotation.GetMapping
-import org.springframework.web.bind.annotation.RestController
+import org.springframework.stereotype.Component
 
-@RestController
+@Component
 class ClassroomScheduler(
-        private val classroomService: ClassroomService
+    private val classroomInfoProvider: ClassroomInfoProvider,
+    private val classroomAppender: ClassroomAppender
 ) {
-    @Scheduled(cron = "0 0 4 1 3 ?") // 매년 3월 1일 04:00:00에 실행
-    @GetMapping("/addClassroom")
-    fun addClassroom(){
-        classroomService.addClassroom()
+    @Scheduled(cron = "0 5 4 1 3 ?") // 매년 3월 1일 04:05:00에 실행
+    fun addClassroom() {
+        val batchSize = 100
+        var startIndex = 0
+        do {
+            val classroomDataContainer = classroomInfoProvider.retrieveClassroomInfo(batchSize, startIndex)
+
+            startIndex += batchSize
+
+            TransactionUtils.writable {
+                classroomAppender.addClassroom(classroomDataContainer.classroomInfo.toClassrooms())
+            }
+        } while (classroomDataContainer.hasNext)
     }
 }

src/main/kotlin/com/asap/asapbackend/batch/school/SchoolInfoProvider.kt

@@ -1,6 +1,5 @@
 package com.asap.asapbackend.batch.school
 
-import com.asap.asapbackend.client.openapi.school.dto.SchoolInfo
 import com.asap.asapbackend.domain.school.domain.model.School
 
 interface SchoolInfoProvider {
@@ -16,8 +15,8 @@ interface SchoolInfoProvider {
         val eduOfficeCode: String, //ATPT_OFCDC_SC_CODE
         val schoolCode: String, //SD_SCHUL_CODE
         val school: String, //SCHUL_NM
-        val address:String //ORG_RDNMA
-    ){
+        val address: String //ORG_RDNMA
+    ) {
         fun toSchool(): School {
             return School(
                 eduOfficeCode = eduOfficeCode,

src/main/kotlin/com/asap/asapbackend/batch/school/SchoolScheduler.kt

@@ -2,23 +2,18 @@ package com.asap.asapbackend.batch.school
 
 import com.asap.asapbackend.domain.school.domain.service.SchoolAppender
 import com.asap.asapbackend.global.util.TransactionUtils
-import io.github.oshai.kotlinlogging.KotlinLogging
 import org.springframework.scheduling.annotation.Scheduled
 import org.springframework.stereotype.Component
 
-private val logger = KotlinLogging.logger {}
-
 @Component
 class SchoolScheduler(
     private val schoolInfoProvider: SchoolInfoProvider,
     private val schoolAppender: SchoolAppender
 ) {
     @Scheduled(cron = "0 0 4 1 3 ?") // 매년 3월 1일 04:00:00에 실행
-//    @Scheduled(fixedRate = 500000) // 500초마다 실행
     fun addSchool() {
         val batchSize = 100
         var startIndex = 1
-        logger.info { "start scheduler" }
         do {
             val schoolDataContainer = schoolInfoProvider.retrieveSchoolInfo(batchSize, startIndex)
 
@@ -27,6 +22,6 @@ class SchoolScheduler(
             TransactionUtils.writable {
                 schoolAppender.appendUniqueSchool(schoolDataContainer.schoolInfo.toSchools())
             }
-        }while (schoolDataContainer.hasNext)
+        } while (schoolDataContainer.hasNext)
     }
 }

src/main/kotlin/com/asap/asapbackend/client/openapi/classroom/ClassroomOpenApiClient.kt

@@ -0,0 +1,53 @@
+package com.asap.asapbackend.client.openapi.classroom
+
+import com.asap.asapbackend.batch.classroom.ClassroomInfoProvider
+import com.asap.asapbackend.client.openapi.classroom.dto.ClassroomOpenApiResponse
+import com.asap.asapbackend.domain.school.domain.repository.SchoolRepository
+import com.fasterxml.jackson.module.kotlin.jacksonObjectMapper
+import org.springframework.stereotype.Component
+import org.springframework.web.reactive.function.client.WebClient
+import org.springframework.web.util.UriBuilder
+import java.time.Year
+
+@Component
+class ClassroomOpenApiClient(
+    private val schoolRepository: SchoolRepository
+) : ClassroomInfoProvider {
+    override fun retrieveClassroomInfo(batchSize: Int, startIndex: Int): ClassroomInfoProvider.ClassroomDataContainer {
+        var endIndex = startIndex + batchSize
+        val hasNext = schoolRepository.count() > endIndex
+        if (!hasNext) {
+            endIndex = schoolRepository.count().toInt()
+        }
+        val schools = schoolRepository.findAll().subList(startIndex, endIndex)
+        val classroomInfoList: MutableList<ClassroomInfoProvider.ClassroomInfo> = mutableListOf()
+        schools.forEach { school ->
+            val apiUrl = "https://open.neis.go.kr/hub/classInfo"
+            val classroomInfoResult = WebClient.create(apiUrl).get()
+                .uri { uriBuilder: UriBuilder ->
+                    uriBuilder
+                        .queryParam("KEY", "32e897d4054342b19fd68dfb1b9ba621")
+                        .queryParam("ATPT_OFCDC_SC_CODE", school.eduOfficeCode)
+                        .queryParam("SD_SCHUL_CODE", school.schoolCode)
+                        .queryParam("AY", Year.now())
+                        .queryParam("Type", "json")
+                        .build()
+                }
+                .retrieve()
+                .bodyToMono(String::class.java)
+                .map {
+                    jacksonObjectMapper().readValue(it, ClassroomOpenApiResponse::class.java)
+                }
+                .block()
+            classroomInfoResult?.classInfo?.forEach { classInfo ->
+                classInfo.row?.let { rows ->
+                    classroomInfoList.addAll(rows.map { it.toClassroomInfo(school) })
+                }
+            }
+        }
+        return ClassroomInfoProvider.ClassroomDataContainer(
+            classroomInfo = classroomInfoList,
+            hasNext = hasNext
+        )
+    }
+}

src/main/kotlin/com/asap/asapbackend/client/openapi/classroom/dto/ClassroomOpenApiResponse.kt

@@ -0,0 +1,47 @@
+package com.asap.asapbackend.client.openapi.classroom.dto
+
+import com.asap.asapbackend.batch.classroom.ClassroomInfoProvider
+import com.asap.asapbackend.domain.school.domain.model.School
+
+data class ClassroomOpenApiResponse(
+    val classInfo: List<ClassInfo>?,
+    val RESULT: Result?
+)
+
+data class ClassInfo(
+    val head: List<Head>?,
+    val row: List<Row>?
+)
+
+data class Head(
+    val list_total_count: Int?,
+    val RESULT: Result?
+)
+
+data class Result(
+    val CODE: String,
+    val MESSAGE: String
+)
+
+data class Row(
+    val ATPT_OFCDC_SC_CODE: String?,
+    val ATPT_OFCDC_SC_NM: String?,
+    val SD_SCHUL_CODE: String?,
+    val SCHUL_NM: String?,
+    val AY: String?,
+    val GRADE: Int,
+    val DGHT_CRSE_SC_NM: String?,
+    val SCHUL_CRSE_SC_NM: String?,
+    val ORD_SC_NM: String?,
+    val DDDEP_NM: String?,
+    val CLASS_NM: String,
+    val LOAD_DTM: String?
+) {
+    fun toClassroomInfo(school: School): ClassroomInfoProvider.ClassroomInfo {
+        return ClassroomInfoProvider.ClassroomInfo(
+            school,
+            GRADE,
+            CLASS_NM
+        )
+    }
+}

src/main/kotlin/com/asap/asapbackend/client/openapi/school/SchoolOpenApiClient.kt

@@ -6,16 +6,16 @@ import org.springframework.stereotype.Component
 import org.springframework.web.reactive.function.client.WebClient
 
 @Component
-class SchoolOpenApiClient : SchoolInfoProvider{
-    override fun retrieveSchoolInfo(batchSize: Int, startIndex: Int): SchoolInfoProvider.SchoolDataContainer{
+class SchoolOpenApiClient : SchoolInfoProvider {
+    override fun retrieveSchoolInfo(batchSize: Int, startIndex: Int): SchoolInfoProvider.SchoolDataContainer {
         val endIndex = startIndex + batchSize
         val apiUrl = "http://openapi.seoul.go.kr:8088/4b78477274736f6d36386d505a614b/json/neisSchoolInfoJS/$startIndex/$endIndex"
         val schoolInfoResult = WebClient.create(apiUrl).get()
             .retrieve()
             .bodyToMono(SchoolOpenApiResponse::class.java)
             .block()
         val neisSchoolInfoJS = schoolInfoResult?.neisSchoolInfoJS
-        val hasNext = (schoolInfoResult?.neisSchoolInfoJS?.list_total_count ?: 0) > endIndex
+        val hasNext = (neisSchoolInfoJS?.list_total_count ?: 0) > endIndex
         return SchoolInfoProvider.SchoolDataContainer(
             schoolInfo = neisSchoolInfoJS?.row?.map { it.toSchoolInfo() } ?: emptyList(),
             hasNext = hasNext