diff options
author | Steven Barth <steven@midlink.org> | 2009-06-14 09:02:40 +0000 |
---|---|---|
committer | Steven Barth <steven@midlink.org> | 2009-06-14 09:02:40 +0000 |
commit | bdf0a4b520bdeb2c892f7eaadf3242c1b824bf11 (patch) | |
tree | d954cf6fdd8f426ce7aff795534837bfb0af1caf /libs/rpcc/luasrc/rpcc.lua | |
parent | c47be2e727d1bb3e2e3aa415ec96be2a5f8a45b7 (diff) |
GSoC: Add RPC client
Diffstat (limited to 'libs/rpcc/luasrc/rpcc.lua')
-rw-r--r-- | libs/rpcc/luasrc/rpcc.lua | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/libs/rpcc/luasrc/rpcc.lua b/libs/rpcc/luasrc/rpcc.lua new file mode 100644 index 0000000000..ef94d3bbd5 --- /dev/null +++ b/libs/rpcc/luasrc/rpcc.lua @@ -0,0 +1,84 @@ +--[[ +LuCIRPCc +(c) 2009 Steven Barth <steven@midlink.org> + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +$Id$ +]]-- + +local util = require "luci.util" +local json = require "luci.json" +local ltn12 = require "luci.ltn12" +local nixio = require "nixio", require "nixio.util" + +local tostring, assert, setmetatable = tostring, assert, setmetatable +local error = error + +--- LuCI RPC Client. +-- @cstyle instance +module "luci.rpcc" + +RQLIMIT = 32 * nixio.const.buffersize + +--- Create a new JSON-RPC stream client. +-- @class function +-- @param fd File descriptor +-- @param v1 Use protocol version 1.0 +-- @return RPC Client +Client = util.class() + +function Client.__init__(self, fd, v1) + self.fd = fd + self.uniqueid = tostring(self):match("0x([a-f0-9]+)") + self.msgid = 1 + self.v1 = v1 +end + +--- Request an RP call and get the response. +-- @param method Remote method +-- @param params Parameters +-- @param notification Notification only? +-- @return response +function Client.request(self, method, params, notification) + local oldchunk = self.decoder and self.decoder.chunk + self.decoder = json.ActiveDecoder(self.fd:blocksource(nil, RQLIMIT)) + self.decoder.chunk = oldchunk + + local reqid = self.msgid .. self.uniqueid + local reqdata = json.Encoder({ + id = (not notification) and (self.msgid .. self.uniqueid) or nil, + jsonrpc = (not self.v1) and "2.0" or nil, + method = method, + params = params + }) + ltn12.pump.all(reqdata:source(), self.fd:sink()) + if not notification then + self.msgid = self.msgid + 1 + local response = self.decoder:get() + assert(response.id == reqid, "Invalid response id") + if response.error then + error(response.error.message or response.error) + end + return response.result + end +end + +--- Create a transparent RPC proxy. +-- @param prefix Method prefix +-- @return RPC Proxy object +function Client.proxy(self, prefix) + prefix = prefix or "" + return setmetatable({}, { + __call = function(proxy, ...) + return self:request(prefix, {...}) + end, + __index = function(proxy, name) + return self:proxy(prefix .. name .. ".") + end + }) +end
\ No newline at end of file |