是否有方法使用自定义策略来映射JDBI3中的枚举类型？

Question

我有一个简单的场景，其中一个枚举被发送到存储，我想要取出它。但是，我不能使用Ordinal或Name策略。存储的值是每个枚举实例中的值。

TLDR:有没有一种方法可以让我注册一个地图绘制者，做一件事在坚持，另一件事在检索？

免责声明:我不使用posgres

// A simple enum
enum class N(val v: Int) {
    A(10),B(15);
    companion object {
        val map = values(v: Int).associateBy { it.v }
        fun from(v: Int) = map.getValue(v)
    }
}

// A simple data class
data class A(val i: Int = 0, val n: N = N.A) {
    interface Dao {
        @SqlUpdate("INSERT INTO a (i) VALUE (?)")
        @GetGeneratedKey
        fun insert(i: Int): Int
        @SqlQuery("SELECT i,n FROM a WHERE i = ?")
        fun select(i: Int): A
    }
}

// register my enum so that it stores the int value
jdbi.registerArgument(object : AbstractArgumentFactory<N>(Types.TINYINT) {
    override fun build(value: N, config: ConfigRegistry): Argument {
        return Argument { pos, stmt, _ -> stmt.setInt(pos, value.value) }
    }
})

// test with Ordinal strategy
jdbi.configure(Enums::class.java) { it.setEnumStrategy(EnumStrategy.BY_ORDINAL) }

最简单的表格

create table a(i int primary key auto_increment, n int);

我正在尝试提取数据，但是我得到了以下错误：Exception in thread "main" org.jdbi.v3.core.result.UnableToProduceResultException: no N value could be matched to the name 15

val dao = jdbi.onDemand(A.Dao::class.java)
val i = dao.insert(N.B)
printf(dao.select())

我知道我正在使用AbstractArgumentFactory，我认为这是我问题的根源。我不认为这个帮手适合我的需要。

有什么办法让我注册一个地图，做一件事在坚持，另一件事在检索？

Answer 1

我想我找到了解决办法。比我预期的要简单得多

jdbi.registerColumnMapper(object : ColumnMapper<N> {
    override fun map(r: ResultSet, columnNumber: Int, ctx: StatementContext): N {
        val v = r.getInt(columnNumber)
        return N.from(v)
    }
})

在我深入到Jdbi项目并看了一下

org.jdbi.v3.core.mapper.EnumMapper

我让它在一个划痕文件中工作