126 lines
3.2 KiB
Lua
126 lines
3.2 KiB
Lua
function printrunning()
|
|
if coroutine.running() == nil then
|
|
print("running is nil");
|
|
else
|
|
print("running is not nil")
|
|
end
|
|
end
|
|
|
|
function foo (a)
|
|
print("foo", a)
|
|
return coroutine.yield(2*a)
|
|
end
|
|
|
|
co = coroutine.create(function (a,b)
|
|
print("co-body", a, b)
|
|
local r = foo(a+1)
|
|
print("co-body", r)
|
|
local r, s = coroutine.yield(a+b, a-b)
|
|
print("co-body", r, s)
|
|
|
|
printrunning()
|
|
print("co.status.inside",coroutine.status(co));
|
|
local co2 = coroutine.create(function()
|
|
print("co.status.inside2",coroutine.status(co));
|
|
end)
|
|
print("co.status.inside",coroutine.status(co));
|
|
coroutine.resume(co2);
|
|
|
|
return b, "end"
|
|
end)
|
|
|
|
function exercise()
|
|
printrunning()
|
|
print("co.status",coroutine.status(co));
|
|
print("main", coroutine.resume(co, 1, 10))
|
|
print("co.status",coroutine.status(co));
|
|
print("main", coroutine.resume(co, "r"))
|
|
print("co.status",coroutine.status(co));
|
|
print("main", coroutine.resume(co, "x", "y"))
|
|
print("co.status",coroutine.status(co));
|
|
print("main", coroutine.resume(co, "x", "y"))
|
|
print("co.status",coroutine.status(co));
|
|
end
|
|
|
|
exercise();
|
|
|
|
co = coroutine.create(function (a,b)
|
|
print("co-body", a, b)
|
|
-- TODO: make java and C behave the same for yielding in pcalls
|
|
-- local status,r = pcall( foo, a+1 )
|
|
foo(a+1)
|
|
print("co-body", status,r)
|
|
local r, s = coroutine.yield(a+b, a-b)
|
|
print("co-body", r, s)
|
|
return b, "end"
|
|
end)
|
|
|
|
exercise();
|
|
|
|
|
|
-- wrap test
|
|
local g = coroutine.wrap(function (a,b)
|
|
print("co-body", a, b)
|
|
local r = foo(a+1)
|
|
print("co-body", r)
|
|
local r, s = coroutine.yield(a+b, a-b)
|
|
print("co-body", r, s)
|
|
return b, "end"
|
|
end )
|
|
|
|
print("g", g(1, 10))
|
|
print("g", g("r"))
|
|
print("g", g("x", "y"))
|
|
local s,e = pcall( g, "x", "y" )
|
|
print("g", string.match(e,'cannot resume dead coroutine') or 'badmessage: '..tostring(e))
|
|
|
|
-- varargs passing
|
|
local echo = function(msg,...)
|
|
print( msg, ...)
|
|
return ...
|
|
end
|
|
local echocr = function(...)
|
|
echo('(echocr) first args', unpack(arg,1,arg.n))
|
|
local a = arg
|
|
while true do
|
|
a = { echo( '(echoch) yield returns', coroutine.yield( unpack(a) ) ) }
|
|
end
|
|
end
|
|
local c = coroutine.create( echocr )
|
|
local step = function(...)
|
|
echo( '(main) resume returns',
|
|
coroutine.resume(c, echo('(main) sending args', ...)) )
|
|
end
|
|
step(111,222,333)
|
|
step()
|
|
step(111)
|
|
step(111,222,333)
|
|
|
|
-- test loops in resume calls
|
|
b = coroutine.create( function( arg )
|
|
while ( true ) do
|
|
print( ' b-resumed', arg, b == coroutine.running() )
|
|
print( ' b-b', coroutine.status(b) )
|
|
print( ' b-c', coroutine.status(c) )
|
|
print( ' b-resume-b',coroutine.resume( b, 'b-arg-for-b' ) )
|
|
print( ' b-resume-c',coroutine.resume( c, 'b-arg-for-c' ) )
|
|
arg = coroutine.yield( 'b-rslt' )
|
|
end
|
|
end )
|
|
c = coroutine.create( function( arg )
|
|
for i=1,3 do
|
|
print( ' c-resumed', arg, c == coroutine.running() )
|
|
print( ' c-b', coroutine.status(b) )
|
|
print( ' c-c', coroutine.status(c) )
|
|
print( ' c-resume-b',coroutine.resume( b, 'b-arg-for-b' ) )
|
|
print( ' c-resume-c',coroutine.resume( c, 'b-arg-for-c' ) )
|
|
arg = coroutine.yield( 'c-rslt' )
|
|
end
|
|
end )
|
|
for i=1,3 do
|
|
print( 'main-b', coroutine.status(b) )
|
|
print( 'main-c', coroutine.status(c) )
|
|
print( 'main-resume-b',coroutine.resume( b, 'main-arg-for-b' ) )
|
|
print( 'main-resume-c',coroutine.resume( c, 'main-arg-for-c' ) )
|
|
end
|