如何在jOOQ查询的中间或末尾创建可复用组合方法？

已有方案解决了「开头相同、后续不同」的多查询复用场景，但如果要复用查询中间或末尾的逻辑呢？比如下面这个限制关联表返回行数的示例：

select(TABLE1.FOO, TABLE2.BAR)
.from(TABLE1)
// 想要复用的代码开始
.leftJoin(TABLE2)
.on(TABLE2.FOREIGN_ID.eq(TABLE1.ID))
.and(
    TABLE2.ID.eq(
        ctx.select(TABLE2.ID)
            .from(TABLE2)
            .where(TABLE2.FOREIGN_ID.eq(TABLE1.ID))
            .limit(1)))
// 想要复用的代码结束
.join(TABLE3)
.on(ETC)

这段中间代码的核心逻辑固定，仅需替换TABLE2、TABLE2.ID、TABLE2.FOREIGN_ID.eq(TABLE1.ID)这几部分。如果目标表有唯一键，TABLE2.ID可隐式获取；如果外键关联关系明确且主表可访问，关联条件也能隐式处理。

我希望能像复用开头逻辑那样，通过链式调用的方式复用这类中间逻辑，比如：

select(TABLE1.FOO, TABLE2.BAR)
.from(TABLE1)
.myUniqueJoinMethod(TABLE2, TABLE2.ID, TABLE2.FOREIGN_ID.eq(TABLE1.ID))
.join(TABLE3)
.on(ETC)

而非用包裹式的独立函数（这种写法不够简洁）：

myUniqueJoinFunction(TABLE2, TABLE2.ID, TABLE2.FOREIGN_ID.eq(TABLE1.ID),
  select(TABLE1.FOO, TABLE2.BAR)
  .from(TABLE1)
).join(TABLE3)
.on(ETC)

需要说明的是，这种限制关联行数的逻辑只是其中一种场景，我希望能实现任意查询组件的拆分复用，就像处理开头查询组件那样灵活。

可行的实现方式 #

1. 扩展jOOQ接口实现链式调用 #

jOOQ的查询API基于接口链式设计，可以通过自定义接口继承SelectJoinStep并添加默认方法，实现自定义链式操作：

public interface CustomSelectJoinStep<R extends Record> extends SelectJoinStep<R> {
    default <T extends Record> CustomSelectJoinStep<R> myUniqueJoin(
        Table<T> targetTable,
        Field<?> targetId,
        Condition joinCondition
    ) {
        return (CustomSelectJoinStep<R>) this
            .leftJoin(targetTable)
            .on(joinCondition)
            .and(targetId.eq(
                DSL.select(targetId)
                    .from(targetTable)
                    .where(joinCondition)
                    .limit(1)
            ));
    }
}

使用时将初始查询强制转换为自定义接口类型即可：

(CustomSelectJoinStep<Record2<String, String>>) select(TABLE1.FOO, TABLE2.BAR)
    .from(TABLE1)
    .myUniqueJoin(TABLE2, TABLE2.ID, TABLE2.FOREIGN_ID.eq(TABLE1.ID))
    .join(TABLE3)
    .on(ETC);

2. 静态工具类配合链式调用 #

如果不想扩展接口，可编写静态工具方法，接收当前SelectJoinStep实例并返回处理后的新实例，保持链式调用的连贯性：

public class JooqQueryUtils {
    public static <R extends Record, T extends Record> SelectJoinStep<R> myUniqueJoin(
        SelectJoinStep<R> step,
        Table<T> targetTable,
        Field<?> targetId,
        Condition joinCondition
    ) {
        return step
            .leftJoin(targetTable)
            .on(joinCondition)
            .and(targetId.eq(
                DSL.select(targetId)
                    .from(targetTable)
                    .where(joinCondition)
                    .limit(1)
            ));
    }
}

结合Java的函数式风格调用（需自行实现let工具方法，或直接嵌套）：

select(TABLE1.FOO, TABLE2.BAR)
    .from(TABLE1)
    .let(step -> JooqQueryUtils.myUniqueJoin(step, TABLE2, TABLE2.ID, TABLE2.FOREIGN_ID.eq(TABLE1.ID)))
    .join(TABLE3)
    .on(ETC);

3. 隐式参数优化（针对唯一键/外键场景） #

如果目标表有唯一键、外键关系明确，可进一步简化参数，自动推导主键和关联条件：

public static <R extends Record, T extends Record> SelectJoinStep<R> myUniqueJoin(
    SelectJoinStep<R> step,
    Table<T> targetTable,
    Table<?> parentTable
) {
    // 自动获取目标表主键
    Field<?> targetId = targetTable.getPrimaryKey().getFields()[0];
    // 自动生成外键关联条件（需jOOQ元数据支持）
    Condition joinCondition = targetTable.field(parentTable.getPrimaryKey().getName())
        .eq(parentTable.getPrimaryKey().getFields()[0]);
    
    return step
        .leftJoin(targetTable)
        .on(joinCondition)
        .and(targetId.eq(
            DSL.select(targetId)
                .from(targetTable)
                .where(joinCondition)
                .limit(1)
        ));
}

使用时只需传入目标表和父表：

JooqQueryUtils.myUniqueJoin(select(TABLE1.FOO, TABLE2.BAR).from(TABLE1), TABLE2, TABLE1)
    .join(TABLE3)
    .on(ETC);

内容的提问来源于stack exchange，提问作者Sparr