在 TCP 服务器中向处理函数纤程发送消息

如果从 socket 模块传递到 tcp_server() 的处理函数是作为纤程运行的,那么是否有可能通过 fiber.channel 与每个 TCP 连接通信?

点赞
用户1182787
用户1182787

是的,它是这样的。

#!/usr/bin/tarantool

local fiber = require('fiber')
local socket = require('socket')

local clients = {}

function rc_handle(s)
    -- 你可以在某些表中保存套接字的引用
    clients[s] = true

    -- 你可以创建一个通道
    -- 我建议将其附加到套接字上
    -- 这样更容易收集垃圾
    s.channel = fiber.channel()

    -- 你也可以获取处理纤程的引用
    -- 它将帮助你区分活着的客户端和死亡的客户端
    s.fiber = fiber.self()

    s:write(string.format('Message for %s:%s: %s',
        s:peer().host, s:peer().port, s.channel:get()
    ))

    -- 不要忘记手动取消引用客户端
    -- 或者您可以将客户端表设置为弱表
    clients[s] = nil
end

server = socket.tcp_server('127.0.0.1', 3003, {
    name = 'srv',
    handler = rc_handle,
})

function greet_all(msg)
    -- 所以你可以将你的消息广播到所有客户端
    for s, _ in pairs(clients) do
        s.channel:put(msg)
    end
end

require('console').start()

当然,这个代码片段远非完美,但我希望它可以帮助你完成工作。

2020-10-02 15:17:56