在递归函数中保持对象引用

时间:2015-07-22 05:28:14

标签: javascript recursion tree lodash

我有一个应用程序,其中一个对象用于显示用户系统上文件的树视图。它的结构如下:

[{
   text: 'C:/',
   type: 'dir',
   nodes: [
      {
         text: 'foo',
         type: 'dir',
         nodes: [] // And so on
      },
      {
         text: 'bar',
         type: 'file'
      }
}]

为了与惯例保持一致,我希望首先显示目录,然后显示要显示的文件。不幸的是,无论项目类型如何,我的数据都按字母顺序检索。

为了解决这个问题,我写了一个很好的递归函数

var sort = function (subtree)
{
   subtree = _.sortBy(subtree, function (item)
   {
      if (item.nodes)
      {
         sort(item.nodes)
      }
      return item.type
   });
}

var tree = someTreeData;
sort(tree);

我使用lodash按文件类型按字母顺序对每个nodes数组进行排序。不幸的是,子树似乎没有引用树对象,因为当我记录它的输出时它仍未被排序。我该如何解决这个问题?

1 个答案:

答案 0 :(得分:2)

您可以使用JavaScript的内置Array.prototype.sort函数,该函数可就地排序。它接受两个参数并执行比较。请注意,在item.notes密钥提取程序中对sortBy进行排序是不合适的。

function isDirectory(node) {
    return !!node.nodes;
}

function sortTree(subtree) {
    subtree.sort(function (a, b) {
        return a.type < b.type ? -1 :
               a.type > b.type ? 1 : 0;
    });

    subtree
        .filter(isDirectory)
        .forEach(function (node) {
            sortTree(node.nodes);
        });
}