如何使用 Jooq insertInto 复制和修改表行

Question

我正在使用 Jooq，并试图在同一个表中生成数据集的近似副本。在此过程中，我想将一个字段的值更新为已知值。我一直在查看文档并尝试变体，但还没有运气。这是我更新 REGISTRATION 表并将“阶段”字段设置为值 6（它是 5）的方法。所以我最终会得到原始数据加上一个只有不同阶段值的重复集。 在伪代码中

insert into Registration (select * from Registration where stage=5) set stage=6 

我在下面尝试了这段代码，并认为我可以添加一个“.set(...)”方法来设置值，但这似乎无效。

create.insertInto(REGISTRATION)
    .select(
        (selectFrom(REGISTRATION)
            .where(REGISTRATION.STAGE.eq(5))
        )
    ).execute();

Answer 1

我不知道支持INSERT .. SELECT .. SET 语法的数据库，如果有这样的语法，它肯定不符合SQL 标准。这里的前进方式是这样写：

在 SQL 中：

INSERT INTO registration (col1, col2, col3, stage, col4, col5)
SELECT col1, col2, col3, 6, col4, col5
FROM registration
WHERE stage = 5;

在 jOOQ 中：

create.insertInto(REGISTRATION)
      .columns(
         REGISTRATION.COL1,
         REGISTRATION.COL2,
         REGISTRATION.COL3,
         REGISTRATION.STAGE,
         REGISTRATION.COL4,
         REGISTRATION.COL5)
      .select(
         select(
           REGISTRATION.COL1,
           REGISTRATION.COL2,
           REGISTRATION.COL3,
           val(6),
           REGISTRATION.COL4,
           REGISTRATION.COL5)
        .from(REGISTRATION)
        .where(REGISTRATION.STAGE.eq(5)))
      .execute();

隐含以下静态导入：

import static org.jooq.impl.DSL.*;

在 jOOQ 中，动态

由于您正在寻找动态 SQL 解决方案，因此可以这样做：

static <T> int copy(
    DSLContext create, Table<?> table, Field<T> field, 
    T oldValue, T newValue
) {
    List<Field<?>> into = new ArrayList<>();
    List<Field<?>> from = new ArrayList<>();

    into.addAll(Stream.of(table.fields())
                      .filter(f -> !field.equals(f))
                      .collect(toList()));
    from.addAll(into);

    into.add(field);
    from.add(val(newValue));

    return
    create.insertInto(table)
          .columns(into)
          .select(
             select(from)
            .from(table)
            .where(field.eq(oldValue))
          .execute();
}

Answer 2

感谢 Lukas 的回答，我将使用它的一个版本，因为它既好又通用。我刚刚开始工作的我自己的答案不太笼统，但对于其他人来说可能是一个有用的参考，特别是因为它考虑了身份字段“id”，否则可能会导致问题。

public void duplicate(int baseStage, int newStage) {

        Field<?>[] allFieldsExceptId = Stream.of(REGISTRATION.fields())
                                            .filter(field -> !field.getName().equals("id"))
                                            .toArray(Field[]::new);

        Field<?>[] newFields = Stream.of(allFieldsExceptId).map(field -> {
            if (field.getName().contentEquals("stage")) {
                return val(newStage);
            } else {
                return field;
            }
        }).toArray(Field[]::new);


        create.insertInto(REGISTRATION)
                .columns(allFieldsExceptId)
                .select(
                        select(newFields)
                            .from(REGISTRATION)
                            .where(REGISTRATION.STAGE.eq(baseStage)))
                .execute();
    }