require("utils.StringUtil")
require("utils.TableUtil")
require "utils.MathUtil"
local config = require "robot_config"
local skynet = require "skynet"
local netpack = require "skynet.netpack"
local socketdriver = require "skynet.socketdriver"
local deque = require "deque"
local string_format = string.format

local poold, pooldObj, onlineCount, startId, endId, inpre
local factory

local queue = {}
local fd2serv = {}
local id2fd = {}
local MSG = {}
local id_max 


function MSG.open( ... )
	log("open", ...)
end

function MSG.close(fd)
	if fd2serv[fd] and not fd2serv[fd].closing then
		fd2serv[fd].closing = true
		skynet.call(fd2serv[fd].agent, "lua", "exit")
		pcall(skynet.send, poold, "lua", "feed")

		log(string_format("logout %s", fd2serv[fd].id))
		id2fd[fd2serv[fd].id] = nil
		fd2serv[fd] = nil
	end
end

function MSG.error(fd, msg)
	log("MSG.error", fd, msg)
	MSG.close(fd)
end

function MSG.data(fd, msg, sz)
	if fd2serv[fd] then
		skynet.redirect(fd2serv[fd].agent, 0, "client", 0, msg, sz)
	end
end

local function dispatch_queue()
	local fd, msg, sz = netpack.pop(queue)
	if fd then
		skynet.fork(dispatch_queue)
		MSG.data(fd, msg, sz)
		for fd, msg, sz in netpack.pop, queue do
			MSG.data(fd, msg, sz)
		end
	end
end
MSG.more = dispatch_queue

skynet.register_protocol {
	name = "socket",
	id = skynet.PTYPE_SOCKET,
	unpack = function (msg, sz)
		return netpack.filter(queue, msg, sz)
	end,
	dispatch = function (_, _, q, type, ...)
		queue = q
		if type then
			MSG[type](...)
		end
	end
}

skynet.register_protocol {
	name = "client",
	id = skynet.PTYPE_CLIENT,
}

local function add_robot()
	local time = skynet.time()
	local robot = factory:pop()
	local fd = socketdriver.connect(config.host, config.port)
	socketdriver.start(fd)
	local id
	if id_max >= endId then
		while true do
			id = math.randomInt(startId, endId)
			if not id2fd[id] then
				break
			end
		end
	else
		id_max = id_max + 1
		id = id_max
	end
	fd2serv[fd] = {agent = robot, id = id}
	id2fd[id] = fd
	pcall(skynet.call, robot, "lua", "start", fd, id, prefix)
	log(string_format("login %s", fd2serv[fd].id))
	-- 定时下线
	skynet.timeout(math.randomInt(config.online_time[1], config.online_time[2]) * 100, function()
		MSG.close(fd)
		socketdriver.close(fd)
	end)
end

local function online()
	local curCount = 0
	for _, _ in pairs(fd2serv) do
		curCount = curCount + 1
	end

	-- 及时补充人数
	if curCount < onlineCount then
		for i = curCount + 1, onlineCount do
			add_robot()
		end
	end

	-- log(string_format("online  %s", curCount))
	skynet.timeout(100, online)
end

local function start(pooldp, pooldObj, onlineCountp, startIdp, endIdp, inprep)
	log("start testing ...")

	factory = deque.clone(pooldObj)
	poold, onlineCount, startId, endId, inpre = pooldp, onlineCountp, startIdp, endIdp, inprep
	id_max = startId - 1

	for i = 1, onlineCount, inpre do
		for j = 1, inpre do
			if i + j - 1 > onlineCount then
				break
			end
			add_robot()
		end
		skynet.sleep(100)
	end

	-- 每秒检查补充在线人数
	online()
end

skynet.start(function()
	skynet.dispatch("lua", function (_, _, command, ...)
		if command == "start" then
			start(...)
			return
		end
	end)
end)