在 Upsert 上跳过空值

Question

我正在使用 pg-promise 来处理我的 Postgres 查询，但我无法找到以下查询的解决方案：

我正在尝试创建一种方法来一次批量更新多行， 这是我的代码：

massUpsert: (orgId, entities) => db.tx((t) => {
    const queries = [];
    entities.forEach((entity) => {
      const { id, name, age, type } = entity;
      queries.push(
        t.one(`INSERT INTO public.table (id, name, age, type) 
             VALUES ($1, $2, $3, $4)
              ON CONFLICT (id) DO update
                SET 
                 name = $2,
                 age = $3,
                 type = $4
              RETURNING *`,
        [id, name, age, type]));
    });
    return t.batch(queries);
}),

现在的问题是，在某些情况下，我的一个或多个字段为 null，我希望数据库保留旧值而不是用 null 替换它。

有什么建议吗？

Answer 1

使用helpers 方法生成查询：

const skipIfNull = name => ({name, skip: c => c.value === null});
    
const cs = new pgp.helpers.ColumnSet([
    '?id',
    skipIfNull('name'),
    skipIfNull('age'),
    skipIfNull('type')
], {table: 'table'});

查看类型 ColumnSet 和 Column。

从样本data生成查询：

const data = {
    id: 1,
    name: null, // will be skipped
    age: 123,
    type: 'tt'
};
    
    const query = pgp.helpers.insert(data, cs) + ' ON CONFLICT(id) DO UPDATE SET ' +
        pgp.helpers.sets(data, cs) + ' RETURNING *';

将生成：

INSERT INTO "table"("id","name","age","type") VALUES(1,null,123,'tt')
ON CONFLICT(id) DO UPDATE SET "age"=123,"type"='tt' RETURNING *

更新：通过assignColumns 的新语法优于sets 方法。

但请注意，根据方法set API，如果您的所有条件列都是null，它将返回一个空字符串，因此生成的查询将无效。您将需要为此添加支票；）

另外，考虑到您正在生成多插入，可以只生成一个多行插入，从而提供更好的性能。为此，请参阅Multi-row insert with pg-promise。

另见：

• skip update columns with pg-promise