在 Promise 中传递查询结果

Question

我正在使用 node-postgres 进行数据库连接。 为此，我有这样的 API：

app.post('/api/insert', urlEncodedParser, function(req, res) {
    // do stuff
})

现在我需要同步步骤：

1. Select 来自表用户的特定 ID
2. 检查表用户的 id 是否已存在于表列表中 3a) 如果表列表中的 id 不存在，则在表列表中插入数据 3b) 如果表列表中的 id 存在，则更新表列表中的数据

所以我有以下代码：

var userid = 0;
var listEntry = 0;

client
.query('SELECT "ID" FROM "User" WHERE "PW" = $1 AND "Name" = $2', [req.body.pw, req.body.name])
.then(res => userid = res.rows[0])
.catch(e => console.error(e.stack))
.query('SELECT " FROM "List" WHERE "UserID" = $1', [userid])
.then(res => listEntry = res.rows[0])
.catch(e => console.error(e.stack))
.query('INSERT INTO "List" ("UserID", "LastDate")', [userid, req.body.date])
.then(res => userid = res.rows[0])
.catch(e => console.error(e.stack))
.query('UPDATE "List" SET "LastDate" = $1', [req.body.date])
.then(res => userid = res.rows[0])
.catch(e => console.error(e.stack))
.finally(() { client.end()
});

我需要改变什么才能拥有上面列出的同步步骤？ 抱歉，我是 javascript 和 node-postgres 的新手。

Answer 1

无论您在一个中返回什么，然后都会在下面结束。 例如：

client.query("SELECT...")
    .then((result) => {
        // do something with the result
        return 42;
    })
    .then((result) => console.log(result))
    .then((result) => console.log(result))

第一个日志打印 42 ，因为前一个then 42 ，第二个打印undefined因为前一个then nothing = undefined 。 所以你实际上想要这样的东西：

client.query('SELECT "ID" FROM "User" WHERE "PW" = $1 AND "Name" = $2', [req.body.pw, req.body.name])
    .then((res) => {
        const userid = res.rows[0];
        return client.query('SELECT " FROM "List" WHERE "UserID" = $1', [userid]).then(() => {
            return client.query('INSERT INTO "List" ("UserID", "LastDate")', [userid, req.body.date]);
        });
    })
    .then(() => {
        return client.query('UPDATE "List" SET "LastDate" = $1', [req.body.date])
    })
    .catch(error => console.error(error))
    .finally(() => client.end());

这是您甚至不需要半全局变量userid和listEntry 。 这通常是一件好事。

使用async/await时，代码可以更加简化。

如果您不想执行以下查询，因为第一个响应中没有行，您可以执行以下操作：

client.query('SELECT "ID" FROM "User" WHERE "PW" = $1 AND "Name" = $2', [req.body.pw, req.body.name])
    .then((res) => {
        if (res.rows.length > 0) {
            const userid = res.rows[0];
            return client.query('SELECT " FROM "List" WHERE "UserID" = $1', [userid]).then(() => {
                return client.query('INSERT INTO "List" ("UserID", "LastDate")', [userid, req.body.date]);
            }).then(() => {
                return client.query('UPDATE "List" SET "LastDate" = $1', [req.body.date])
            })
        }
    })
    .catch(error => console.error(error))
    .finally(() => client.end());