4

我正在尝试为 PGInterval 和 Duration 编写自定义数据类型绑定,以将 jOOQ 与 TimescaleDB 一起使用。遗憾的是,jOOQ 在为数据库例程生成函数时不使用它。

这是我的绑定类:

import org.jooq.*
import org.jooq.conf.ParamType
import org.jooq.impl.DSL
import org.postgresql.util.PGInterval
import java.sql.SQLFeatureNotSupportedException
import java.sql.Types
import java.time.Duration
import java.util.*

class PostgresIntervalDurationBinding: Binding<Any, Duration> {
    override fun converter(): Converter<Any, Duration> {
        return object : Converter<Any, Duration> {
            override fun from(t: Any?): Duration {
                return if (t == null) Duration.ZERO else Duration.ofSeconds(pgIntervalToSeconds(t as PGInterval))
            }

            override fun to(u: Duration?): Any? {
                return if (u == null || u === Duration.ZERO) null else PGInterval().seconds = u.seconds.toDouble()
            }

            override fun fromType(): Class<Any> {
                return Any::class.java
            }

            override fun toType(): Class<Duration> {
                return Duration::class.java
            }
        }

    }

    override fun sql(ctx: BindingSQLContext<Duration>?) {
        if (ctx?.render()?.paramType() == ParamType.INLINED)
            ctx.render()?.visit(DSL.inline(ctx.convert(converter()).value()))?.sql("::interval")
        else
            ctx?.render()?.sql("?::interval")
    }

    override fun register(ctx: BindingRegisterContext<Duration>?) {
        ctx?.statement()?.registerOutParameter(ctx.index(), Types.VARCHAR)
    }

    override fun set(ctx: BindingSetStatementContext<Duration>?) {
        ctx?.statement()?.setString(ctx.index(), Objects.toString(ctx.convert(converter()).value(), null))
    }

    override fun get(ctx: BindingGetResultSetContext<Duration>?) {
        ctx?.convert(converter())?.value(ctx.resultSet().getString(ctx.index()))
    }

    override fun get(ctx: BindingGetStatementContext<Duration>?) {
        ctx?.convert(converter())?.value(ctx.statement().getString(ctx.index()))
    }

    override fun set(ctx: BindingSetSQLOutputContext<Duration>?) {
        throw SQLFeatureNotSupportedException()
    }

    override fun get(ctx: BindingGetSQLInputContext<Duration>?) {
        throw SQLFeatureNotSupportedException()
    }

    companion object {
        fun pgIntervalToSeconds(t: PGInterval): Long {
            var seconds = 0L
            with(t){
                seconds += Duration.ofSeconds(this.seconds.toLong()).seconds
                seconds += Duration.ofMinutes(this.minutes.toLong()).seconds
                seconds += Duration.ofHours(this.hours.toLong()).seconds
                seconds += Duration.ofDays(this.days.toLong()).seconds
                if (months > 0 || years > 0) throw SQLFeatureNotSupportedException()
            }
            return seconds
        }
    }
}

这是我在 pom 中的配置:

<database>
<name>org.jooq.meta.postgres.PostgresDatabase</name>
<includes>.*</includes>
<inputSchema>public</inputSchema>
<excludes>set_adaptive_chunking
    | flyway_schema_history
</excludes>
<forcedTypes>
    <forcedType>
        <userType>java.time.Duration</userType>

        <binding>de.ninjaneers.dmc.jooq.databindings.PostgresIntervalDurationBinding
        </binding>
        <expression>.*interval.*</expression>

        <types>.*</types>
    </forcedType>
</forcedTypes>
</database>

例如,我希望 jOOQ 生成例程

time_bucket(bucket_with interval, ts timestamp with time zone) 

作为

timeBucket(Field<Duration> bucketWidth, Field<Timestamp> ts) 

但我明白了

timeBucket(Field<Object> bucketWidth, Field<Timestamp> ts)
4

1 回答 1

0

您混淆了两个配置标志:

<expression>.*interval.*</expression>
<types>.*</types>

<expression>是与您要替换其类型的列/属性/参数的标识符匹配的正则表达式。<types>匹配数据类型。这就是你想要的:

<types>.*interval.*</types>
<expression>.*</expression>
于 2018-12-13T13:22:52.663 回答