mirror of
https://github.com/cloudwu/skynet.git
synced 2026-07-25 12:43:09 +00:00
socketchannel is an independent mod now, channel:request will throw channel.error when disconnected.
This commit is contained in:
Cloud Wu
@@ -1,5 +1,6 @@
|
|||||||
local bson = require "bson"
|
local bson = require "bson"
|
||||||
local socket = require "socket"
|
local socket = require "socket"
|
||||||
|
local socketchannel = require "socketchannel"
|
||||||
local skynet = require "skynet"
|
local skynet = require "skynet"
|
||||||
local driver = require "mongo.driver"
|
local driver = require "mongo.driver"
|
||||||
local rawget = rawget
|
local rawget = rawget
|
||||||
@@ -78,7 +79,7 @@ end
|
|||||||
function mongo.client( obj )
|
function mongo.client( obj )
|
||||||
obj.port = obj.port or 27017
|
obj.port = obj.port or 27017
|
||||||
obj.__id = 0
|
obj.__id = 0
|
||||||
obj.__sock = socket.channel {
|
obj.__sock = socketchannel.channel {
|
||||||
host = obj.host,
|
host = obj.host,
|
||||||
port = obj.port,
|
port = obj.port,
|
||||||
response = dispatch_reply,
|
response = dispatch_reply,
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
local skynet = require "skynet"
|
local skynet = require "skynet"
|
||||||
local socket = require "socket"
|
local socket = require "socket"
|
||||||
|
local socketchannel = require "socketchannel"
|
||||||
local config = require "config"
|
local config = require "config"
|
||||||
local redis_conf = skynet.getenv "redis"
|
local redis_conf = skynet.getenv "redis"
|
||||||
local name = config (redis_conf)
|
local name = config (redis_conf)
|
||||||
@@ -84,7 +85,7 @@ end
|
|||||||
|
|
||||||
function redis.connect(dbname)
|
function redis.connect(dbname)
|
||||||
local db_conf = name[dbname]
|
local db_conf = name[dbname]
|
||||||
local channel = socket.channel {
|
local channel = socketchannel.channel {
|
||||||
host = db_conf.host,
|
host = db_conf.host,
|
||||||
port = db_conf.port or 6379,
|
port = db_conf.port or 6379,
|
||||||
auth = redis_login(db_conf.auth, db_conf.db),
|
auth = redis_login(db_conf.auth, db_conf.db),
|
||||||
@@ -175,7 +176,7 @@ function redis.watch(dbname)
|
|||||||
__subscribe = {},
|
__subscribe = {},
|
||||||
__psubscribe = {},
|
__psubscribe = {},
|
||||||
}
|
}
|
||||||
local channel = socket.channel {
|
local channel = socketchannel.channel {
|
||||||
host = db_conf.host,
|
host = db_conf.host,
|
||||||
port = db_conf.port or 6379,
|
port = db_conf.port or 6379,
|
||||||
auth = watch_login(obj, db_conf.auth),
|
auth = watch_login(obj, db_conf.auth),
|
||||||
|
|||||||
@@ -158,7 +158,7 @@ function suspend(co, result, command, param, size)
|
|||||||
end
|
end
|
||||||
session_coroutine_id[co] = nil
|
session_coroutine_id[co] = nil
|
||||||
session_coroutine_address[co] = nil
|
session_coroutine_address[co] = nil
|
||||||
error(debug.traceback(co,command))
|
error(debug.traceback(co,tostring(command)))
|
||||||
end
|
end
|
||||||
if command == "CALL" then
|
if command == "CALL" then
|
||||||
c.trace_register(trace_handle, param)
|
c.trace_register(trace_handle, param)
|
||||||
@@ -444,13 +444,13 @@ local function dispatch_message(...)
|
|||||||
if not fork_succ then
|
if not fork_succ then
|
||||||
if succ then
|
if succ then
|
||||||
succ = false
|
succ = false
|
||||||
err = fork_err
|
err = tostring(fork_err)
|
||||||
else
|
else
|
||||||
err = err .. "\n" .. fork_err
|
err = tostring(err) .. "\n" .. tostring(fork_err)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
assert(succ, err)
|
assert(succ, tostring(err))
|
||||||
end
|
end
|
||||||
|
|
||||||
function skynet.newservice(name, ...)
|
function skynet.newservice(name, ...)
|
||||||
|
|||||||
@@ -149,7 +149,7 @@ function socket.start(id, func)
|
|||||||
return connect(id, func)
|
return connect(id, func)
|
||||||
end
|
end
|
||||||
|
|
||||||
local function clear_socket(id)
|
function socket.shutdown(id)
|
||||||
local s = socket_pool[id]
|
local s = socket_pool[id]
|
||||||
if s then
|
if s then
|
||||||
if s.buffer then
|
if s.buffer then
|
||||||
@@ -180,7 +180,7 @@ function socket.close(id)
|
|||||||
end
|
end
|
||||||
s.connected = false
|
s.connected = false
|
||||||
end
|
end
|
||||||
clear_socket(id)
|
socket.shutdown(id)
|
||||||
assert(s.lock_set == nil or next(s.lock_set) == nil)
|
assert(s.lock_set == nil or next(s.lock_set) == nil)
|
||||||
socket_pool[id] = nil
|
socket_pool[id] = nil
|
||||||
end
|
end
|
||||||
@@ -306,296 +306,4 @@ function socket.abandon(id)
|
|||||||
socket_pool[id] = nil
|
socket_pool[id] = nil
|
||||||
end
|
end
|
||||||
|
|
||||||
-- channel support auto reconnect , and capture socket error in request/response transaction
|
|
||||||
-- { host = "", port = , auth = function(so) , response = function(so) session, data }
|
|
||||||
|
|
||||||
local channel = {}
|
|
||||||
local channel_socket = {}
|
|
||||||
local channel_meta = { __index = channel }
|
|
||||||
local channel_socket_meta = {
|
|
||||||
__index = channel_socket,
|
|
||||||
__gc = function(cs)
|
|
||||||
local fd = cs[1]
|
|
||||||
cs[1] = false
|
|
||||||
if fd then
|
|
||||||
clear_socket(fd)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
}
|
|
||||||
local socket_error = channel_socket -- alias for error object
|
|
||||||
|
|
||||||
function socket.channel(desc)
|
|
||||||
local c = {
|
|
||||||
__host = assert(desc.host),
|
|
||||||
__port = assert(desc.port),
|
|
||||||
__auth = desc.auth,
|
|
||||||
__response = desc.response,
|
|
||||||
__request = {}, -- request seq { response func or session }
|
|
||||||
__thread = {}, -- coroutine seq or session->coroutine map
|
|
||||||
__result = {}, -- response result { coroutine -> result }
|
|
||||||
__result_data = {},
|
|
||||||
__connecting = {},
|
|
||||||
__sock = false,
|
|
||||||
__closed = false,
|
|
||||||
__authcoroutine = false,
|
|
||||||
}
|
|
||||||
|
|
||||||
return setmetatable(c, channel_meta)
|
|
||||||
end
|
|
||||||
|
|
||||||
local function close_channel_socket(self)
|
|
||||||
if self.__sock then
|
|
||||||
local so = self.__sock
|
|
||||||
self.__sock = false
|
|
||||||
-- never raise error
|
|
||||||
pcall(socket.close,so[1])
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function wakeup_all(self, errmsg)
|
|
||||||
for i = 1, #self.__thread do
|
|
||||||
local co = self.__thread[i]
|
|
||||||
self.__thread[i] = nil
|
|
||||||
self.__result[co] = socket_error
|
|
||||||
self.__result_data[co] = errmsg
|
|
||||||
skynet.wakeup(co)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function dispatch_response(self)
|
|
||||||
local response = self.__response
|
|
||||||
if response then
|
|
||||||
-- response() return session
|
|
||||||
while self.__sock do
|
|
||||||
local ok , session, result_ok, result_data = pcall(response, self.__sock)
|
|
||||||
if ok and session then
|
|
||||||
local co = self.__thread[session]
|
|
||||||
self.__thread[session] = nil
|
|
||||||
if co then
|
|
||||||
self.__result[co] = result_ok
|
|
||||||
self.__result_data[co] = result_data
|
|
||||||
skynet.wakeup(co)
|
|
||||||
else
|
|
||||||
print("socket: unknown session :", session)
|
|
||||||
end
|
|
||||||
else
|
|
||||||
close_channel_socket(self)
|
|
||||||
local errormsg
|
|
||||||
if session ~= socket_error then
|
|
||||||
errormsg = session
|
|
||||||
end
|
|
||||||
for k,co in pairs(self.__thread) do
|
|
||||||
-- throw error (errormsg)
|
|
||||||
self.__thread[k] = nil
|
|
||||||
self.__result[co] = socket_error
|
|
||||||
self.__result_data[co] = errormsg
|
|
||||||
skynet.wakeup(co)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
else
|
|
||||||
-- pop response function from __request
|
|
||||||
while self.__sock do
|
|
||||||
local func = table.remove(self.__request, 1)
|
|
||||||
if func == nil then
|
|
||||||
if not socket.block(self.__sock[1]) then
|
|
||||||
close_channel_socket(self)
|
|
||||||
wakeup_all(self)
|
|
||||||
end
|
|
||||||
else
|
|
||||||
local ok, result_ok, result_data = pcall(func, self.__sock)
|
|
||||||
if ok then
|
|
||||||
local co = table.remove(self.__thread, 1)
|
|
||||||
self.__result[co] = result_ok
|
|
||||||
self.__result_data[co] = result_data
|
|
||||||
skynet.wakeup(co)
|
|
||||||
else
|
|
||||||
close_channel_socket(self)
|
|
||||||
local errmsg
|
|
||||||
if result ~= socket_error then
|
|
||||||
errmsg = result_ok
|
|
||||||
end
|
|
||||||
wakeup_all(self, errmsg)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function connect_once(self)
|
|
||||||
assert(not self.__sock and not self.__authcoroutine)
|
|
||||||
local fd = socket.open(self.__host, self.__port)
|
|
||||||
if not fd then
|
|
||||||
return false
|
|
||||||
end
|
|
||||||
self.__authcoroutine = coroutine.running()
|
|
||||||
self.__sock = setmetatable( {fd} , channel_socket_meta )
|
|
||||||
skynet.fork(dispatch_response, self)
|
|
||||||
|
|
||||||
if self.__auth then
|
|
||||||
local ok , message = pcall(self.__auth, self)
|
|
||||||
if not ok then
|
|
||||||
close_channel_socket(self)
|
|
||||||
if message ~= socket_error then
|
|
||||||
print("socket: auth failed", message)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
self.__authcoroutine = false
|
|
||||||
return ok
|
|
||||||
end
|
|
||||||
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
|
|
||||||
local function try_connect(self)
|
|
||||||
local t = 100
|
|
||||||
while not self.__closed do
|
|
||||||
if connect_once(self) then
|
|
||||||
return
|
|
||||||
end
|
|
||||||
if t > 1000 then
|
|
||||||
print("socket: try to reconnect", self.__host, self.__port)
|
|
||||||
skynet.sleep(t)
|
|
||||||
t = 0
|
|
||||||
else
|
|
||||||
skynet.sleep(t)
|
|
||||||
end
|
|
||||||
t = t + 100
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
local function block_connect(self)
|
|
||||||
if self.__sock then
|
|
||||||
local authco = self.__authcoroutine
|
|
||||||
if not authco then
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
if authco == coroutine.running() then
|
|
||||||
-- authing
|
|
||||||
return true
|
|
||||||
end
|
|
||||||
end
|
|
||||||
if self.__closed then
|
|
||||||
return false
|
|
||||||
end
|
|
||||||
if #self.__connecting > 0 then
|
|
||||||
-- connecting in other coroutine
|
|
||||||
local co = coroutine.running()
|
|
||||||
table.insert(self.__connecting, co)
|
|
||||||
skynet.wait()
|
|
||||||
-- check connection again
|
|
||||||
return block_connect(self)
|
|
||||||
end
|
|
||||||
|
|
||||||
self.__connecting[1] = true
|
|
||||||
try_connect(self)
|
|
||||||
self.__connecting[1] = nil
|
|
||||||
for i=2, #self.__connecting do
|
|
||||||
local co = self.__connecting[i]
|
|
||||||
self.__connecting[i] = nil
|
|
||||||
skynet.wakeup(co)
|
|
||||||
end
|
|
||||||
|
|
||||||
-- check again
|
|
||||||
return block_connect(self)
|
|
||||||
end
|
|
||||||
|
|
||||||
function channel:connect()
|
|
||||||
if self.__closed then
|
|
||||||
self.__closed = false
|
|
||||||
end
|
|
||||||
|
|
||||||
return block_connect(self)
|
|
||||||
end
|
|
||||||
|
|
||||||
function channel:request(request, response)
|
|
||||||
assert(block_connect(self))
|
|
||||||
|
|
||||||
if not socket.write(self.__sock[1], request) then
|
|
||||||
return self:request(request, response)
|
|
||||||
end
|
|
||||||
|
|
||||||
if response == nil then
|
|
||||||
-- no response
|
|
||||||
return
|
|
||||||
end
|
|
||||||
|
|
||||||
local co = coroutine.running()
|
|
||||||
|
|
||||||
if self.__response then
|
|
||||||
-- response is session
|
|
||||||
self.__thread[response] = co
|
|
||||||
else
|
|
||||||
-- response is a function, push it to __request
|
|
||||||
table.insert(self.__request, response)
|
|
||||||
table.insert(self.__thread, co)
|
|
||||||
end
|
|
||||||
skynet.wait()
|
|
||||||
|
|
||||||
local result = self.__result[co]
|
|
||||||
self.__result[co] = nil
|
|
||||||
local result_data = self.__result_data[co]
|
|
||||||
self.__result_data[co] = nil
|
|
||||||
|
|
||||||
if result == socket_error then
|
|
||||||
if result_data then
|
|
||||||
print("socket: dispatch", request, result_data)
|
|
||||||
end
|
|
||||||
return self:request(request, response)
|
|
||||||
else
|
|
||||||
assert(result, result_data)
|
|
||||||
return result_data
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function channel:response(response)
|
|
||||||
assert(block_connect(self))
|
|
||||||
|
|
||||||
assert(type(response) == "function")
|
|
||||||
|
|
||||||
local co = coroutine.running()
|
|
||||||
table.insert(self.__request, response)
|
|
||||||
table.insert(self.__thread, co)
|
|
||||||
|
|
||||||
skynet.wait()
|
|
||||||
|
|
||||||
local result = self.__result[co]
|
|
||||||
self.__result[co] = nil
|
|
||||||
local result_data = self.__result_data[co]
|
|
||||||
self.__result_data[co] = nil
|
|
||||||
|
|
||||||
if result == socket_error then
|
|
||||||
if result_data then
|
|
||||||
print("socket: dispatch", request, result_data)
|
|
||||||
end
|
|
||||||
return self:response(response)
|
|
||||||
else
|
|
||||||
assert(result, result_data)
|
|
||||||
return result_data
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
function channel:close()
|
|
||||||
if not self.__closed then
|
|
||||||
self.__closed = true
|
|
||||||
close_channel_socket(self)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
channel_meta.__gc = channel.close
|
|
||||||
|
|
||||||
local function wrapper_socket_function(f)
|
|
||||||
return function(self, ...)
|
|
||||||
local result = f(self[1], ...)
|
|
||||||
if not result then
|
|
||||||
error(socket_error)
|
|
||||||
else
|
|
||||||
return result
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
channel_socket.read = wrapper_socket_function(socket.read)
|
|
||||||
channel_socket.readline = wrapper_socket_function(socket.readline)
|
|
||||||
|
|
||||||
return socket
|
return socket
|
||||||
|
|||||||
300
lualib/socketchannel.lua
Normal file
300
lualib/socketchannel.lua
Normal file
@@ -0,0 +1,300 @@
|
|||||||
|
local skynet = require "skynet"
|
||||||
|
local socket = require "socket"
|
||||||
|
|
||||||
|
-- channel support auto reconnect , and capture socket error in request/response transaction
|
||||||
|
-- { host = "", port = , auth = function(so) , response = function(so) session, data }
|
||||||
|
|
||||||
|
local socket_channel = {}
|
||||||
|
local channel = {}
|
||||||
|
local channel_socket = {}
|
||||||
|
local channel_meta = { __index = channel }
|
||||||
|
local channel_socket_meta = {
|
||||||
|
__index = channel_socket,
|
||||||
|
__gc = function(cs)
|
||||||
|
local fd = cs[1]
|
||||||
|
cs[1] = false
|
||||||
|
if fd then
|
||||||
|
socket.shutdown(fd)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
}
|
||||||
|
|
||||||
|
local socket_error = setmetatable({}, {__tostring = function() return "[Error: socket]" end }) -- alias for error object
|
||||||
|
socket_channel.error = socket_error
|
||||||
|
|
||||||
|
function socket_channel.channel(desc)
|
||||||
|
local c = {
|
||||||
|
__host = assert(desc.host),
|
||||||
|
__port = assert(desc.port),
|
||||||
|
__auth = desc.auth,
|
||||||
|
__response = desc.response,
|
||||||
|
__request = {}, -- request seq { response func or session }
|
||||||
|
__thread = {}, -- coroutine seq or session->coroutine map
|
||||||
|
__result = {}, -- response result { coroutine -> result }
|
||||||
|
__result_data = {},
|
||||||
|
__connecting = {},
|
||||||
|
__sock = false,
|
||||||
|
__closed = false,
|
||||||
|
__authcoroutine = false,
|
||||||
|
}
|
||||||
|
|
||||||
|
return setmetatable(c, channel_meta)
|
||||||
|
end
|
||||||
|
|
||||||
|
local function close_channel_socket(self)
|
||||||
|
if self.__sock then
|
||||||
|
local so = self.__sock
|
||||||
|
self.__sock = false
|
||||||
|
-- never raise error
|
||||||
|
pcall(socket.close,so[1])
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function wakeup_all(self, errmsg)
|
||||||
|
for i = 1, #self.__thread do
|
||||||
|
local co = self.__thread[i]
|
||||||
|
self.__thread[i] = nil
|
||||||
|
self.__result[co] = socket_error
|
||||||
|
self.__result_data[co] = errmsg
|
||||||
|
skynet.wakeup(co)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function dispatch_response(self)
|
||||||
|
local response = self.__response
|
||||||
|
if response then
|
||||||
|
-- response() return session
|
||||||
|
while self.__sock do
|
||||||
|
local ok , session, result_ok, result_data = pcall(response, self.__sock)
|
||||||
|
if ok and session then
|
||||||
|
local co = self.__thread[session]
|
||||||
|
self.__thread[session] = nil
|
||||||
|
if co then
|
||||||
|
self.__result[co] = result_ok
|
||||||
|
self.__result_data[co] = result_data
|
||||||
|
skynet.wakeup(co)
|
||||||
|
else
|
||||||
|
print("socket: unknown session :", session)
|
||||||
|
end
|
||||||
|
else
|
||||||
|
close_channel_socket(self)
|
||||||
|
local errormsg
|
||||||
|
if session ~= socket_error then
|
||||||
|
errormsg = session
|
||||||
|
end
|
||||||
|
for k,co in pairs(self.__thread) do
|
||||||
|
-- throw error (errormsg)
|
||||||
|
self.__thread[k] = nil
|
||||||
|
self.__result[co] = socket_error
|
||||||
|
self.__result_data[co] = errormsg
|
||||||
|
skynet.wakeup(co)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
else
|
||||||
|
-- pop response function from __request
|
||||||
|
while self.__sock do
|
||||||
|
local func = table.remove(self.__request, 1)
|
||||||
|
if func == nil then
|
||||||
|
if not socket.block(self.__sock[1]) then
|
||||||
|
close_channel_socket(self)
|
||||||
|
wakeup_all(self)
|
||||||
|
end
|
||||||
|
else
|
||||||
|
local ok, result_ok, result_data = pcall(func, self.__sock)
|
||||||
|
if ok then
|
||||||
|
local co = table.remove(self.__thread, 1)
|
||||||
|
self.__result[co] = result_ok
|
||||||
|
self.__result_data[co] = result_data
|
||||||
|
skynet.wakeup(co)
|
||||||
|
else
|
||||||
|
close_channel_socket(self)
|
||||||
|
local errmsg
|
||||||
|
if result ~= socket_error then
|
||||||
|
errmsg = result_ok
|
||||||
|
end
|
||||||
|
wakeup_all(self, errmsg)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function connect_once(self)
|
||||||
|
assert(not self.__sock and not self.__authcoroutine)
|
||||||
|
local fd = socket.open(self.__host, self.__port)
|
||||||
|
if not fd then
|
||||||
|
return false
|
||||||
|
end
|
||||||
|
self.__authcoroutine = coroutine.running()
|
||||||
|
self.__sock = setmetatable( {fd} , channel_socket_meta )
|
||||||
|
skynet.fork(dispatch_response, self)
|
||||||
|
|
||||||
|
if self.__auth then
|
||||||
|
local ok , message = pcall(self.__auth, self)
|
||||||
|
if not ok then
|
||||||
|
close_channel_socket(self)
|
||||||
|
if message ~= socket_error then
|
||||||
|
print("socket: auth failed", message)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
self.__authcoroutine = false
|
||||||
|
return ok
|
||||||
|
end
|
||||||
|
|
||||||
|
self.__authcoroutine = false
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
|
||||||
|
local function try_connect(self)
|
||||||
|
local t = 100
|
||||||
|
while not self.__closed do
|
||||||
|
if connect_once(self) then
|
||||||
|
return
|
||||||
|
end
|
||||||
|
if t > 1000 then
|
||||||
|
print("socket: try to reconnect", self.__host, self.__port)
|
||||||
|
skynet.sleep(t)
|
||||||
|
t = 0
|
||||||
|
else
|
||||||
|
skynet.sleep(t)
|
||||||
|
end
|
||||||
|
t = t + 100
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function block_connect(self)
|
||||||
|
if self.__sock then
|
||||||
|
local authco = self.__authcoroutine
|
||||||
|
if not authco then
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
if authco == coroutine.running() then
|
||||||
|
-- authing
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
end
|
||||||
|
if self.__closed then
|
||||||
|
return false
|
||||||
|
end
|
||||||
|
if #self.__connecting > 0 then
|
||||||
|
-- connecting in other coroutine
|
||||||
|
local co = coroutine.running()
|
||||||
|
table.insert(self.__connecting, co)
|
||||||
|
skynet.wait()
|
||||||
|
-- check connection again
|
||||||
|
return block_connect(self)
|
||||||
|
end
|
||||||
|
|
||||||
|
self.__connecting[1] = true
|
||||||
|
try_connect(self)
|
||||||
|
self.__connecting[1] = nil
|
||||||
|
for i=2, #self.__connecting do
|
||||||
|
local co = self.__connecting[i]
|
||||||
|
self.__connecting[i] = nil
|
||||||
|
skynet.wakeup(co)
|
||||||
|
end
|
||||||
|
|
||||||
|
-- check again
|
||||||
|
return block_connect(self)
|
||||||
|
end
|
||||||
|
|
||||||
|
function channel:connect()
|
||||||
|
if self.__closed then
|
||||||
|
self.__closed = false
|
||||||
|
end
|
||||||
|
|
||||||
|
return block_connect(self)
|
||||||
|
end
|
||||||
|
|
||||||
|
function channel:request(request, response)
|
||||||
|
assert(block_connect(self))
|
||||||
|
|
||||||
|
if not socket.write(self.__sock[1], request) then
|
||||||
|
error(socket_error)
|
||||||
|
end
|
||||||
|
|
||||||
|
if response == nil then
|
||||||
|
-- no response
|
||||||
|
return
|
||||||
|
end
|
||||||
|
|
||||||
|
local co = coroutine.running()
|
||||||
|
|
||||||
|
if self.__response then
|
||||||
|
-- response is session
|
||||||
|
self.__thread[response] = co
|
||||||
|
else
|
||||||
|
-- response is a function, push it to __request
|
||||||
|
table.insert(self.__request, response)
|
||||||
|
table.insert(self.__thread, co)
|
||||||
|
end
|
||||||
|
skynet.wait()
|
||||||
|
|
||||||
|
local result = self.__result[co]
|
||||||
|
self.__result[co] = nil
|
||||||
|
local result_data = self.__result_data[co]
|
||||||
|
self.__result_data[co] = nil
|
||||||
|
|
||||||
|
if result == socket_error then
|
||||||
|
if result_data then
|
||||||
|
print("socket: dispatch", request, result_data)
|
||||||
|
end
|
||||||
|
error(socket_error)
|
||||||
|
else
|
||||||
|
assert(result, result_data)
|
||||||
|
return result_data
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
function channel:response(response)
|
||||||
|
assert(block_connect(self))
|
||||||
|
|
||||||
|
assert(type(response) == "function")
|
||||||
|
|
||||||
|
local co = coroutine.running()
|
||||||
|
table.insert(self.__request, response)
|
||||||
|
table.insert(self.__thread, co)
|
||||||
|
|
||||||
|
skynet.wait()
|
||||||
|
|
||||||
|
local result = self.__result[co]
|
||||||
|
self.__result[co] = nil
|
||||||
|
local result_data = self.__result_data[co]
|
||||||
|
self.__result_data[co] = nil
|
||||||
|
|
||||||
|
if result == socket_error then
|
||||||
|
if result_data then
|
||||||
|
print("socket: dispatch", request, result_data)
|
||||||
|
end
|
||||||
|
error(socket_error)
|
||||||
|
else
|
||||||
|
assert(result, result_data)
|
||||||
|
return result_data
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
function channel:close()
|
||||||
|
if not self.__closed then
|
||||||
|
self.__closed = true
|
||||||
|
close_channel_socket(self)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
channel_meta.__gc = channel.close
|
||||||
|
|
||||||
|
local function wrapper_socket_function(f)
|
||||||
|
return function(self, ...)
|
||||||
|
local result = f(self[1], ...)
|
||||||
|
if not result then
|
||||||
|
error(socket_error)
|
||||||
|
else
|
||||||
|
return result
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
channel_socket.read = wrapper_socket_function(socket.read)
|
||||||
|
channel_socket.readline = wrapper_socket_function(socket.readline)
|
||||||
|
|
||||||
|
return socket_channel
|
||||||
Reference in New Issue
Block a user