0

我有一个简单的场景，其中正在发送枚举以进行存储，我想将其拉出。但是，我不能使用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，我认为这是我问题的根源。我认为这个助手不能灵活地满足我的需求。

有没有办法让我注册一个映射器，它在持久性上做一件事，在检索上做另一件事？

4

1 回答 1

2

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

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

我让它在一个临时文件中工作

于 2020-09-21T17:31:50.387 回答