在Crystal中实现Python os.walk时遇递归块展开错误的问询

你遇到的问题是Crystal编译器对递归中的yield处理逻辑导致的——Crystal会尝试在编译时展开递归调用里的yield代码，但目录结构的深度是运行时才能确定的，编译器没法预判递归的终止点，所以会触发无限代码生成的警告/错误。这和Python动态处理迭代器的逻辑完全不同，得调整实现方式。

为什么你的递归写法行不通？ #

你的代码里直接在递归调用walk时让它yield，Crystal的静态编译特性会试图为每一层递归生成对应的yield代码，但因为递归深度不确定，编译器无法停止代码展开，最终触发相关提示。

两种可行的解决思路 #

我们可以通过封装迭代器或者用栈模拟递归的方式，绕过编译器对递归yield的展开逻辑：

方案1：用Enumerator封装递归遍历

把yield逻辑放到Enumerator.new的块中，递归调用时通过each传递子迭代器的内容，这样编译器只会生成一次迭代器代码，不会无限展开：

def walk(start_path : String = @root)
  Enumerator.new do |yielder|
    dir = Dir.new(start_path)
    next unless Dir.exists?(dir.path)

    # 过滤掉.和..目录
    entries = dir.entries.reject { |e| e == "." || e == ".." }
    dirs, files = entries.partition { |e| Dir.exists?(File.join(dir.path, e)) }

    # 输出当前目录信息
    yielder << {dir.path, dirs, files}

    # 递归处理子目录，将子迭代器的内容传递给当前yielder
    dirs.each do |dir_name|
      walk(File.join(dir.path, dir_name)).each do |entry|
        yielder << entry
      end
    end
  end
end

方案2：迭代式遍历（栈模拟DFS，避免递归栈溢出）

如果目录深度很大，递归可能导致栈溢出，用栈模拟深度优先遍历更稳妥，同时彻底避开递归yield的问题：

def walk(start_path : String = @root)
  Enumerator.new do |yielder|
    stack = [start_path]

    until stack.empty?
      current_path = stack.pop
      dir = Dir.new(current_path)
      next unless Dir.exists?(dir.path)

      entries = dir.entries.reject { |e| e == "." || e == ".." }
      dirs, files = entries.partition { |e| Dir.exists?(File.join(current_path, e)) }

      yielder << {current_path, dirs, files}

      # 反向入栈保证遍历顺序和递归一致
      dirs.reverse.each do |dir_name|
        stack << File.join(current_path, dir_name)
      end
    end
  end
end

使用示例 #

两种实现都可以像Python的os.walk一样直接使用：

walk("/your/target/dir") do |path, dirs, files|
  puts "当前目录：#{path}"
  puts "子目录：#{dirs.join(", ")}"
  puts "文件：#{files.join(", ")}"
  puts "---"
end

内容的提问来源于stack exchange，提问作者mlobl