Question

是否可以使用Elixir构建Python样式的递归生成器？像这样：

def traverse(parent_dir):
    dirs, files = get_dirs_and_files_as_lists(parent_dir)

    for d in dirs:
        yield from traverse(d)

    for f in files:
        yield f

对于所有要线性处理的文件，没有急切的不定长度列表所隐含的开销：

for f in traverse(dir):
    process(f)

使用流可以做到这一点或等效的工作；不幸的是，我不知道如何。

我想要这样的东西，只是懒惰：

def traverse_eagerly(parent_dir) do
  {dirs, files} = get_dirs_and_files_as_lists(parent_dir)

  for x <- dirs do
    traverse_eagerly(x)
  end
  |> Enum.concat()
  |> Enum.concat(files)
end

Answer 1

您在这里不需要Stream，但如果需要，这里是：

defmodule Traverse do
  @spec traverse(root :: binary(), yielder :: (binary() -> any())) ::
     :ok | {:error, posix()}
  def traverse(root, yielder) do
    # https://hexdocs.pm/elixir/master/File.html?#ls/1
    with {:ok, list} <- File.ls(root) do
      list
      |> Stream.each(fn file_or_dir ->
        if File.dir?(file_or_dir),
          do: traverse(file_or_dir, yielder), # TCO
          else: yielder.(file_or_dir)
      end)
      |> Stream.run()
    end
  end
end

并这样称呼：

Traverse.traverse(".", &IO.inspect/1)

Answer 2

解决方案似乎很简单：将Enum替换为Stream。

def traverse_lazily(parent_dir) do
  {dirs, files} = get_dirs_and_files_as_lists(parent_dir)

  for x <- dirs do
    traverse_lazily(x)
  end
  |> Stream.concat()
  |> Stream.concat(files)
end

以下按预期工作：

s = traverse_lazily(a_dir_of_choice)

for x <- s, do: whatever_you_please(x)

语言非常好。完美的解决方案，如您所愿。除非我缺少任何东西，否则就是:)。欢迎发表评论！

Elixir：递归生成器

2 个答案: