Question

我试图遵循pg-promise库作者here推荐的性能模式。

基本上Vitaly建议使用插页：

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的标准。

有没有办法实现这个目标？

谢谢！

Answer 1

我为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');
                });
        });
});

那应该这样做。

另请参阅：ColumnSet。

Pg-promise性能提升：ON CONFLICT

1 个答案: