Postgres 相当于 MySQL 的 SET？

Question

我正在尝试从 MySQL 迁移到 Postgres，但遇到了一些问题。我有一个表单，用户填写大约 40 个字段，这些值被插入到数据库中。使用 MySQL 我习惯这样做：

INSERT INTO table_name SET name="John Smith", email="jsmith@gmail.com", website="jsmith.org";

我还在使用带有 nodejs 的 mysql 模块，这就是我的代码目前的样子：

var data = {
  name: req.body.name,
  email: req.body.email,
  website: req.body.website,
  ...
  ...
}

var query = connection.query('INSERT INTO table_name SET ?', data)

由于SET 不是有效的SQL，如果我想使用Postgres，使用pg 模块的查询将如下所示：

client.query('INSERT INTO table_name (name, email, website) VALUES ($1, $2, $3)', req.body.name, req.body.email, req.body.website)

考虑到我有将近 40 个字段，这将变得非常乏味。

有没有更好的方法来做到这一点，还是我不得不手动编写查询？

谢谢。

Answer 1

因为你已经有一个包含相关数据的对象

var data = {
  name: req.body.name,
  email: req.body.email,
  website: req.body.website,
  ...
  ...
}

您可以编写一个查询函数来包装pg 客户端的query 方法并添加对象即值支持：

function insert(client, table, values) {
    const keys = Object.keys(values),
          // IMPORTANT: escape column identifiers. If `values` should come
          // from an uncontrolled source, naive concatenation would allow
          // SQL injection.
          columns = keys.map(k => client.escapeIdentifier(k)).join(', '),
          placeholders = keys.map((k, i) => '$' + (i + 1)).join(', ');

    return client.query(`INSERT INTO ${client.escapeIdentifier(table)} 
                         (${columns}) VALUES (${placeholders})`,
                        // According to docs `query` accepts an array of
                        // values, not values as positional arguments.
                        // If this is not true for your version, use the
                        // spread syntax (...args) to apply values.
                        keys.map(k => values[k]));
}

...

insert(client, 'table_name', data);

或者，您可能可以使用sql-template-strings 重写您的查询，这看起来相当不错。

Answer 2

这在使用pg-promise时很容易实现，完整代码如下：

var pgp = require('pg-promise')();
var db = pgp(/*connection details*/);

app.get('/get', (req, res)=> {
    var insert = pgp.helpers.insert(req.body, null, {table: 'table_name'});
    db.none(insert)
        .then(()=> {
            // success
        })
        .catch(error=> {
            // error
        });
});

API：helpers.insert、none

方法helpers.insert 可以为您生成完整的INSERT 语句，格式正确。

它不仅是最简单的方法，而且也是最灵活的方法，因为您无需更改任何内容即可开始生成多插入语句，即如果insert 方法的第一个参数是一个数组对象，您将获得多插入查询。

---

如果您只需要请求中的特定属性，那么最好的方法是单独指定它们，作为请求之外的ColumnSet（可重复使用，以获得最佳性能）：

var cs = new pgp.helpers.ColumnSet(['name', 'email', 'website'], {table: 'table_name'});

app.get('/get', (req, res)=> {
    var insert = pgp.helpers.insert(req.body, cs);
    db.none(insert)
        .then(()=> {
            // success
        })
        .catch(error=> {
            // error
        });
});

返回新 id-s 数组的示例

无转化

app.get('/get', (req, res)=> {
    var insert = pgp.helpers.insert(req.body, cs) + 'RETURNING id';
    db.many(insert)
        .then(data=> {
            // success
        })
        .catch(error=> {
            // error
        });
});

有转化

app.get('/get', (req, res)=> {
    var insert = pgp.helpers.insert(req.body, cs) + 'RETURNING id';
    db.map(insert, [], a=>a.id)
        .then(ids=> {
            // ids = array of id-s
        })
        .catch(error=> {
            // error
        });
});