122 lines
3.3 KiB
Plaintext
122 lines
3.3 KiB
Plaintext
# coroutine.nas by ValKmjolnir
|
|
# 2022/5/19
|
|
use std.process_bar;
|
|
use std.padding;
|
|
|
|
if(os.platform()=="windows"){
|
|
system("chcp 65001");
|
|
system("color");
|
|
}
|
|
|
|
var fib=func(){
|
|
var (a,b)=(1,1);
|
|
coroutine.yield(a);
|
|
coroutine.yield(b);
|
|
while(1){
|
|
(a,b)=(b,a+b);
|
|
coroutine.yield(b);
|
|
}
|
|
return;
|
|
}
|
|
var co=[coroutine.create(fib),coroutine.create(fib)];
|
|
for(var i=0;i<45;i+=1){
|
|
var res=[coroutine.resume(co[0]),coroutine.resume(co[1])];
|
|
if(res[0]==nil or res[1]==nil or res[0][0]!=res[1][0])
|
|
die("different coroutines don't share the same local scope");
|
|
}
|
|
|
|
# test if coroutine can get upvalues
|
|
func(){
|
|
var x=1;
|
|
var co=coroutine.create(func(){
|
|
for(var j=0;j<128;j+=1){
|
|
coroutine.yield(x,i,j);
|
|
x+=1;
|
|
}
|
|
});
|
|
for(var i=0;i<16;i+=1){
|
|
var res=coroutine.resume(co);
|
|
if(res==nil or res[0]!=x or res[1]!=i)
|
|
die("coroutine should have the ability to get upvalues");
|
|
}
|
|
}();
|
|
|
|
# test coroutine.resume passing arguments to coroutine
|
|
func{
|
|
var co=coroutine.create(func(){
|
|
var (a,b)=coroutine.yield(a+b);
|
|
println("coroutine.yield get ",a," ",b);
|
|
(a,b)=coroutine.yield(a+b);
|
|
println("coroutine.yield get ",a," ",b);
|
|
return "end";
|
|
});
|
|
|
|
for(var i=0;i<5;i+=1)
|
|
println("coroutine.resume get ",coroutine.resume(co,i,i+1));
|
|
}();
|
|
|
|
# test crash in coroutines
|
|
var co=coroutine.create(func{
|
|
var b=func(){b()}
|
|
coroutine.yield(b);
|
|
b();
|
|
coroutine.yield(0);
|
|
});
|
|
|
|
println("coroutine yield: ",coroutine.resume(co));
|
|
println("coroutine state:\e[32m ",coroutine.status(co),"\e[0m");
|
|
println("coroutine error: ",coroutine.resume(co));
|
|
println("coroutine state:\e[91m ",coroutine.status(co),"\e[0m");
|
|
println("coroutine yield: ",coroutine.resume(co));
|
|
println("coroutine state:\e[91m ",coroutine.status(co),"\e[0m");
|
|
|
|
var co=coroutine.create(func{
|
|
var a=1;
|
|
var b=func(){
|
|
b();
|
|
}
|
|
coroutine.yield(b);
|
|
coroutine.yield(b());
|
|
});
|
|
|
|
println("coroutine yield: ",coroutine.resume(co));
|
|
println("coroutine state:\e[32m ",coroutine.status(co),"\e[0m");
|
|
println("coroutine error: ",coroutine.resume(co));
|
|
println("coroutine state:\e[91m ",coroutine.status(co),"\e[0m");
|
|
println("coroutine yield: ",coroutine.resume(co));
|
|
println("coroutine state:\e[91m ",coroutine.status(co),"\e[0m");
|
|
println("ok");
|
|
|
|
# pressure test
|
|
for(var t=0;t<10;t+=1){
|
|
var productor=func(){
|
|
while(1){
|
|
coroutine.yield(i);
|
|
}
|
|
}
|
|
|
|
var co=coroutine.create(productor);
|
|
var tm=maketimestamp();
|
|
|
|
var counter=0;
|
|
var bar=process_bar.high_resolution_bar(40);
|
|
var consumer=func(){
|
|
counter+=1;
|
|
for(var i=0;i<t+1;i+=1)
|
|
coroutine.resume(co);
|
|
if(counter-int(counter/1000)*1000==0){
|
|
var rate=counter/2e5;
|
|
print(" ",bar.bar(rate)," ",
|
|
padding.leftpad(str(int(rate*100)),3),"% | ",
|
|
str(1e3*int(counter/tm.elapsedMSec())),
|
|
" tasks/s \r");
|
|
}
|
|
}
|
|
|
|
tm.stamp();
|
|
for(var i=0;i<1e5;i+=1)
|
|
consumer();
|
|
println(" ",bar.bar(1)," 100% | ",
|
|
str(int(1e3*counter/tm.elapsedMSec())),
|
|
" tasks/s ");
|
|
} |