diff options
| author | ruki <[email protected]> | 2020-01-31 00:31:03 +0800 |
|---|---|---|
| committer | ruki <[email protected]> | 2020-01-30 21:06:19 +0800 |
| commit | 4d8a3d38c1a199a2cff62b3ae725b00bfc67a296 (patch) | |
| tree | cfaf5d3f0f4dc771a747de7155554f716ff39aa2 /tests/modules/socket | |
| parent | 4e2bb608354145ac86d85b418e9a2319eaed189e (diff) | |
improve scheduler and socket connect
Diffstat (limited to 'tests/modules/socket')
| -rw-r--r-- | tests/modules/socket/sched_tcp/echo_client.lua | 13 | ||||
| -rw-r--r-- | tests/modules/socket/sched_tcp/echo_server.lua | 2 |
2 files changed, 9 insertions, 6 deletions
diff --git a/tests/modules/socket/sched_tcp/echo_client.lua b/tests/modules/socket/sched_tcp/echo_client.lua index 41490317c..257cb451f 100644 --- a/tests/modules/socket/sched_tcp/echo_client.lua +++ b/tests/modules/socket/sched_tcp/echo_client.lua @@ -37,11 +37,14 @@ function _session(addr, port) print("connect %s:%d ..", addr, port) local sock = socket.connect(addr, port) - print("%s: connected!", sock) - - scheduler.co_start(_session_recv, sock) - scheduler.co_start(_session_send, sock) - table.insert(socks, sock) + if sock then + print("%s: connected!", sock) + table.insert(socks, sock) + scheduler.co_start(_session_recv, sock) + scheduler.co_start(_session_send, sock) + else + print("connect %s:%d failed", addr, port) + end end function main(count) diff --git a/tests/modules/socket/sched_tcp/echo_server.lua b/tests/modules/socket/sched_tcp/echo_server.lua index 7f3db2d8c..f922608f6 100644 --- a/tests/modules/socket/sched_tcp/echo_server.lua +++ b/tests/modules/socket/sched_tcp/echo_server.lua @@ -42,9 +42,9 @@ function _listen(addr, port) local sock_client = sock:accept() if sock_client then print("%s: accepted", sock_client) + table.insert(sock_clients, sock_client) scheduler.co_start(_session_recv, sock_client) scheduler.co_start(_session_send, sock_client) - table.insert(sock_clients, sock_client) end end for _, sock_client in ipairs(sock_clients) do |