lua - 一个默默失败的 Lua 迭代器？

Question

我对一个简单的迭代器有一个非常简单的问题。

假设我设计了一个函数，files() ，它遍历文件夹中的所有文件:

for file in files("/path/to/folder") do
  print(file)
end

现在，这看起来很完美，但这里有一个问题:如果文件夹不存在，或者我们没有读取权限怎么办？

我们将如何指出这样的错误？

一种解决方案是让 files()返回 nil, "no read permission"在这种情况下。然后我们就可以将调用封装到 files()内assert() :

for file in assert(files("/path/to/folder")) do
  print(file)
end

这似乎解决了问题。但这迫使我们的用户始终使用 assert() .如果用户不关心错误怎么办？对于这类用户，我们希望我们的 files()表现得好像文件夹是空的。但是 Lua -- 万一files()表示错误——将尝试调用返回的 nil这将导致错误(“尝试调用 nil 值”)。

所以，

我们如何设计迭代器，files() ，这会同时满足关心错误的用户和不关心错误的用户吗？

如果不可能，您会建议什么替代方案？

Answer 1

第一:而不是返回 nil + 错误消息考虑在 files 中引发错误函数(使用 error )。这样你就不会忘记 assert调用，并且您不会遇到令人困惑的“尝试调用 nil 值”错误。

您可以将额外的 bool 参数传递给 files当你不想引发错误时——你应该返回一个空函数( function() end )而不是调用 error在这种情况下。

更通用的方法如下:

-- an iterator that immediately stops a for loop
local function dummy_iter() end

-- catch errors and skip for loop in that case
function iterpcall( g, ... )
  local ok, f, st, var = pcall( g, ... )
  if ok then
    return f, st, var
  else
    return dummy_iter
  end
end


for file in iterpcall( files, "/path/to/folder" ) do
  print( file )
  for line in iterpcall( io.lines, file ) do -- works for other iterators as well
    print( line )
  end
end

iterpcall的实现以上只处理迭代器生成器( files 或 io.lines )中引发的错误，而不是迭代器函数( f )本身。你必须包装 f在带有 pcall 的封口中要做到这一点。