mirror of
https://github.com/cloudwu/skynet.git
synced 2026-07-24 12:20:41 +00:00
32 lines
597 B
Lua
32 lines
597 B
Lua
local socket = require "socket"
|
|
|
|
local readbytes = socket.read
|
|
local writebytes = socket.write
|
|
|
|
local sockethelper = {}
|
|
local socket_error = setmetatable({} , { __tostring = function() return "[Socket Error]" end })
|
|
|
|
sockethelper.socket_error = socket_error
|
|
|
|
function sockethelper.readfunc(fd)
|
|
return function (sz)
|
|
local ret = readbytes(fd, sz)
|
|
if ret then
|
|
return ret
|
|
else
|
|
error(socket_error)
|
|
end
|
|
end
|
|
end
|
|
|
|
function sockethelper.writefunc(fd)
|
|
return function(content)
|
|
local ok = writebytes(fd, content)
|
|
if not ok then
|
|
error(socket_error)
|
|
end
|
|
end
|
|
end
|
|
|
|
return sockethelper
|