Nodejs 一个怪异的问题

Node.js 一个怪异的问题

最近我在开发一个 Node.js 应用程序时遇到了一个非常奇怪的问题。具体来说，在处理一些特定的请求时，某些函数的行为与预期不符。经过一番排查，我发现问题出在异步操作上。

问题描述

假设我有一个简单的 API 端点 /get-data，它应该从数据库中获取数据并返回给客户端。代码大致如下：

const express = require('express');
const app = express();
const port = 3000;

app.get('/get-data', async (req, res) => {
    const data = await fetchDataFromDatabase();
    console.log('Fetched data:', data);
    res.json(data);
});

async function fetchDataFromDatabase() {
    return new Promise((resolve, reject) => {
        setTimeout(() => {
            resolve({ id: 1, name: 'Test Data' });
        }, 1000);
    });
}

app.listen(port, () => {
    console.log(`Server running on http://localhost:${port}`);
});

问题现象

当我访问 /get-data 端点时，控制台会输出类似以下信息：

Server running on http://localhost:3000
Fetched data: undefined

然而，返回给客户端的数据却是正确的：

{
    "id": 1,
    "name": "Test Data"
}

排查过程

起初我以为是 fetchDataFromDatabase 函数出了问题，但通过调试发现该函数确实返回了正确的数据。后来我注意到 console.log('Fetched data:', data); 这一行执行的时间比预期要早，这让我怀疑是不是异步操作的问题。

解决方案

最终，我发现问题在于 setTimeout 的使用方式。虽然 setTimeout 是异步的，但它并不会阻塞事件循环。因此，当 await 操作完成后，代码会继续执行，而不会等待 setTimeout 完成。为了解决这个问题，我们可以使用 Promise 和 async/await 来确保异步操作按顺序执行。

修改后的代码如下：

const express = require('express');
const app = express();
const port = 3000;

app.get('/get-data', async (req, res) => {
    try {
        const data = await fetchDataFromDatabase();
        console.log('Fetched data:', data);
        res.json(data);
    } catch (error) {
        console.error('Error fetching data:', error);
        res.status(500).send('Internal Server Error');
    }
});

async function fetchDataFromDatabase() {
    return new Promise((resolve, reject) => {
        setTimeout(() => {
            resolve({ id: 1, name: 'Test Data' });
        }, 1000);
    });
}

app.listen(port, () => {
    console.log(`Server running on http://localhost:${port}`);
});

通过上述修改，我们确保了 console.log 在 fetchDataFromDatabase 完成后才执行，从而解决了问题。

希望这个例子能帮助遇到类似问题的朋友！