如何递归和异步构建未知大小的树？

Question

我正在尝试在我的网站的给定帖子中检索我的 cmets，但由于 node.js 的异步性质，我无法构建嵌套 cmets。

getBlock([], function(){});    

function getBlock(comments, callback) {
    comments.forEach(function(comment) {
        getChildComments(comment, function(err, children) {
            if (children) {
                getBlock(children, callback);
            }
            comment.Comments = children;

            /* not sure how to decide when to be done?*/
            callback(null, comments);
        });
    });
}

上面的代码适用于同步代码，但不适用于异步代码，因为我无法判断comments 何时包含要返回到浏览器的所有数据。

我试图跟踪递归调用，并在剩下 0 个调用时结束，但这有问题，有时会根据树结构提前返回。

Answer 1

你可以保留一个计数仍然要做的工作，当它达到零时，你调用调用者的回调函数。递归树中执行函数的每个实例都会定义自己的回调，这样只有顶层实例的调用才会调用第一条语句中的回调（函数体之外）：

function getBlock(comments, callback) {
    if (!comments || !comments.length) {
        // Nothing to do, call back synchronously
        callback(comments);
        return;
    }
    var leftOver = comments.length;
    comments.forEach(function(comment) {
        getChildComments(comment, function(err, children) {
            comment.Comments = children;
            // provide custom callback:
            getBlock(children, function () {
                // only call parent's callback when all is done here:
                if (--leftOver === 0) callback(comments);
            });
        });
    });
}

与您的示例代码不同，上述代码不能使用空数组调用，而应使用要检索其下方层次结构的注释对象数组。要获取所有内容，您将传递一个包含一个虚拟评论对象的数组，该对象将具有未定义的id（与没有父级的 cmets 的 parentId 引用匹配）。像这样的：

getBlock([container], function(){
    console.log(container);
});

下面是一个工作实现，它使用模拟数据和setTimeout来模拟异步getChildComments：

function Comment(id, text, parentId) {
    this.id = id;
    this.text = text;
    this.parentId = parentId;
}

var mockData = [
    new Comment(1, "Michal Jackson died today"),
    new Comment(2, "How did he die?", 1),
    new Comment(3, "His doctor gave him too much of the white stuff", 2),
    new Comment(4, "He died in his sleep", 2),
    new Comment(5, "Oh my god, this can't be true!?", 1),
    new Comment(6, "He will be greatly missed", 1),
    new Comment(7, "I am working in my garden"),
    new Comment(8, "Happy birthday, friend!"),
    new Comment(9, "Thank you!", 8),   
];

function getChildComments(parentComment, callback) {
    // Mock asynchronous implementation, for testing the rest of the code
    setTimeout(function () {
        var children = mockData.filter(function (comment) {
            return comment.parentId === parentComment.id;
        });
        callback(null, children); 
    }, 0);
}

var container = new Comment(); // dummy node to collect complete hierarchy into
getBlock([container], function(){
    console.log(container);
});

function getBlock(comments, callback) {
    if (!comments || !comments.length) {
        // Nothing to do, call back synchronously
        callback(comments);
        return;
    }
    var leftOver = comments.length;
    comments.forEach(function(comment) {
        getChildComments(comment, function(err, children) {
            comment.Comments = children;
            // provide custom callback:
            getBlock(children, function () {
                // only call parent's callback when all is done here:
                if (--leftOver === 0) callback(comments);
            });
        });
    });
}

性能考虑

以上是对“如何递归异步构建未知大小的树”的直接回答，但这可能不是获得最终结果的最有效方法。

您从 Postgres 数据库中获取数据，并且可能为每次调用 getChildComments 执行查询：这可能需要相对较长的时间才能完成，并且会给您的数据库引擎带来相当大的负载。

执行单个查询来检索整个 cmets 层次结构可能更有效。