在 jOOQ 中使用 unnest 作为字段而不是表

Roh*_*bhu 4 postgresql jooq

这是我尝试在 PostgreSQL 中运行的查询:

SELECT * FROM message WHERE id IN (
    SELECT unnest(message_ids) "mid"
        FROM session_messages WHERE session_id = '?' ORDER BY "mid" ASC
);
Run Code Online (Sandbox Code Playgroud)

但是,我无法做某事:

create.selectFrom(Tables.MESSAGE).where(Tables.MESSAGE.ID.in(
    create.select(DSL.unnest(..))
Run Code Online (Sandbox Code Playgroud)

因为DSL.unnest是 a Table<?>,这是有道理的,因为它试图获取List类似对象(主要是文字)并将其转换为表。

我有一种感觉,我需要找到一种方法将函数包装在我的字段名称周围,但我不知道如何继续。

笔记。该字段message_ids的类型为bigint[]

编辑

所以,这就是我现在正在做的事情,它的工作原理与预期完全一样,但我不确定这是否是最好的方法:

Field<Long> unnestMessageIdField = DSL.field(
                "unnest(" +  SESSION_MESSAGES.MESSAGE_IDS.getName() + ")",
                Long.class)
        .as("mid");

Field<Long> messageIdField = DSL.field("mid", Long.class);

MESSAGE.ID.in(
        ctx.select(messageIdField).from(
            ctx.select(unnestMessageIdField)
               .from(Tables.CHAT_SESSION_MESSAGES)
                    .where(Tables.CHAT_SESSION_MESSAGES.SESSION_ID.eq(sessionId))
            )
            .where(condition)
)
Run Code Online (Sandbox Code Playgroud)

编辑2

在浏览完https://github.com/jOOQ/jOOQ/blob/master/jOOQ/src/main/java/org/jooq/impl/DSL.java上的代码后,我想正确的方法是:

DSL.function("unnest", SQLDataTypes.BIGINT.getArrayType(), SESSION_MESSAGES.MESSAGE_IDS)
Run Code Online (Sandbox Code Playgroud)

编辑3

由于卢卡斯一如既往地在这里解决我的 jOOQ 困境,我将利用这一点:)

尝试以某种签名形式概括此函数

public <T> Field<T> unnest(Field<T[]> arrayField) {
    return DSL.function("unnest", <??>, arrayField);
}
Run Code Online (Sandbox Code Playgroud)

我不知道如何获取数据类型。似乎有一种方法可以DataType<T[]>DataType<T>using中获得DataType::getArrayDataType(),但反过来是不可能的。我找到了这个类ArrayDataType,但它似乎是包私有的,所以我不能使用它(即使我可以,它也不会公开该字段elementType)。

Luk*_*der 5

旧的 PostgreSQL 版本有一个时髦的想法,即可以从子句中生成一个表SELECT,并将其扩展到“外部”表,就像在子句中声明它一样FROM。这是一个非常不起眼的 PostgreSQL 遗留问题,这个例子是摆脱它并使用它的好机会LATERAL。您的查询相当于以下查询:

SELECT * 
FROM message 
WHERE id IN (
    SELECT "mid"
    FROM session_messages 
    CROSS JOIN LATERAL unnest(message_ids) AS t("mid")
    WHERE session_id = '?' 
);
Run Code Online (Sandbox Code Playgroud)

这可以更容易地转换为 jOOQ,如下所示:

DSL.using(configuration)
   .select()
   .from(MESSAGE)
   .where(MESSAGE.ID).in(
        select(field(name("mid"), MESSAGE.ID.getDataType()))
       .from(SESSION_MESSAGES)
       .crossJoin(lateral(unnest(SESSION_MESSAGES.MESSAGE_IDS)).as("t", "mid"))
       .where(SESSION_MESSAGES.SESSION_ID.eq("'?'"))
   )
Run Code Online (Sandbox Code Playgroud)