时间戳

首页 > 解决方案 > javascript - 如何在 node.js 中等待异步循环

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

问题描述

我正在使用 Node.js 和 socket.io 制作一个私人消息传递系统。我需要在 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));
    });
});

我使用message参数get_message_user只是为了绕过变量来产生结果。

标签： javascriptnode.js

解决方案

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

1) 将get_message_userandget_messages_between_users函数转换为基于 Promise 的函数。

（注意：正如下面评论中所建议的那样，最佳解决方案是使用当前数据库库支持的基于 Promise 的函数，或者切换到使用 Promises 进行这些操作的函数。或者，可以使用一个实用程序来轻松地打开回调将函数转换为基于 Promise 的函数。Promisify就是这样一种实用程序

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

3）在套接字回调中，调用get_messages_between_users函数并await获取异步结果

get_message_user4) 返回结果列表后，循环遍历它并创建一个对async 函数的调用数组，以及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));

    });

});

参考：

注意：无需支付 Udemy 课程费用。你可以在 Udacity 中查看这些关于异步编程的优秀课程：

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

推荐阅读