You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.4 KiB
Lua
63 lines
1.4 KiB
Lua
|
|
local skynet = require "skynet"
|
|
require "skynet.manager"
|
|
local setting_template = require "settings"
|
|
|
|
local skynet_node_name = ...
|
|
|
|
local CMD = {}
|
|
local pool = {}
|
|
|
|
local next_id = 0
|
|
local maxconn = 1
|
|
|
|
local function next_conn()
|
|
local id = next_id % maxconn + 1
|
|
next_id = next_id + 1
|
|
if id > maxconn then
|
|
id = 1
|
|
end
|
|
return pool[id]
|
|
end
|
|
|
|
local function getconn(key)
|
|
if key and (type(key) == "number" or tonumber(key)) then
|
|
local id = math.floor((tonumber(key) - 1) % maxconn) + 1
|
|
return pool[id]
|
|
else
|
|
return next_conn()
|
|
end
|
|
end
|
|
|
|
local function call_mysqldb_slave(addr, ...)
|
|
return skynet.call(addr, "lua", "query", ...)
|
|
end
|
|
|
|
local function start()
|
|
local settings = setting_template.db_cnf[skynet_node_name]
|
|
INFO("mysqldbpool 启动", skynet_node_name, inspect(settings))
|
|
maxconn = tonumber(settings.mysqldb_maxinst) or 1
|
|
for i = 1, maxconn do
|
|
local mysqldb_slave = skynet.newservice("mysqldb_slave")
|
|
skynet.call(mysqldb_slave, "lua", "start", settings.mysqldb_cnf)
|
|
table.insert(pool, mysqldb_slave)
|
|
end
|
|
end
|
|
|
|
|
|
function CMD.query(sql)
|
|
local executer = getconn()
|
|
return call_mysqldb_slave(executer, sql)
|
|
end
|
|
|
|
|
|
skynet.start(function()
|
|
start()
|
|
|
|
skynet.dispatch("lua", function(_, _, cmd, ...)
|
|
local f = assert(CMD[cmd], cmd .. "not found")
|
|
skynet.retpack(f(...))
|
|
end)
|
|
|
|
skynet.register('.' .. SERVICE_NAME)
|
|
end) |