Spring Cloud Function을 이용해 Spring Application을 AWS Lambda에 배포하기

kshired·2022년 10월 6일

aws lambda spring batch

시작하며

현재 몬스테라 팀의 CS Broker 프로젝트는 AI 서버의 모델이 업데이트 되거나 채점 기준이 변경 되었을 경우, 많은 양의 채점 기록을 전부 재채점해야합니다.

이 상황에서 유용하게 사용할 수 있는 것이 Spring Batch 였습니다.

그러던 도중 하나의 문제에 봉착하게 됩니다.

배포를 어디에 할까?

이 프로젝트에서, 재채점 작업은 그렇게 자주 일어나는 작업이 아닙니다.

특정한 시점에, 요청이 있을 때만 재채점 작업을 진행하면 됩니다. 그렇기에, 이를 위해 서버를 하나 더 올리는 것은 비용 낭비라고 생각했습니다.

이런 비용 낭비를 줄이기 위해, AWS Lambda 서비스를 사용하기로 마음 먹었습니다.

Spring Cloud Function

운이 좋게도, Spring 프로젝트 중에 AWS Lambda나 Azure Functions과 같이 서버리스 서비스에 사용하기 쉽도록 해주는 Spring Cloud Function 이라는 것이 있다는 것을 알게 되었습니다.

이제 이것을 적용해보겠습니다.

주의! 이 프로젝트는 Kotlin 으로 개발되었습니다.

build.gradle.kts

아래와 같은 값들을 build.gradle.kts 에 추가합니다.


plugins {
    id("com.github.johnrengelman.shadow") version "7.1.2"
}

depenencies {
    implementation("org.springframework.cloud:spring-cloud-starter-function-web:3.2.5")
    implementation("org.springframework.cloud:spring-cloud-function-kotlin:3.2.5")
    implementation("org.springframework.cloud:spring-cloud-function-adapter-aws:3.2.5")
    implementation("com.amazonaws:aws-lambda-java-events:3.11.0")
    implementation("com.amazonaws:aws-lambda-java-core:1.2.1")
    runtimeOnly("com.amazonaws:aws-lambda-java-log4j2:1.5.1")
}

tasks.withType<Jar> {
	// Spring Application 시작점
    manifest {
        attributes["Start-Class"] = "io.csbroker.batchserver.BatchServerApplication" 
    }
}

tasks.assemble {
    dependsOn("shadowJar")
}

tasks.withType<com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar> {
    archiveFileName.set("batch.jar")
    dependencies {
        exclude("org.springframework.cloud:spring-cloud-function-web")
    }
    mergeServiceFiles()
    append("META-INF/spring.handlers")
    append("META-INF/spring.schemas")
    append("META-INF/spring.tooling")
    transform(com.github.jengelman.gradle.plugins.shadow.transformers.PropertiesFileTransformer::class.java) {
        paths.add("META-INF/spring.factories")
        mergeStrategy = "append"
    }
}

spring-cloud-function-web 은 로컬에서, 테스트 용도로 사용하기 위함입니다.
Start-Class 의 값은 적절하게 변경해야합니다.
shadowjar 는 모든 라이브러리를 포함한 fat-jar를 생성하기 위함입니다.
실제로 gradle 을 통해 jar 파일을 생성할 때는, spring-cloud-function-web 을 제외하게됩니다.

application.yml

아래와 같은 값들을 application.yml 에 추가합니다.

spring:
  cloud:
    function:
      definition: regradingProblem # 함수 이름
    main:
      banner-mode: off
logging:
  level:
    root: info

함수 작성

package io.csbroker.batchserver.function

import io.csbroker.batchserver.service.RegradingJobLauncherService
import io.csbroker.batchserver.util.log
import org.slf4j.MDC
import org.springframework.context.annotation.Bean
import org.springframework.stereotype.Component
import java.time.LocalDateTime
import java.time.ZoneId
import java.util.UUID

@Component
class RegradingFunction(
    private val regradingJobLauncherService: RegradingJobLauncherService
) {
    @Bean
    fun regradingProblem(): (Long) -> (String) {
        return {
            MDC.put("traceId", UUID.randomUUID().toString())
            log.info("==> re-grading request coming with problem id : $it")
            regradingJobLauncherService.regradingProblem(it, LocalDateTime.now(ZoneId.of("Asia/Seoul")))
            MDC.get("traceId")
        }
    }

}