You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
1.1 KiB
40 lines
1.1 KiB
#!/usr/bin/sysbench
|
|
|
|
require("account_common")
|
|
|
|
|
|
-- 初始化数据库 第一个调用的lua函数
|
|
function thread_init()
|
|
print("thread_init ID:" .. thread_id .. " max:" .. sysbench.opt.threads)
|
|
drv = sysbench.sql.driver()
|
|
con = drv:connect()
|
|
end
|
|
|
|
function thread_done()
|
|
print("thread_done ID:" .. thread_id)
|
|
con:disconnect()
|
|
end
|
|
|
|
-- 每个db 1万个账号
|
|
local account_map = { }
|
|
local insert_index = 1;
|
|
function insert_one_data()
|
|
local one_thread_count = account_table_max / sysbench.opt.threads;
|
|
local db_start = thread_id * one_thread_count;
|
|
local cur_start = (insert_index-1) % one_thread_count + 1;
|
|
local db_index = db_start + cur_start;
|
|
-- print("insert_one_data insert_index:" .. insert_index .. " thread_id:" .. thread_id .. " db_index:" .. db_index)
|
|
insert_index = insert_index + 1;
|
|
|
|
account_map[db_index] = account_map[db_index] or 1;
|
|
if (account_map[db_index] > 10000) then -- 1万就停止
|
|
return
|
|
end
|
|
|
|
con:query(insert_account_cmd(db_index, account_map[db_index]));
|
|
account_map[db_index] = account_map[db_index] + 1;
|
|
end
|
|
|
|
function event()
|
|
insert_one_data();
|
|
end
|
|
|