87 lines
2.4 KiB
Lua

require "oil"
local Viewer = require "loop.debug.Viewer"
--------------------------------------------------------------------------------
local orb = oil.init{ flavor = "intercepted;corba;typed;cooperative;base" }
local viewer = Viewer{ maxdepth = 2 }
local interceptor = {}
--------------------------------------------------------------------------------
local send_context_idl = orb:loadidl [[
struct ServerInfo {
long memory;
};
]]
function interceptor:sendrequest(request)
print("intercepting request to "..request.operation.."("..viewer:tostring(unpack(request, 1, request.count))..")")
local encoder = orb:newencoder()
encoder:put({
memory = gcinfo(),
}, send_context_idl)
request.service_context = {
{
context_id = 1234,
context_data = encoder:getdata()
}
}
end
--------------------------------------------------------------------------------
local receive_context_idl = orb:loadidl [[
struct ClientInfo {
double start;
double ending;
};
]]
function interceptor:receivereply(reply)
print("intercepting reply of opreation "..reply.operation)
print("\tsuccess:", reply.success)
print("\tresults:", unpack(reply, 1, reply.count))
for _, context in ipairs(reply.service_context) do
if context.context_id == 4321 then
local decoder = orb:newdecoder(context.context_data)
local result = decoder:get(receive_context_idl)
print("\ttime:", result.ending - result.start)
return
end
end
io.stderr:write("context 4321 not found! Canceling ...\n")
reply.success = false
reply.count = 1
reply[1] = orb:newexcept{ "ACCESS_DENIED" }
end
--------------------------------------------------------------------------------
orb:setclientinterceptor(interceptor)
--------------------------------------------------------------------------------
if select("#", ...) == 0 then
io.stderr:write "usage: lua client.lua <time of client 1>, <time of client 2>, ..."
os.exit(-1)
end
local arg = {...}
--------------------------------------------------------------------------------
oil.main(function()
local proxy = orb:newproxy(assert(oil.readfrom("server.ior")))
local function showprogress(id, time)
print(id, "about to request work for "..time.." seconds")
if proxy:do_something_for(time)
then print(id, "result received successfully")
else print(id, "got an unexpected result")
end
end
for id, time in ipairs(arg) do
oil.newthread(showprogress, id, tonumber(time))
end
end)