将 Async/Await 与 node-postgres 一起使用

Using Async/Await with node-postgres

我正在使用 node-postgres 查询我的数据库,想知道如何使用 async/await 并正确处理错误

这里是我使用的一个非常简单的查询示例

const { Pool } = require('pg');

let config;
if (process.env.NODE_ENV === 'production' || process.env.NODE_ENV === 'staging') {
  config = { connectionString: process.env.DATABASE_URL, ssl: true };
} else {
  config = {
    host: 'localhost',
    user: 'myuser',
    database: 'mydatabase',
  };
}

const pool = new Pool(config);

async function getAllUsers() {
  let response;
  try {
    response = await pool.query('select * FROM users');
  } catch (error) {
    throw error;
  }
  return response.rows;
}

然后在我的 routes.js 我有

app.get('/all_users', async (req, res) => {
  const users = await queries.getAllUsers();
  console.log(users); // returns all users fine
});

到目前为止,这是我的理解,但我认为我没有正确处理这个问题,因为当涉及到错误时,我的应用程序会冻结并抛出 UnhandledPromiseRejectionWarning。因此,如果我提供了不正确的 table,例如

async function getAllUsers() {
  let response;
  try {
    response = await pool.query('select * FROM notable');
  } catch (error) {
    throw error;
  }
  return response.rows;
}

UnhandledPromiseRejectionWarning: error: relation "notable" does not exist

应用程序会在 30 秒后崩溃,我没有妥善处理这个错误

有人能指出我在这里遗漏了什么吗?

async 函数或 Promise 抛出未捕获的错误时,或者当 catcher 也 抛出时,例如您的

throw error;

这意味着函数的调用者将面临一个被拒绝的Promise来处理。如果您在调用方中使用 await,那么您还必须在调用方中使用 try/catch 才能正确捕获错误:

app.get('/all_users', async (req, res) => {
  try {
    const users = await queries.getAllUsers();
    console.log(users);
  } catch(e) {
    // handle errors
  }
});

无需在消费者中使用 try/catch 即可解决错误的另一种方法是不 throw 您的 catch 中的错误:

async function getAllUsers() {
  let response;
  try {
    response = await pool.query('select * FROM users');
    return response.rows;
  } catch (error) {
    // handle error
    // do not throw anything
  }
}

但这会使消费者更难知道何时出现错误。

在这种特殊情况下,async/await/try/catch 结构增加了很多语法噪音,但对 IMO 没有太大好处 - 目前,您可能会考虑改用普通的 Promises:

const getAllUsers = () => pool.query('select * FROM users')
  .then(response => response.rows);

// and:
app.get('/all_users', (req, res) => {
  queries.getAllUsers()
    .then((users) => {
      console.log(users);
    })
    .catch((err) => {
      // handle errors
    });
});

asyncawait 当您有一些 .then 想让您的代码看起来更平坦时,它会发光。如果只有一个 .then,IMO 将其转换为 async/await 语法没有太大好处。当然,这取决于你。