Long 의 wrapper 클래스를 value class 로 만들어서 사용중인데 이 타입에 대한 sum 쿼리를 할 수 있는 방법이 있을까요?

[WoongE]

안녕하세요!
재미있는 라이브러리를 만들어주신 덕분에 흥미롭게 PoC 를 해보고 있습니다. :)

라이브러리 사용법 관련해서 질문이 있습니다.

제가 현재 작업하고 있는 프로젝트에서는 돈 관련 필드에 대해서는 전부 Money 클래스(Long 타입을 value class 로 래핑)를 만들어서 사용하고 있습니다. 그리고 이 Money 클래스를 Jpa Entity 클래스에서도 필드 타입으로 사용하고 있습니다.

JDSL 에서 Money 의 eq, gt, lt 등 비교구문은 문제없이 작동했으나, sum 이 불가능하여 방법을 찾아봤습니다만,
JpqlSum 에 MoneySum 을 추가할 방법이 현재로서는 사실상 없는 거 같더라고요.

혹시 이 부분에 대한 가이드가 있을까요?

[WoongE]

data class MoneySum(
    val expr: Expression<Money>,
    val distinct: Boolean = false,
) : Expression<Long>

class JpqlMoneySumSerializer : JpqlSerializer<MoneySum> {
    override fun handledType(): KClass<MoneySum> = MoneySum::class

    override fun serialize(part: MoneySum, writer: JpqlWriter, context: RenderContext) {
        val delegate = context.getValue(JpqlRenderSerializer)

        writer.write("SUM")

        writer.writeParentheses {
            if (part.distinct) {
                writer.write("DISTINCT")
                writer.write(" ")
            }

            delegate.serialize(part.expr, writer, context)
        }
    }
}

class MyJpql : Jpql() {
    companion object Constructor : JpqlDsl.Constructor<MyJpql> {
        override fun newInstance(): MyJpql = MyJpql()
    }

    fun <T : Any> sum(expr: KProperty1<T, Money?>): Expression<Long> = MoneySum(Paths.path(expr), false)

    fun sum(expr: Expression<Money>, distinct: Boolean = false): Expression<Long> = MoneySum(expr, distinct)
}

@Configuration
class CustomJpqlRenderContextConfig {
    @Bean
    fun jpqlSerializer(): JpqlSerializer<*> = ValueClassAwareJpqlValueSerializer(JpqlValueSerializer())

    @Bean
    fun jpqlMoneySumSerializer(): JpqlSerializer<*> = JpqlMoneySumSerializer()
}

해결했습니다. ^^;

[shouwn]

WoongE 님 안녕하세요. 해결되어서 다행이네요.

Embedded 클래스에 대한 sum 과 같은 집계 연산은 기본 기능으로 제공하기 어렵습니다.
왜냐하면 클래스에 필드가 여럿 있을 수 있고 문자열이 들어가 있을 수 있기 때문입니다.

작성해주신 방법으로도 동작하는 것은 오늘 처음 알았습니다.
왜냐하면 JPQL, 정확히는 Hibernate도 위와 같은 제약사항으로 파싱 시 타입 관련 에러가 발생할 것으로 생각되기 때문입니다.
하지만 동작한다니 다행이네요.

Embedded 클래스의 집계 함수 정의 시에는 아래 방식들도 추천드립니다.
customExpression 이라는 함수도 있어서 Serializer 대신에 사용하실 수 있습니다.
https://kotlin-jdsl.gitbook.io/docs/jpql-with-kotlin-jdsl/expressions#custom-expression

@Embeddable
data class BookPrice(
    @Column(name = "price", scale = 2)
    val value: BigDecimal,
)

bookRepository.findAll {
    select(
        sum(path(Book::price)(BookPrice::value)),
    ).from(
        entity(Book::class),
    )
}

class MyJpql : Jpql() {
    companion object Constructor : JpqlDsl.Constructor<MyJpql> {
        override fun newInstance(): MyJpql = MyJpql()
    }

    fun sum(expr: Expression<BookPrice>, distinct: Boolean = false): Expression<BigDecimal> {
        return if (distinct) {
            customExpression(BigDecimal::class, "SUM(DISTINCT {0}.value)", expr)
        } else {
            customExpression(BigDecimal::class, "SUM({0}.value)", expr)
        }
    }
}

bookRepository.findAll(MyJpql) {
    select(
        sum(path(Book::price)),
    ).from(
        entity(Book::class),
    )
}

요즘 다른 업무로 Kotlin JDSL에 신경을 많이 못 쓰고 있는데 이렇게 사용해주셔서 감사합니다. 😭