improve poller

2 files changed, 3 insertions, 3 deletions

diff --git a/tests/modules/socket/sched_tcp/echo_client.lua b/tests/modules/socket/sched_tcp/echo_client.lua
index 50e8c6dcd..f9c993729 100644
--- a/tests/modules/socket/sched_tcp/echo_client.lua
+++ b/tests/modules/socket/sched_tcp/echo_client.lua
@@ -22,7 +22,7 @@ end
 function main(count)
     count = count and tonumber(count) or 1
     for i = 1, count do
-        scheduler.run(_session, "127.0.0.1", 9001)
+        scheduler.co_start(_session, "127.0.0.1", 9001)
     end
     scheduler.runloop()
 end
diff --git a/tests/modules/socket/sched_tcp/echo_server.lua b/tests/modules/socket/sched_tcp/echo_server.lua
index 4ae2119cc..18f3d990f 100644
--- a/tests/modules/socket/sched_tcp/echo_server.lua
+++ b/tests/modules/socket/sched_tcp/echo_server.lua
@@ -28,13 +28,13 @@ function _listen(addr, port)
         local sock_client = sock:accept()
         if sock_client then
             print("%s: accepted", sock_client) 
-            scheduler.run(_session, sock_client) 
+            scheduler.co_start(_session, sock_client) 
         end
     end
     sock:close()
 end
 
 function main()
-    scheduler.run(_listen, "127.0.0.1", 9001)
+    scheduler.co_start(_listen, "127.0.0.1", 9001)
     scheduler.runloop()
 end