javascript - 如何等待 node.js 中的异步循环

Question

我正在使用 Node.js 和 socket.io 制作一个私有(private)消息系统。我需要在 2 个用户之间发送最新消息。 get_messages_between_users 函数和 get_message_user 函数分别工作正常。但是如果我像下面那样使用它们，messages 数组在套接字发出数据时保持为空。由于Javascript的异步结构，这实际上是正常的，但是我该如何解决这个问题呢？这是代码。

function get_message_user(message, callback) {
    var sql = "SELECT * from users WHERE id = " + message.from_user
    var query = dbh.query(sql, function(err, result) {
        callback(message, result[0]);
    });
};

function get_messages_between_users(user1, user2, callback) {
    var sql = "SELECT id, from_user, to_user, content, msg_date, status FROM messages WHERE ((to_user = "+user1+" AND from_user = "+user2+") OR (to_user = "+user2+" AND from_user = "+user1+")) ORDER BY id ASC LIMIT 20;"
    var query = dbh.query(sql, function(err, result) {
        callback(result);
    });
};

io.on('connection',function (socket) {
    //Other stuff for handling clients
    socket.on("page messages", function (data) {
        var messages = [];
        get_messages_between_users(socket.session.user_id, data.to_user, function(result){
            for(var i = 0; i < result.length; i++){
                get_message_user(result[i], function(message, user){
                    var obj = {id: message.id, user_id: user.id, username: user.username, avatar:user.avatar_url, content: message.content, msg_date:message.msg_date};
                    messages.push(obj);
                });
            }
        });

        //The problem is messages array is empty here.
        socket.emit('message history', JSON.stringify(messages));
    });
});

我在 get_message_user 中使用 message 参数只是为了绕过结果变量。

Answer 1

为了重构代码以处理所有这些相互依赖的异步操作，您需要做几件事:

1) 将 get_message_user 和 get_messages_between_users 函数转换为基于 Promise 的函数。

(注意:如以下评论中所建议，最佳解决方案是使用当前数据库库支持的基于 promise 的函数，或者切换到使用 Promise 的函数操作。或者，可以使用一个实用程序轻松地将回调函数转换为基于 Promise 的函数。一个这样的实用程序是 Promisify

2) 将socket.on("page message")回调转换为异步函数

3) 在套接字回调中，调用 get_messages_between_users 函数并为异步结果调用 await

4) 返回结果列表后，遍历它并创建一个调用 get_message_user 异步函数的数组，并为所有这些异步请求的结果创建 await使用 Promise.all() 函数。

5) 返回结果后，映射(或循环)它们，并将数据添加到消息数组并通过 socket.emit 函数发出数据

function get_message_user(message) {
    // Return a Promise that will be handled via a then() method
    return new Promise((resolve,reject)=>{
        let sql = "SELECT * from users WHERE id = " + message.from_user;
        dbh.query(sql, function(err, result) {
            if ( err ){ reject(err); } // Handle Error via a catch() method
            resolve([message,result[0]]);
        });
    })
};

function get_messages_between_users(user1, user2) {
    return new Promise((resolve,reject)=>{
        let sql = "SELECT id, from_user, to_user, content, msg_date, status FROM messages WHERE ((to_user = "+user1+" AND from_user = "+user2+") OR (to_user = "+user2+" AND from_user = "+user1+")) ORDER BY id ASC LIMIT 20;"
        dbh.query(sql, function(err, result) {
            if ( err ){ reject(err); }
            resolve(result);
        });
    })
};

io.on('connection',function (socket) {

    // We need an async function here, in order to use await on async operations
    socket.on("page messages", async function (data) {

        let messages = [];
        // Wait for the asynchronous operation to give back a result using await
        const result = await get_messages_between_users(socket.session.user_id, data.to_user);
        const requests = []; // This will hold a list of async requests for each result
        for ( let i = 0; i < result.length; i++){
            requests.push( get_message_user(result[i]) );
        }
        const results = await Promise.all(requests);
        results.map((result)=>{
            let message = result[0];
            let user = result[1];
            let obj = {
              id: message.id,
              user_id: user.id,
              username: user.username,
              avatar:user.avatar_url,
              content: message.content,
              msg_date:message.msg_date
            };
            messages.push(obj);
        });

        socket.emit('message history', JSON.stringify(messages));

    });

});

引用资料:

• > Promises
• > Promise.all()
• > Async/Await

注意:无需为 Udemy 类(class)付费。你可以在 Udacity 查看这些关于异步编程的优秀类(class):

• > Asynchronous JavaScript Requests
• > JavaScript Promises

警告:正如上面的评论所正确指出的那样，处理问题中发布的 SQL 查询的代码需要进行大量重构才能使代码安全。此答案中的代码仅用于演示如何处理多个异步请求。其余代码仍然不安全并且必须重构。