云风的个人空间 : LuaCothread

首页 :: 索引 :: 修订历史 :: 你好, 54.144.81.21
你的足迹: » LuaCothread
这是 lua cothread 库 ,说明见这里:[InterWiki]lua cothread

-- cothread.lua
 
local co = coroutine
local error = error
local setmetatable = setmetatable
local unpack = unpack
local assert = assert
local type = type
local pairs = pairs
 
module "cothread"
 
resume = error
run = error
chan = error
sleep = error
select = error
 
local queue = error
local _active 
local timer_add = error
local timer_pop = error
local _thread_count = 0
 
do
	local _queue = {}
	local _queue_meta = { __index = _queue }
 
	function _queue:push(v)
		self[self.tail]=v
		self.tail = self.tail + 1
	end
 
	function _queue:pop()
		if self.tail == self.head then
			self.tail = 1
			self.head = 1
		else
			local ret = self[self.head]
			self[self.head] = nil
			self.head = self.head + 1
			return ret
		end
	end
 
	function _queue:move(q)
		for v in q.pop , q do
			self:push(v)
		end
	end
 
	function queue()
		return setmetatable( {head = 1, tail = 1} , _queue_meta )
	end
end
 
do
	_active = queue()
 
	local function _resume_active()
		for v in _active.pop , _active do
			assert(co.resume(v))
		end
	end
 
	function resume(ti)
		_resume_active()
		_active:move(timer_pop())
		return _thread_count
	end
end
 
function run(f,...)
	local c = co.create(
		function() 
			_thread_count = _thread_count + 1
			f(unpack(arg)) 
			_thread_count = _thread_count - 1
		end)
	_active:push(c)
end
 
function sleep(ti)
	if ti and ti > 0 then
		timer_add(co.running(), ti-1)
	else
		_active:push(co.running())
	end
	co.yield()
end
 
do 
	local _chan = {}
	local _chan_meta = { __index = _chan }
 
	function chan()
		local ret =  setmetatable(
			{ 
				_status = "empty",
				_read = queue(),
				_write = queue(),
				_closing = false,
				_value = nil
			} , _chan_meta)
		return ret
	end
 
	function _chan:read()
		if self._status == "empty" then
			if self._closing then
				return nil,true
			end
			self._read:push {co.running()}
			co.yield()
		end
		if self._status == "ready" then
			local ret = self._value
			self._value = nil
			self._status = self._closing and "closed" or "empty"
			local w = self._write:pop()
			if w then
				_active:push(w)
			end
			return ret
		end
		return nil, true
	end
 
	function _chan:write(v)
		if self._status == "ready" then
			if self._closing then
				return
			end
			self._write:push(co.running())
			co.yield()
		end
		if self._status == "empty" then
			self._value = v
			self._status = self._closing and "closed" or "ready"
			while true do
				local r = self._read:pop()
				if r then
					if r[1] then
						_active:push(r[1])
						r[1] = nil
						return true
					end
				else 
					return true
				end
			end
		end
	end
 
	function _chan:closed()
		return self._status == "closed"
	end
 
	function _chan:close()
		self._closing = true
	end
 
end
 
function select(t)
	if t.default then
		for ch,func in pairs(t) do
			if type(ch) == "table" then
				if ch._status == "ready" then
					local v = ch._value
					ch._value = nil
					ch._status = ch._closing and "closed" or "empty"
					local w = ch._write:pop()
					if w then
						_active:push(w)
					end
					func(v)
					return
				end
			end
		end
		t.default()
		_active:push(co.running())
	else
		local reader = { co.running() }
		for ch,func in pairs(t) do
			if ch._status == "ready" then
				reader[1] = nil
				local v = ch._value
				ch._value = nil
				ch._status = ch._closing and "closed" or "empty"
				local w = ch._write:pop()
				if w then
					_active:push(w)
				end
				func(v)
				return
			elseif ch._status == "empty" then
				if ch._closing then
					reader[1] = nil
					func(nil,true)
				else
					ch._read:push(reader)
				end
			end
		end
	end
	co.yield()
end
 
do
	local _timer = {}
	local _current = 1
	for i = 1,100 do
		_timer[i] = queue()
	end
 
	local _long_timer = queue()
 
	function timer_add(v , ti)
		if ti < 100 then
			local idx = _current + ti
			_timer[idx<=100 and idx or idx-100 ]:push(v)
		else
			_long_timer:push(ti + _current)
			_long_timer:push(v)
		end
	end
 
	function timer_pop()
		local ret = _timer[_current]
		_current = _current + 1
		if _current > 100 then
			_current = 1
			while true do
				local ti = _long_timer:pop()
				if ti == nil then
					break
				end
				local v = _long_timer:pop()
				timer_add(v, ti - 101)
			end
		end
 
		return ret
	end
end


这里有个简单的例子:

require "cothread"
 
function f(a,b,c)
	print(a)
	cothread.sleep()
	print(b)
	cothread.sleep()
	print(c)
end
 
cothread.run(f,1,2,3)
cothread.run(f,4,5,6)
 
cothread.resume()


输出结果为

1
4
2
5
3
6


看一个时间控制的例子:

require "cothread"
 
function f(ti)
	for i=1,3 do
		cothread.sleep(ti)
		print(ti,i)
	end
end
 
function main()
	cothread.run(f,2)
	cothread.run(f,3)
end
 
cothread.run(main)
 
ti = 0
 
while cothread.resume() > 0 do
	ti = ti + 1
	print("time : ",ti)
end


输出为:
time :  1
time :  2
2       1
time :  3
3       1
time :  4
2       2
time :  5
time :  6
3       2
2       3
time :  7
time :  8
time :  9
3       3


接下来是一个简单的使用 chan 通讯的例子:

require "cothread"
 
function input(c)
	for i = 1,5 do
		c:write(i)
	end
	c:close()
end
 
function output(c)
	repeat
		local v = c:read()
		if v then
			print(v)
		end
	until v == nil
end
 
function main()
	c = cothread.chan()
	cothread.run(input, c)
	cothread.run(output, c)
end
 
cothread.run(main)
cothread.resume()


输出结果为

1
2
3
4
5


接下来是一个使用 cothread.select 的稍微复杂的例子:

require "cothread"
 
function input(c,from)
	for i = from,from + 5 do
		c:write(i)
	end
	c:close()
end
 
function output(c1,c2)
	repeat
		cothread.select {
			[c1] = function(v)
				print("c1",v)
			end ,
			[c2] = function (v)
				print("c2",v)
			end 
		}
	until c1:closed() and c2:closed()
end
 
function main()
	c = cothread.chan()
	c2 = cothread.chan()
	cothread.run(input, c,1)
	cothread.run(input, c2,10)
	cothread.run(output, c,c2)
end
 
cothread.run(main)
cothread.resume()


输出为:
c1      1
c2      10
c1      2
c2      11
c1      3
c2      12
c1      4
c2      13
c1      5
c2      14
c1      6
c2      15