TDEngine压力测试思路

TDEngine 压测思路

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

压测控制stress_testd

加载时操作

1.加载压测文件夹模块&＃xff0c;目录stress_test

2.注册玩家登陆成功回调事件&＃xff0c;并注册玩家心跳回调

3.账号名起始字段

4.注册压测定时器

启动压测模块

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

-- 有则新登陆number玩家再进行操作 if number thenlocal arg &＃61; {number &＃61; number,extra_data &＃61; {test_modules &＃61; test_modules}}login(arg) -- 没有则对已登陆的玩家进行操作 elselocal player_list &＃61; child_objects(PLAYER_TDCLS)for _, player in ipairs(player_list) doplayer:set("test_modules", test_modules)end end

玩家登陆

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

-- 批量登陆(number个玩家) function login(arg)-- 统计登陆数login_number &＃61; login_number &＃43; 1arg.number &＃61; arg.number - 1-- 构造账号名(命名规则 "时间戳_编号")local account &＃61; 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)

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

-- 每个玩家的子模块的时间间隔 local interval &＃61; player:query("interval") if not interval thenreturn end -- 每个玩家的子模块的累计时间 local accumulate &＃61; player:query("accumulate") or {} -- 每个玩家需要测试的子模块 local test_modules &＃61; 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] &＃61; 0end-- 累加时间accumulate[test_module] &＃61; accumulate[test_module] &＃43; HEARTBEAT_INTERVAL-- 累计时间 >&＃61; 间隔时间&＃xff0c;则执行子模块操作if accumulate[test_module] >&＃61; interval[test_module] thenlocal child_module &＃61; _G[test_module]if child_module and type(child_module.operation) &＃61;&＃61; "function" then-- 调用子模块的统一接口child_module.operation(player)elsetrace("找不到压力测试子模块(%o) 或者该子模块未定义&＃39;operation&＃39;接口!", test_module)test_modules[test_module] &＃61; nilend-- 累计时间清零&＃xff0c;重新计算accumulate[test_module] &＃61; 0endelsetrace("要求测试的压力子模块(%o)并未定义!", test_module)test_modules[test_module] &＃61; nilend end -- 保存累计时间 player:set("accumulate", accumulate)