[将回调函数的结果推入数组时如何等待迭代完成

Question

实现array.push的正确方法是什么，以便在forEach迭代完成后返回“ array_of_results”？

const postgres = require("./postgres");

function get_array(value) {
    var array_of_results = []
    value.forEach( item => {
        postgres.query(item["id"],function(res){ 
            console.log(res) //gives proper res after empty array
            array_of_results.push(res);
        })
    });
    console.log(array_of_results)// prints empty array
    return array_of_results;
}

编辑：和postgres.js看起来像：

const { Pool } = require("pg");
const pool = new Pool();

var query_string = "select...."

function query(id, call) {
    pool.query(query_string, [id], (err, res) => {
        if (err) {
            console.log(err.stack)
        } else {
            call(res.rows[0])
        }
    })
}

module.exports = {
    query
}

Answer 1

有几种方法可以做到这一点，但首先您需要了解实际发生的情况。

在postgres.query(item["id"],function(res){中，您正在通过以下方式调用postgres.query：（1）项目ID和（2）回调函数。发生该调用，然后立即在您的调用代码中继续。因此，现在您刚刚向数据库发送了一堆请求，然后立即返回一个空数组。这些回调（2）尚未被调用。

要使数据返回到调用函数，您需要传递回调而不是使用return，或更改为异步/等待。

在循环的每个迭代中使用async / await效率不高，因为您正在等待每个调用顺序返回。对于最有效的方法，您将需要触发请求并等待所有请求完成。您可以通过使用promises来做到这一点。

您可以修改代码，以便在每次循环迭代时将一个promise放入数组，然后在promise数组上调用（并等待）Promise.all。

这是您的基本重写：

postgres.js：

function query(id) {
    return new Promise((resolve, reject) => {
        pool.query(query_string, [id], (err, res) => {
            if (err) {
                console.log(err.stack)
                reject(err)
            } else {
                resolve(res.rows[0])
            }
        })
    })
}

module.exports = {
    query
}

get_array实现：

async function get_array(value) {
    var array_of_promises = [], array_of_results = []
    value.forEach( item => {
        array_of_promises.push(postgres.query(item["id"]));
    });
    array_of_results = await Promise.all(array_of_promises);
    console.log(array_of_results)// prints populated array
    return array_of_results;
}

请注意，当您呼叫get_array时，必须在呼叫前使用await，例如将var array = get_array(items)更改为var array = await get_array(items)并在函数中使用await要求将其声明为异步函数。

如果无法将其声明为异步函数，则可以更改调用代码以使用Promise：

var arrayPromise = get_array(items);
arrayPromise.then((results) => {
    // do something with results
    // but remember you cannot _return_ from within a callback, as discussed above
});