mirror of
https://github.com/cloudwu/skynet.git
synced 2026-07-22 02:53:09 +00:00
bed435660e
* fix: httpc支持ipv6 * httpc ip:port解析放在c层 * lua-socket.c listen增加ip:port字符串解析, 相关cluster代码修改 #2124 * socketchannel changehost port可置空 * socketchannel changehost恢复port判空 * httpc解析域名加端口号逻辑优化 * 注释中字符串模式内容格式优化 --------- Co-authored-by: zhuyin.zhu <zhuyin.zhu@bytedance.com>
84 lines
2.1 KiB
Lua
84 lines
2.1 KiB
Lua
local skynet = require "skynet"
|
|
local sc = require "skynet.socketchannel"
|
|
local socket = require "skynet.socket"
|
|
local cluster = require "skynet.cluster.core"
|
|
|
|
local channel
|
|
local session = 1
|
|
local node, nodename, init_host, init_port = ...
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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)
|
|
if not host then
|
|
skynet.error("Close cluster sender", channel.__host, channel.__port)
|
|
channel:close()
|
|
else
|
|
channel:changehost(host, port)
|
|
channel:connect(true)
|
|
end
|
|
skynet.ret(skynet.pack(nil))
|
|
end
|
|
|
|
skynet.start(function()
|
|
channel = sc.channel {
|
|
host = init_host,
|
|
port = tonumber(init_port),
|
|
response = read_response,
|
|
nodelay = true,
|
|
}
|
|
skynet.dispatch("lua", function(session , source, cmd, ...)
|
|
local f = assert(command[cmd])
|
|
f(...)
|
|
end)
|
|
end)
|