multi cluster sender

service/clustersender.lua

@@ -0,0 +1,87 @@
+local skynet = require "skynet"
+local sc = require "skynet.socketchannel"
+local socket = require "skynet.socket"
+local cluster = require "skynet.cluster.core"
+local ignoreret = skynet.ignoreret
+
+local channel
+local session = 1
+
+local command = {}
+
+local function send_request(addr, msg, sz)
+	-- msg is a local pointer, cluster.packrequest will free it
+	local current_session = session
+	local request, new_session, padding = cluster.packrequest(addr, session, msg, sz)
+	session = new_session
+
+	-- node_channel[node] may yield or throw error
+	local tracetag = skynet.tracetag()
+	if tracetag then
+		if tracetag:sub(1,1) ~= "(" then
+			-- add nodename
+			local newtag = string.format("(%s-%s-%d)%s", nodename, node, session, tracetag)
+			skynet.tracelog(tracetag, string.format("session %s", newtag))
+			tracetag = newtag
+		end
+		skynet.tracelog(tracetag, string.format("cluster %s", node))
+		channel:request(cluster.packtrace(tracetag))
+	end
+	return channel:request(request, current_session, padding)
+end
+
+function command.req(...)
+	local ok, msg = pcall(send_request, ...)
+	if ok then
+		if type(msg) == "table" then
+			skynet.ret(cluster.concat(msg))
+		else
+			skynet.ret(msg)
+		end
+	else
+		skynet.error(msg)
+		skynet.response()(false)
+	end
+end
+
+function command.push(addr, msg, sz)
+	local request, new_session, padding = cluster.packpush(addr, session, msg, sz)
+	if padding then	-- is multi push
+		session = new_session
+	end
+
+	-- node_channel[node] may yield or throw error
+	channel:request(request, nil, padding)
+end
+
+local function read_response(sock)
+	local sz = socket.header(sock:read(2))
+	local msg = sock:read(sz)
+	return cluster.unpackresponse(msg)	-- session, ok, data, padding
+end
+
+function command.changenode(host, port)
+	local c = sc.channel {
+			host = host,
+			port = tonumber(port),
+			response = read_response,
+			nodelay = true,
+		}
+	succ, err = pcall(c.connect, c, true)
+	if succ then
+		channel = c
+		skynet.ret(skynet.pack(nil))
+	else
+		skynet.response()(false)
+	end
+end
+
+skynet.start(function()
+	skynet.dispatch("lua", function(session , source, cmd, ...)
+		local f = assert(command[cmd])
+		f(...)
+	end)
+end)
+
+
+