Merge pull request #3 from Team-Retrip/feature-01/batch

Spring Batch와 Es 연동

Author: xjvmdutl

build.gradle.kts

@@ -35,14 +35,15 @@ dependencies {
     implementation("org.springdoc:springdoc-openapi-starter-webmvc-api:2.8.5")
     implementation("org.springdoc:springdoc-openapi-starter-webmvc-ui:2.8.5")
     implementation("com.querydsl:querydsl-jpa:5.1.0:jakarta")
+
     kapt("com.querydsl:querydsl-apt:5.1.0:jakarta") // annotationProcessor → kapt로 변경
 
     implementation("com.fasterxml.jackson.module:jackson-module-kotlin")
     implementation("org.jetbrains.kotlin:kotlin-reflect")
-    implementation("com.h2database:h2")
+    runtimeOnly("com.mysql:mysql-connector-j")
     compileOnly("org.projectlombok:lombok")
     annotationProcessor("org.projectlombok:lombok")
-
+    testImplementation("org.springframework.batch:spring-batch-test")
     testImplementation("org.springframework.boot:spring-boot-starter-test")
     testImplementation("org.jetbrains.kotlin:kotlin-test-junit5")
     testRuntimeOnly("org.junit.platform:junit-platform-launcher")

src/main/kotlin/com/retrip/map/MapApplication.kt

@@ -4,7 +4,7 @@ import org.springframework.boot.autoconfigure.SpringBootApplication
 import org.springframework.boot.runApplication
 
 @SpringBootApplication
-class MapApplication
+class MapApplication()
 
 fun main(args: Array<String>) {
     runApplication<MapApplication>(*args)

src/main/kotlin/com/retrip/map/application/in/service/LocationIndexService.kt

@@ -0,0 +1,17 @@
+package com.retrip.map.application.`in`.service
+
+import com.retrip.map.application.`in`.usecase.LocationIndexUseCase
+import com.retrip.map.application.out.repository.LocationElasticRepository
+import com.retrip.map.infra.adapter.out.search.elasticsearch.entity.LocationDocument
+import org.springframework.stereotype.Service
+import org.springframework.transaction.annotation.Transactional
+
+@Service
+@Transactional
+class LocationIndexService(
+    private val locationElasticRepository: LocationElasticRepository
+) : LocationIndexUseCase {
+    override fun indexLocationDocuments(documents: List<LocationDocument>?) {
+        documents?.let { locationElasticRepository.saveAll(it) }
+    }
+}

src/main/kotlin/com/retrip/map/application/in/usecase/LocationIndexUseCase.kt

@@ -0,0 +1,7 @@
+package com.retrip.map.application.`in`.usecase
+
+import com.retrip.map.infra.adapter.out.search.elasticsearch.entity.LocationDocument
+
+interface LocationIndexUseCase {
+    fun indexLocationDocuments(documents: List<LocationDocument>?)
+}

src/main/kotlin/com/retrip/map/application/out/repository/LocationElasticRepository.kt

@@ -0,0 +1,11 @@
+package com.retrip.map.application.out.repository
+
+import com.retrip.map.infra.adapter.out.search.elasticsearch.entity.LocationDocument
+import org.springframework.data.elasticsearch.repository.ElasticsearchRepository
+import org.springframework.stereotype.Repository
+import java.util.UUID
+
+@Repository
+interface LocationElasticRepository: ElasticsearchRepository<LocationDocument, UUID> {
+
+}

src/main/kotlin/com/retrip/map/application/out/repository/LocationQueryRepository.kt

@@ -1,10 +1,13 @@
 package com.retrip.map.application.out.repository
 
 import com.retrip.map.application.`in`.response.LocationResponse
+import com.retrip.map.domain.entity.Location
 import org.springframework.data.domain.Page
 import org.springframework.data.domain.Pageable
+import java.time.LocalDate
 import java.util.UUID
 
 interface LocationQueryRepository {
     fun findLocations(id: UUID?, page: Pageable): Page<LocationResponse>
+    fun findLocationsByEditedAt( editedAt: LocalDate): List<Location>
 }

src/main/kotlin/com/retrip/map/domain/entity/Location.kt

@@ -9,13 +9,15 @@ import jakarta.persistence.Column
 import jakarta.persistence.Embedded
 import jakarta.persistence.Entity
 import jakarta.persistence.Id
+import jakarta.persistence.Table
 import jakarta.persistence.Version
 import lombok.AccessLevel
 import lombok.NoArgsConstructor
 import lombok.Setter
 import java.util.*
 
 @Entity
+@Table(name = "location")
 @NoArgsConstructor(access = AccessLevel.PROTECTED)
 @Setter(value = AccessLevel.PROTECTED)
 class Location(

src/main/kotlin/com/retrip/map/infra/adapter/in/batch/DestinationReader.kt

@@ -0,0 +1,51 @@
+package com.retrip.map.infra.adapter.`in`.batch
+
+import com.retrip.map.domain.entity.Location
+import com.retrip.map.infra.adapter.out.search.elasticsearch.entity.LocationDocument
+import org.springframework.batch.core.Job
+import org.springframework.batch.core.Step
+import org.springframework.batch.core.configuration.annotation.EnableBatchProcessing
+import org.springframework.batch.core.job.builder.JobBuilder
+import org.springframework.batch.core.repository.JobRepository
+import org.springframework.batch.core.step.builder.StepBuilder
+import org.springframework.beans.factory.annotation.Value
+import org.springframework.context.annotation.Bean
+import org.springframework.context.annotation.Configuration
+import org.springframework.transaction.PlatformTransactionManager
+
+@Configuration
+//@EnableBatchProcessing
+class MapBatchConfig(
+    @Value("\${spring.batch.job.name}")
+    private val jobName: String,
+    private val reader: MapReader,
+    private val processor: MapProcessor,
+    private val writer: MapWriter,
+) {
+
+    @Bean
+    fun mapJob(
+        jobRepository: JobRepository,
+        mapStep: Step
+    ): Job {
+        return JobBuilder(jobName, jobRepository)
+            .start(mapStep)
+            .build()
+
+    }
+
+    @Bean
+    fun mapStep(
+        jobRepository: JobRepository,
+        transactionManager: PlatformTransactionManager
+    ): Step {
+        return StepBuilder("map-step", jobRepository)
+            .chunk<List<Location>, List<LocationDocument>>(10, transactionManager)
+            .reader(reader)
+            .processor(processor)
+            .writer(writer)
+            .build()
+    }
+
+
+}

src/main/kotlin/com/retrip/map/infra/adapter/in/batch/MapBatchConfig.kt

@@ -0,0 +1,15 @@
+package com.retrip.map.infra.adapter.`in`.batch
+
+import com.retrip.map.domain.entity.Location
+import com.retrip.map.infra.adapter.out.search.elasticsearch.entity.LocationDocument
+import org.springframework.batch.item.ItemProcessor
+import org.springframework.stereotype.Component
+
+@Component
+class MapProcessor()
+    : ItemProcessor<List<Location>, List<LocationDocument>> {
+    override fun process(item: List<Location>): List<LocationDocument>? {
+        return item.map { LocationDocument.of(it) }
+
+    }
+}