作者:shihuaping0918@163.com,转载请注明作者
这个sockethelper可讲的东西不多,它提供一个connect函数给外部使用,这个connect能够支持阻塞和非阻塞连接两种方式。
另外提供readfunc和writefunc两个函数,这两个函数将fd参数隐藏起来,形成闭包,再返回闭包。
local socket = require "skynet.socket"
local skynet = require "skynet"
local readbytes = socket.read
local writebytes = socket.write
local sockethelper = {}
--不支持tostring
local socket_error = setmetatable({} , { __tostring = function() return "[Socket Error]" end })
sockethelper.socket_error = socket_error
--预读字符串str
local function preread(fd, str)
return function (sz)
if str then --预读字符串不为空
if sz == #str or sz == nil then
local ret = str
str = nil
return ret
else --数据不匹配
if sz < #str then --读到的数据长度不够
local ret = str:sub(1,sz)
str = str:sub(sz + 1)
return ret
else --读到的数据长度够长了
sz = sz - #str
local ret = readbytes(fd, sz) --再读剩下想要读的内容
if ret then
return str .. ret
else
error(socket_error)
end
end
end
else --预读为空
local ret = readbytes(fd, sz)
if ret then
return ret
else
error(socket_error)
end
end
end
end
--功能就是从fd上读数据
-- 返回的是闭包
-- 这个闭包把fd给隐藏了
function sockethelper.readfunc(fd, pre)
if pre then
return preread(fd, pre)
end
return function (sz)
local ret = readbytes(fd, sz)
if ret then
return ret
else
error(socket_error)
end
end
end
sockethelper.readall = socket.readall
--往fd上写数据
-- 返回的也是闭包
-- 闭包把fd隐藏了
function sockethelper.writefunc(fd)
return function(content)
local ok = writebytes(fd, content)
if not ok then
error(socket_error)
end
end
end
--连接到服务器
--阻塞和非阻塞都支持
function sockethelper.connect(host, port, timeout)
local fd
if timeout then --非阻塞,使用协程实现
local drop_fd
local co = coroutine.running()
-- asynchronous connect
skynet.fork(function()
fd = socket.open(host, port)
if drop_fd then
-- sockethelper.connect already return, and raise socket_error
socket.close(fd)
else
-- socket.open before sleep, wakeup.
skynet.wakeup(co)
end
end)
skynet.sleep(timeout) --这里也是协程实现的,sleep的时候让出cpu
if not fd then
-- not connect yet
drop_fd = true
end
else
-- block connect
fd = socket.open(host, port)
end
if fd then
return fd
end
error(socket_error)
end
function sockethelper.close(fd)
socket.close(fd)
end
function sockethelper.shutdown(fd)
socket.shutdown(fd)
end
return sockethelper