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

编程问答 2022-06-12

问题描述

我有一个简单的场景，其中正在发送一个枚举进行存储，我想将其取出。但是，我不能使用顺序或名称策略。存储的值是每个枚举实例内部的值。

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 ，我认为这是问题的根源。我认为该帮助程序不能满足我的需求。

我是否可以注册一个映射器，该映射器在持久性上做一件事而在检索上做另一件事？

解决方法

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

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

我使它可以在临时文件中工作

javajavajdbijdbi3kotlinkotlinkotlinmysql