lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-25 04:33:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

example for sproto

Browse Source
This commit is contained in:
Cloud Wu
2014-09-07 21:29:53 +08:00
parent 01269c88ff
commit 872b3c2743
4 changed files with 102 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
lualib/sproto.lua
View File

@@ -1,11 +1,12 @@
local core = require "sproto.core"
local assert = assert
local sproto = {}
local rpc = {}
local host = {}
local weak_mt = { __mode = "kv" }
local sproto_mt = { __index = sproto }
local rpc_mt = { __index = rpc }
local host_mt = { __index = host }
function sproto_mt:__gc()
core.deleteproto(self.__cobj)
@@ -27,14 +28,14 @@ function sproto.parse(ptext)
return sproto.new(pbin)
end
function sproto:rpc( packagename )
function sproto:host( packagename )
packagename = packagename or "package"
local obj = {
__proto = self,
__package = core.querytype(self.__cobj, packagename),
__session = {},
}
return setmetatable(obj, rpc_mt)
return setmetatable(obj, host_mt)
end
local function querytype(self, typename)
@@ -89,35 +90,22 @@ local function queryproto(self, pname)
end
local header_tmp = {}
function rpc:request(name, args, session)
local proto = queryproto(self.__proto, name)
header_tmp.type = proto.tag
header_tmp.session = session
local header = core.encode(self.__package, header_tmp)
if session then
self.__session[session] = assert(proto.response)
end
if args then
local content = core.encode(proto.request, args)
return core.pack(header .. content)
else
return core.pack(header)
end
end
local function gen_response(self, response, session)
return function(args)
header_tmp.type = nil
header_tmp.session = session
local header = core.encode(self.__package, header_tmp)
local content = core.encode(response, args)
return core.pack(header .. content)
if response then
local content = core.encode(response, args)
return core.pack(header .. content)
else
return core.pack(header)
end
end
end
function rpc:dispatch(...)
function host:dispatch(...)
local bin = core.unpack(...)
header_tmp.type = nil
header_tmp.session = nil
@@ -126,26 +114,46 @@ function rpc:dispatch(...)
if header.type then
-- request
local proto = queryproto(self.__proto, header.type)
local result
if proto.request then
local result = core.decode(proto.request, content)
if header_tmp.session then
return "REQUEST", proto.name, result, gen_response(self, proto.response, header_tmp.session)
else
return "REQUEST", proto.name, result
end
result = core.decode(proto.request, content)
end
if header_tmp.session then
return "REQUEST", proto.name, result, gen_response(self, proto.response, header_tmp.session)
else
if header_tmp.session then
return "REQUEST", proto.name, nil, gen_response(self, proto.response, header_tmp.session)
else
return "REQUEST", proto.name
end
return "REQUEST", proto.name, result
end
else
-- response
local session = assert(header_tmp.session, "session not found")
local response = assert(self.__session[session], "Unknown session")
self.__session[session] = nil
return "RESPONSE", session, core.decode(response, content)
if response == true then
return "RESPONSE", session
else
local result = core.decode(response, content)
return "RESPONSE", session, result
end
end
end
function host:attach(sp)
return function(name, args, session)
local proto = queryproto(sp, name)
header_tmp.type = proto.tag
header_tmp.session = session
local header = core.encode(self.__package, header_tmp)
if session then
self.__session[session] = proto.response or true
end
if args then
local content = core.encode(proto.request, args)
return core.pack(header .. content)
else
return core.pack(header)
end
end
end