使用 pg-promise 的嵌套事务

Nested transactions with pg-promise

我正在使用 NodeJS、PostgreSQL 和令人惊叹的 pg-promise 库。就我而言,我想执行三个主要查询:

  1. 在 table 'tweets' 中插入一条推文。
  2. 如果推文中有主题标签,请将它们插入另一个 table 'hashtags'
  3. 他们 link 在第三条推文和主题标签中 table 'hashtagmap'(多对多关系 table)

这是请求正文的示例 (JSON):

{
"id":"12344444",
"created_at":"1999-01-08 04:05:06 -8:00",
"userid":"@postman",
"tweet":"This is the first test from postman!",
"coordinates":"",
"favorite_count":"0",
"retweet_count":"2",
"hashtags":{
    "0":{
        "name":"test",
        "relevancetraffic":"f",
        "relevancedisaster":"f"
    },
    "1":{
        "name":"postman",
        "relevancetraffic":"f",
        "relevancedisaster":"f"
    },
    "2":{
        "name":"bestApp",
        "relevancetraffic":"f",
        "relevancedisaster":"f"
    }
}

除主题标签外,上述所有字段都应包含在 table "tweets" 中,而主题标签又应包含在 table "hashtags".[=15= 中]

这是我使用的代码,基于来自 NodeJS 模块内的 pg-promise 文档的嵌套事务。我想我需要嵌套交易,因为我需要知道 tweet_idhashtag_id 以便 link 它们在 hashtagmap table.

// Columns
var tweetCols = ['id','created_at','userid','tweet','coordinates','favorite_count','retweet_count'];

var hashtagCols = ['name','relevancetraffic','relevancedisaster'];

//pgp Column Sets
var cs_tweets = new pgp.helpers.ColumnSet(tweetCols, {table: 'tweets'});

var cs_hashtags = new pgp.helpers.ColumnSet(hashtagCols, {table:'hashtags'});
return{
// Transactions
add: body =>
    rep.tx(t => {
        return t.one(pgp.helpers.insert(body,cs_tweets)+" ON CONFLICT(id) DO UPDATE SET coordinates = "+body.coordinates+" RETURNING id")
            .then(tweet => {
                var queries = [];
                for(var i = 0; i < body.hashtags.length; i++){
                    queries.push(
                        t.tx(t1 => {
                            return t1.one(pgp.helpers.insert(body.hashtags[i],cs_hashtags) + "ON CONFLICT(name) DO UPDATE SET fool ='f' RETURNING id")
                                .then(hash =>{
                                    t1.tx(t2 =>{
                                        return t2.none("INSERT INTO hashtagmap(tweetid,hashtagid) VALUES("+tweet.id+","+hash.id+") ON CONFLICT DO NOTHING");
                                    });
                                });
                        }));
                }
                return t.batch(queries);
            });
    })
}

问题出在这段代码上,我能够成功插入推文,但之后什么也没有发生。我无法将主题标签或 link 主题标签插入推文。

抱歉,我是编码新手,所以我想我不明白如何从交易中正确 return 以及如何执行这个简单的任务。希望您能够帮助我。

提前致谢。

对于那些会遇到类似问题的人,我会post回答。

首先是我的错误:

  1. 在 for 循环中:body.hashtag.length 不存在,因为我正在处理一个对象(这里是非常基本的错误)。更改为 Object.keys(body.hashtags).length
  2. 为什么要使用这么多交易?按照 vitaly-t 在 中的回答,我删除了额外的交易。我还不清楚如何打开一个事务并将一个查询的结果用于同一事务中的另一个查询。

这是最终代码:

    // Columns
var tweetCols = ['id','created_at','userid','tweet','coordinates','favorite_count','retweet_count'];

var hashtagCols = ['name','relevancetraffic','relevancedisaster'];

//pgp Column Sets
var cs_tweets = new pgp.helpers.ColumnSet(tweetCols, {table: 'tweets'});

var cs_hashtags = new pgp.helpers.ColumnSet(hashtagCols, {table:'hashtags'});

return {
    /* Tweets */
    // Add a new tweet and update the corresponding hashtags
    add: body =>
        rep.tx(t => {
            return t.one(pgp.helpers.insert(body,cs_tweets)+" ON CONFLICT(id) DO UPDATE SET coordinates = "+body.coordinates+" RETURNING id")
                .then(tweet => {
                    var queries = [];
                    for(var i = 0; i < Object.keys(body.hashtags).length; i++){
                        queries.push(
                            t.one(pgp.helpers.insert(body.hashtags[i],cs_hashtags) + "ON CONFLICT(name) DO UPDATE SET fool ='f' RETURNING id")
                                .then(hash =>{
                                    t.none("INSERT INTO hashtagmap(tweetid,hashtagid) VALUES("+tweet.id+","+hash.id+") ON CONFLICT DO NOTHING");
                                })
                            );
                    }
                    return t.batch(queries);
                });
        }),

改进 Jean Phelippe 自己的回答:

// Columns
var tweetCols = ['id', 'created_at', 'userid', 'tweet', 'coordinates', 'favorite_count', 'retweet_count'];

var hashtagCols = ['name', 'relevancetraffic', 'relevancedisaster'];

//pgp Column Sets
var cs_tweets = new pgp.helpers.ColumnSet(tweetCols, {table: 'tweets'});

var cs_hashtags = new pgp.helpers.ColumnSet(hashtagCols, {table: 'hashtags'});

return {
    /* Tweets */
    // Add a new tweet and update the corresponding hash tags
    add: body =>
        db.tx(t => {
            return t.one(pgp.helpers.insert(body, cs_tweets) + ' ON CONFLICT(id) DO UPDATE SET coordinates = ' + body.coordinates + ' RETURNING id')
                .then(tweet => {
                    var queries = Object.keys(body.hashtags).map((_, idx) => {
                        return t.one(pgp.helpers.insert(body.hashtags[i], cs_hashtags) + 'ON CONFLICT(name) DO UPDATE SET fool =  RETURNING id', 'f')
                            .then(hash => {
                                return t.none('INSERT INTO hashtagmap(tweetid, hashtagid) VALUES(, ) ON CONFLICT DO NOTHING', [+tweet.id, +hash.id]);
                            });
                    });
                    return t.batch(queries);
                });
        })
            .then(data => {
                // transaction was committed;
                // data = [null, null,...] as per t.none('INSERT INTO hashtagmap...
            })
            .catch(error => {
                // transaction rolled back
            })
},

注释:

  • 根据我之前的笔记,您必须链接所有查询,否则您最终会得到松散的承诺
  • 远离嵌套事务,除非您确切了解它们在 PostgreSQL 中的工作方式(read this,特别是 Limitations 部分)。
  • 避免手动查询格式化,这不安全,始终依赖图书馆的查询格式化。
  • 除非您将交易结果传递到其他地方,否则您至少应该提供 .catch 处理程序。

P.S。对于像 +tweet.id 这样的语法,它与 parseInt(tweet.id) 相同,只是更短,以防它们是字符串 ;)