如何使用 node-postgres 将多行正确插入 PG？

Question

可以像这样插入一行：

client.query("insert into tableName (name, email) values ($1, $2) ", ['john', 'john@gmail.com'], callBack)

这种方法会自动排除任何特殊字符。

如何一次插入多行？

我需要实现这个：

"insert into tableName (name, email) values ('john', 'john@gmail.com'), ('jane', 'jane@gmail.com')"

我可以使用 js 字符串运算符手动编译这些行，但是我需要以某种方式添加特殊字符转义。

Answer 1

使用pg-format，如下所示。

var format = require('pg-format');

var values = [
  [7, 'john22', 'john22@gmail.com', '9999999922'], 
  [6, 'testvk', 'testvk@gmail.com', '88888888888']
];
client.query(format('INSERT INTO users (id, name, email, phone) VALUES %L', values),[], (err, result)=>{
  console.log(err);
  console.log(result);
});

Answer 2

另一种使用 PostgreSQL json 函数的方式：

client.query('INSERT INTO table (columns) ' +
  'SELECT m.* FROM json_populate_recordset(null::your_custom_type, $1) AS m',
  [JSON.stringify(your_json_object_array)], function(err, result) {
    if (err) {
      console.log(err);
    } else {
      console.log(result);
    }
});

Answer 3

您将不得不动态生成查询。虽然有可能，但这是有风险的，如果你做错了，很容易导致 SQL 注入漏洞。查询中的参数索引和传入的参数之间也很容易出现一个错误。

话虽如此，下面是一个示例，说明如何编写此代码，假设您有一组看起来像 {name: string, email: string} 的用户：

client.query(
  `INSERT INTO table_name (name, email) VALUES ${users.map(() => `(?, ?)`).join(',')}`,
  users.reduce((params, u) => params.concat([u.name, u.email]), []),
  callBack,
)

另一种方法是使用像@databases/pg（我写的）这样的库：

await db.query(sql`
  INSERT INTO table_name (name, email)
  VALUES ${sql.join(users.map(u => sql`(${u.name}, ${u.email})`), ',')}
`)

@databases 要求使用sql 标记查询，并使用它来确保您传递的任何用户数据始终自动转义。这也让您可以内联编写参数，我认为这使代码更具可读性。

Answer 4

按照这篇文章：来自pg-promise 库的Performance Boost 及其建议的方法：

// Concatenates an array of objects or arrays of values, according to the template,
// to use with insert queries. Can be used either as a class type or as a function.
//
// template = formatting template string
// data = array of either objects or arrays of values
function Inserts(template, data) {
    if (!(this instanceof Inserts)) {
        return new Inserts(template, data);
    }
    this._rawDBType = true;
    this.formatDBType = function () {
        return data.map(d=>'(' + pgp.as.format(template, d) + ')').join(',');
    };
}

一个使用它的例子，就像你的情况一样：

var users = [['John', 23], ['Mike', 30], ['David', 18]];

db.none('INSERT INTO Users(name, age) VALUES $1', Inserts('$1, $2', users))
    .then(data=> {
        // OK, all records have been inserted
    })
    .catch(error=> {
        // Error, no records inserted
    });

它也适用于对象数组：

var users = [{name: 'John', age: 23}, {name: 'Mike', age: 30}, {name: 'David', age: 18}];

db.none('INSERT INTO Users(name, age) VALUES $1', Inserts('${name}, ${age}', users))
    .then(data=> {
        // OK, all records have been inserted
    })
    .catch(error=> {
        // Error, no records inserted
    });

UPDATE-1

有关通过单个 INSERT 查询的高性能方法，请参阅 Multi-row insert with pg-promise。

UPDATE-2

这里的信息现在已经很老了，请参阅Custom Type Formatting 的最新语法。以前的_rawDBType 现在是rawType，formatDBType 更名为toPostgres。

Answer 5

client.query("insert into tableName (name, email) values ($1, $2),($3, $4) ", ['john', 'john@gmail.com','john', 'john@gmail.com'], callBack)

没有帮助吗？ 此外，您可以手动生成一个字符串进行查询：

insert into tableName (name, email) values (" +var1 + "," + var2 + "),(" +var3 + ", " +var4+ ") "

如果您在这里阅读 https://github.com/brianc/node-postgres/issues/530 ，您可以看到相同的实现。