TDEngine 压测思路
消息分发复用TDEngine底层的逻辑,包括连接建立,消息读取,消息发送,消息执行,定时器执行
压测控制stress_testd
加载时操作
1.加载压测文件夹模块,目录stress_test
2.注册玩家登陆成功回调事件,并注册玩家心跳回调
3.账号名起始字段
4.注册压测定时器
启动压测模块
调用STRESS_TEST_D.start(500, "CHAT_TESTD"),第一个参数是需要登陆的人数,第二个参数是需要测试的模块,模块名以逗号隔开,如果有设置登陆的人数,则进行登陆,如果没有设置登陆人数则对在线的玩家设置测试模块
-- 有则新登陆number玩家再进行操作
if number thenlocal arg = {number = number,extra_data = {test_modules = test_modules}}login(arg)
-- 没有 则对已登陆的玩家进行操作
elselocal player_list = child_objects(PLAYER_TDCLS)for _, player in ipairs(player_list) doplayer:set("test_modules", test_modules)end
end
玩家登陆
一次调用函数只登陆一个,如果还有需要登陆的用户,则启动定时器,间隔一定时间后登陆
-- 批量登陆(number个玩家)
function login(arg)-- 统计登陆数login_number = login_number + 1arg.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 thenreturn
end
-- 每个玩家的子模块的累计时间
local accumulate = player:query("accumulate") or {}
-- 每个玩家需要测试的子模块
local test_modules = player:query("test_modules") or {}
-- 循环执行需要测试的子模块
for test_module, _ in pairs(test_modules) doif interval[test_module] then-- 初始化累加时间if not accumulate[test_module] thenaccumulate[test_module] = 0end-- 累加时间accumulate[test_module] = accumulate[test_module] + HEARTBEAT_INTERVAL-- 累计时间 >= 间隔时间,则执行子模块操作if accumulate[test_module] >= interval[test_module] thenlocal child_module = _G[test_module]if child_module and type(child_module.operation) == "function" then-- 调用子模块的统一接口child_module.operation(player)elsetrace("找不到压力测试子模块(%o) 或者 该子模块未定义'operation'接口!", test_module)test_modules[test_module] = nilend-- 累计时间清零,重新计算accumulate[test_module] = 0endelsetrace("要求测试的压力子模块(%o)并未定义!", test_module)test_modules[test_module] = nilend
end
-- 保存累计时间
player:set("accumulate", accumulate)
测试模块要求
测试模块需实现下面两个方法
-- 达到每个玩家的该模块间隔时间,则调用该函数
function operation(player)
--操作的时间间隔,毫秒
function random_interval()
总结
通过丰富测试模块,及丰富PLAYER_CLASS的数据,可以达模拟一个玩家正常的操作流程,可以达到更真实的测试效果。
项目
TDEngine