CONNMIX 开发 WebSocket 视频弹幕

CONNMIX 开发 WebSocket 视频弹幕

使用WebSocket制作一个单机版弹幕系统非常简单，但是当单机性能达到瓶颈，需要扩展为集群部署时就会面临很多分布式问题，使用CONNMIX则无需担心这些问题，很少的代码即可完成一个高性能分布式WebSocket集群。

要求

• connmix >= v1.0.4

设计思路

• 客户端在ws连接成功后，订阅一个视频ID的通道 video:<videoid>
• 在发送弹幕的接口中，调用connmix任意节点的 /v1/mesh/publish 接口往对应 video_id 发送实时弹幕，所有订阅该通道的ws客户端都将会收到消息。
• 以上都是增量推送设计，全量通常都是在页面加载时通过一个全量api接口获取。

交互协议设计

• 当用户发送 100001@video 我们在 lua 代码中就执行订阅 video:100001 通道。
• 前端切换视频时，可先发送取消弹幕消息，然后发送新的订阅弹幕消息。

安装引擎

• CONNMIX https://connmix.com/docs/1.0/#/zh-cn/install-engine

修改配置

在 connmix.yaml 配置文件的 options 选项，修改websocket的url路径

options:
  - name: path
    value: /barrage-videos

CONNMIX 编码

修改 entry.websocket.lua 的 on_message 方法如下：

• 当接受到subscribe、unsubscribe消息时，执行订阅/取消订阅对应的通道

function on_message(msg)
    --print(msg)
    if msg["type"] ~= "text" then
        conn:close()
        return
    end

    local conn = mix.websocket()

    local data, err = mix.json_decode(msg["data"])
    if err then
        mix_log(mix_DEBUG, "json_decode error: " .. err)
        conn:close()
        return
    end

    local op = data["op"]
    local channel_raw = data["channel"]
    local channel_table = mix.str_split(channel_raw, "@")
    if table.getn(channel_table) ~= 2 then
        mix_log(mix_DEBUG, "invalid channel: " .. channel_raw)
        conn:close()
        return
    end
    local video_id = channel_table[1] --Lua的table索引默认从1开始
    local channel_type = channel_table[2]
    
    if op == "subscribe" and channel_type == "video" then
        local err = conn:subscribe("video:" .. video_id)
        if err then
            mix_log(mix_DEBUG, "subscribe error: " .. err)
            conn:close()
            return
        end
    end
    
    if op == "unsubscribe" and channel_type == "video" then
        local err = conn:unsubscribe("video:" .. video_id)
        if err then
            mix_log(mix_DEBUG, "unsubscribe error: " .. err)
            conn:close()
            return
        end
    end

    conn:send('{"result":true}')
end

API 编码

接下来在现有系统的框架中实现主动弹幕推送

• 可以在 spring、laravel 框架中写一个发送弹幕接口
• 该接口中验证完用户身份后，执行以下http请求完成推送
• 如果发送请求非常频繁，可以改用 websocket-api推送 提升性能

curl --request POST 'http://127.0.0.1:6789/v1/mesh/publish' \
--header 'Content-Type: application/json' \
--data-raw '{
    "c": "video:100001",
    "d": "{\"event\":\"@video\",\"data\":{\"uid\":1001,\"video_id\":100001,\"msg\":\"Hello,World!\"}}"
}'

测试

使用 wstool 进行测试

• 连接 ws://127.0.0.1:6790/barrage-videos
• 发送 {"op":"subscribe","channel":"100001@video"}
• 接收到 {"result":true}
• 执行 curl 主动推送
• 接收到 {"event":"@video","data":{"uid":1001,"video_id":100001,"msg":"Hello,World!"}}