summaryrefslogtreecommitdiff
path: root/tests/modules/socket/sched_tcp/file_server.lua
blob: 4461f977e6191a808bb33be819e400809e6fa4b4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import("core.base.socket")
import("core.base.scheduler")

function _session(sock, filepath)

    local file = io.open(filepath, 'rb')
    if file then
        local send = sock:sendfile(file, {block = true})
        print("%s: send %s %d bytes!", sock, filepath, send)
        file:close()
    end
    sock:close()
end

function _listen(addr, port, filepath)

    local sock = socket.bind(addr, port)
    sock:listen(100)
    print("%s: listening %s:%d ..", sock, addr, port)
    while true do
        local sock_client = sock:accept()
        if sock_client then
            print("%s: accepted", sock_client)
            scheduler.co_start(_session, sock_client, filepath)
        end
    end
    sock:close()
end

function main(filepath)
    scheduler.co_start(_listen, "127.0.0.1", 9092, filepath)
end