TDEngine 压力测试思路

TDEngine 压测思路

消息分发复用TDEngine底层的逻辑，包括连接建立，消息读取，消息发送，消息执行，定时器执行

压测控制stress_testd

加载时操作

1.加载压测文件夹模块，目录stress_test

2.注册玩家登陆成功回调事件，并注册玩家心跳回调

3.账号名起始字段

4.注册压测定时器

启动压测模块

调用STRESS_TEST_D.start(500, "CHAT_TESTD")，第一个参数是需要登陆的人数，第二个参数是需要测试的模块，模块名以逗号隔开，如果有设置登陆的人数，则进行登陆，如果没有设置登陆人数则对在线的玩家设置测试模块

-- 有则新登陆number玩家再进行操作
if number then
    local arg = {
        number     = number,
        extra_data = {test_modules = test_modules}
    }
    login(arg)
-- 没有 则对已登陆的玩家进行操作
else
    local player_list = child_objects(PLAYER_TDCLS)
    for _, player in ipairs(player_list) do
        player:set("test_modules", test_modules)
    end
end

玩家登陆

一次调用函数只登陆一个，如果还有需要登陆的用户，则启动定时器，间隔一定时间后登陆

-- 批量登陆(number个玩家)
function login(arg)
    -- 统计登陆数
    login_number = login_number + 1
    arg.number   = arg.number - 1
    -- 构造账号名(命名规则 "时间戳_编号")
    local account = string.format("%d_%d", start_time, login_number)
    -- 登陆一个玩家
    LOGIN_D.login(account, "default_password", arg.extra_data)
    if arg.number > 0 then
        -- 间隔一段时间后再登陆下一玩家
        set_timer(TIME_LOGIN, login, arg)
    end
end

登陆成功处理

function func_login_ok(player)

登陆成功主要设置玩家的测试模块及模块的测试间隔

心跳函数处理

-- 玩家心跳函数(负责玩家所有测试子模块的操作)
function heartbeat_handler(player)

心跳函数主要负责定时调用相应的测试模块，每个模块都有自己的独立计时，如果累计时间超过执行一次需要的时间则进的一次调用，并把累计时间清0

-- 每个玩家的子模块的时间间隔
local interval = player:query("interval")
if not interval then
    return 
end
-- 每个玩家的子模块的累计时间
local accumulate = player:query("accumulate") or {}
-- 每个玩家需要测试的子模块
local test_modules = player:query("test_modules") or {}
-- 循环执行需要测试的子模块
for test_module, _ in pairs(test_modules) do
    if interval[test_module] then
        -- 初始化累加时间
        if not accumulate[test_module] then
            accumulate[test_module] = 0
        end
        -- 累加时间
        accumulate[test_module] = accumulate[test_module] + HEARTBEAT_INTERVAL
        -- 累计时间 >= 间隔时间，则执行子模块操作
        if accumulate[test_module] >= interval[test_module] then
            local child_module = _G[test_module]
            if child_module and type(child_module.operation) == "function" then
                -- 调用子模块的统一接口
                child_module.operation(player)
            else
                trace("找不到压力测试子模块(%o) 或者 该子模块未定义'operation'接口!", test_module)
                test_modules[test_module] = nil
            end
            -- 累计时间清零，重新计算
            accumulate[test_module] = 0
        end
    else
        trace("要求测试的压力子模块(%o)并未定义!", test_module)
        test_modules[test_module] = nil
    end
end
-- 保存累计时间
player:set("accumulate", accumulate)

测试模块要求

测试模块需实现下面两个方法

-- 达到每个玩家的该模块间隔时间，则调用该函数
function operation(player)
--操作的时间间隔，毫秒
function random_interval()

总结

通过丰富测试模块，及丰富PLAYER_CLASS的数据，可以达模拟一个玩家正常的操作流程，可以达到更真实的测试效果。

项目

TDEngine