利用协程实现的复制和迭代器

我正在尝试使用copas适应nrk/redis-lua。起初,对于像getset这样的命令,这对我来说似乎很容易:只需使用copas.wrap()包装客户端即可。

但实际上这并不起作用,因为在使用协程的迭代器中使用了对receive的调用。下面的示例说明了会发生什么:

local copas  = require "copas"
local socket = require "socket"

local function iterate (x)
  local client = socket.tcp ()
  client:connect("127.0.0.1", 6379)
  client:settimeout (0)
  client = copas.wrap (client)
  return coroutine.wrap (function ()
    for k, v in pairs (x) do
      client:receive "*l"
      coroutine.yield (k, v)
    end
    client:close ()
  end)
end

local data = {}
for i = 1, 2 do
  data [i] = i
end

local function worker ()
  local id = coroutine.running ()
  for k, v in iterate (data) do
    print (id, k, v)
  end
end

copas.addthread (worker)
copas.addthread (worker)

copas.loop ()

输出如下:

thread: 0x1b31990   tcp{client}: 0x1b31e48  table: 0x1b2eac0
thread: 0x1b31990   tcp{client}: 0x1b31e48  table: 0x1b2eac0
...

第一个问题:我必须设置client:settimeout (0),否则对client:receive的调用将会阻塞。但这将创建一个活跃的等待...有没有更好的方法?

第二个问题:有没有一种方法来解决迭代器内的yield问题?我可以将迭代器重写为闭包,以避免使用协程,但是有没有一种不那么侵入式的解决问题的方法?

点赞