3

我正在尝试遵循 pg-promise 库作者在此处推荐的性能模式。

基本上 Vitaly 建议使用 inserts 这样做：

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

// We can use Inserts as an inline function also:

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
    });

使用以下辅助函数：

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(',');
    };
}

我的问题是，当插入物有限制时，你会怎么做？我的代码：

db.tx(function (t) {
    return t.any("SELECT... ",[params])
        .then(function (data) {

          var requestParameters = [];

          async.each(data,function(entry){
            requestParameters.push(entry.application_id,entry.country_id,collectionId)
          });

          db.none(
            " INSERT INTO application_average_ranking (application_id,country_id,collection_id) VALUES ($1)" +
            " ON CONFLICT ON CONSTRAINT constraint_name" +
            " DO UPDATE SET country_id=$2,collection_id=$3",
            [Inserts('$1, $2, $3',requestParameters),entry.country_id,collectionId])

            .then(data=> {
              console.log('success');
            })
            .catch(error=> {
              console.log('insert error');
            });

        });

});

显然，我无法访问参数，因为我不在异步循环中。

我也尝试做这样的事情：

        db.none(
        " INSERT INTO application_average_ranking (application_id,country_id,collection_id) VALUES ($1)" +
        " ON CONFLICT ON CONSTRAINT constraint_name" +
        " DO UPDATE SET (application_id,country_id,collection_id) = $1",
        Inserts('$1, $2, $3',requestParameters));

但当然，它不尊重 postgresql 的标准。

有没有办法做到这一点？

谢谢 ！

4

1 回答 1

3

我为Performance Boost文章编写了该示例以生成简单形式的多插入，仅此而已，这对于本文来说已经足够了。

你在这里尝试做的事情有点复杂，显然function Inserts(template, data)没有那种逻辑。

我无法从脑海中告诉您如何更改它以适应您的场景，但它可能会变得非常复杂，甚至可能根本不值得做。

幸运的是，您不必这样做。我一次又一次地被问到某些格式化助手，这是我最近发布的 - helpers namespace。您需要更新到最新版本的库（当前为 4.1.9）才能以您需要的方式使用它。

将您的示例更改为以下内容：

var h = pgp.helpers;
var cs = new h.ColumnSet(['?application_id', 'country_id', 'collection_id'],
    {table: 'application_average_ranking'});

db.tx(t => {
    return t.any("SELECT... ", [params])
        .then(function (data) {

            var insertData = data.map(d => {
                return {
                    application_id: d.application_id,
                    country_id: d.country_id,
                    collection_id: collectionId
                };
            });

            var updateData = {
                country_id: entry.country_id,
                collection_id: collectionId
            };

            var query = h.insert(insertData, cs) +
                " ON CONFLICT ON CONSTRAINT constraint_name DO UPDATE SET " +
                h.sets(updateData, cs);

            db.none(query)
                .then(data => {
                    console.log('success');
                })
                .catch(error => {
                    console.log('insert error');
                });
        });
});

那应该这样做。

另请参阅：列集。

于 2016-05-13T13:48:47.973 回答