parent
b2d555933e
commit
5beb83ba9e
@ -0,0 +1,101 @@
|
|||||||
|
bc = {}
|
||||||
|
|
||||||
|
_bc_backend = require("_bc_backend")
|
||||||
|
|
||||||
|
function bc:init(master_id)
|
||||||
|
local o = {}
|
||||||
|
setmetatable(o, self)
|
||||||
|
self.__index = self
|
||||||
|
-- Init backend and cb handler
|
||||||
|
o._listen_callbacks = {}
|
||||||
|
o._verb_handlers = {}
|
||||||
|
o._listeners = {}
|
||||||
|
local function cb_handler(cb_type, param, arg)
|
||||||
|
if cb_type == "listen" then
|
||||||
|
if type(o._listen_callbacks[param]) == "function" then
|
||||||
|
o._listen_callbacks[param]()
|
||||||
|
end
|
||||||
|
elseif cb_type == "verb" then
|
||||||
|
o._verb_handlers[param](arg)
|
||||||
|
elseif cb_type == "noun" then
|
||||||
|
return o._mynouns[param]
|
||||||
|
elseif cb_type == "reg_nl" then
|
||||||
|
-- Create array if there is not one already
|
||||||
|
o._listeners[param] = o._listeners[param] or {}
|
||||||
|
-- Insert parameters
|
||||||
|
table.insert(o._listeners[param], arg)
|
||||||
|
return #o._listeners[param]
|
||||||
|
elseif cb_type == "unreg_nl" then
|
||||||
|
o._listeners[param][arg] = true;
|
||||||
|
else
|
||||||
|
error("Wrong cb type")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
o._backend = _bc_backend:init(cb_handler)
|
||||||
|
-- Fetch list of nouns and verbs from master
|
||||||
|
local list = o._backend:fetch_list(master_id)
|
||||||
|
o._nouns = list.nouns
|
||||||
|
o._verbs = list.verbs
|
||||||
|
-- Initialize own nouns
|
||||||
|
o._mynouns = {}
|
||||||
|
for noun, addr in pairs(o._nouns) do
|
||||||
|
if addr == o._backend.local_addr then
|
||||||
|
o._mynouns[noun] = 0
|
||||||
|
end
|
||||||
|
end
|
||||||
|
list = nil
|
||||||
|
return o
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:get_noun(n)
|
||||||
|
if self._nouns[n] == self._backend.local_addr then
|
||||||
|
return self._mynouns[n]
|
||||||
|
else
|
||||||
|
return self._backend:request_noun(self._nouns[n], n)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:set_noun(n, val)
|
||||||
|
if self._nouns[n] == self._backend.local_addr then
|
||||||
|
-- Call listeners
|
||||||
|
if type(self._listeners[n]) == "table" then
|
||||||
|
for i, listener in ipairs(self._listeners[n]) do
|
||||||
|
-- Check condition
|
||||||
|
if type(listener) == "table" and
|
||||||
|
( (listener.query == "onchange" and self._mynouns[n] ~= val)
|
||||||
|
or (listener.query == "onrising" and self._mynouns[n] < val)
|
||||||
|
or (listener.query == "onfalling" and self._mynouns[n] > val)
|
||||||
|
or (listener.query == "onabove" and val > listener.qparam)
|
||||||
|
or (listener.query == "onbelow" and val < listener.qparam)
|
||||||
|
or (listener.query == "onvalue" and val == listener.qparam)
|
||||||
|
) then
|
||||||
|
self._backend:call_listener(listener.addr, n, i, val)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
-- Set value
|
||||||
|
self._mynouns[n] = val
|
||||||
|
return true
|
||||||
|
end
|
||||||
|
return false
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:on_verb(v, handler)
|
||||||
|
self._verb_handlers[v] = handler
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:call_verb(v, param)
|
||||||
|
self._backend:call_verb(self._verbs[v], v, param)
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:listen_noun(n, query, qparam, cb)
|
||||||
|
local id = self._backend:register_callback(self._nouns[n], n, query, qparam)
|
||||||
|
self._listen_callbacks[id] = cb
|
||||||
|
return id
|
||||||
|
end
|
||||||
|
|
||||||
|
function bc:listen_cancel(n, id)
|
||||||
|
self._backend:cancel_callback(n, id)
|
||||||
|
end
|
||||||
|
|
||||||
|
return bc
|
||||||
@ -1,101 +1,154 @@
|
|||||||
bc = {}
|
-- For testing
|
||||||
|
modem = require("modem")
|
||||||
|
|
||||||
|
-- For deployment
|
||||||
|
-- modem = require("component").modem
|
||||||
|
|
||||||
|
serializer = require("serialization")
|
||||||
|
|
||||||
_bc_backend = require("_bc_backend")
|
Queue = {first = 0, last = -1}
|
||||||
|
|
||||||
function bc:init(master_id)
|
function Queue:new()
|
||||||
local o = {}
|
local o = {}
|
||||||
setmetatable(o, self)
|
setmetatable(o, self)
|
||||||
self.__index = self
|
self.__index = self
|
||||||
-- Init backend and cb handler
|
|
||||||
o._listen_callbacks = {}
|
|
||||||
o._verb_handlers = {}
|
|
||||||
o._listeners = {}
|
|
||||||
local function cb_handler(cb_type, param, arg)
|
|
||||||
if cb_type == "listen" then
|
|
||||||
if type(o._listen_callbacks[param]) == "function" then
|
|
||||||
o._listen_callbacks[param]()
|
|
||||||
end
|
|
||||||
elseif cb_type == "verb" then
|
|
||||||
o._verb_handlers[param](arg)
|
|
||||||
elseif cb_type == "noun" then
|
|
||||||
return o._mynouns[param]
|
|
||||||
elseif cb_type == "reg_nl" then
|
|
||||||
-- Create array if there is not one already
|
|
||||||
o._listeners[param] = o._listeners[param] or {}
|
|
||||||
-- Insert parameters
|
|
||||||
table.insert(o._listeners[param], arg)
|
|
||||||
return #o._listeners[param]
|
|
||||||
elseif cb_type == "unreg_nl" then
|
|
||||||
o._listeners[param][arg] = true;
|
|
||||||
else
|
|
||||||
error("Wrong cb type")
|
|
||||||
end
|
|
||||||
end
|
|
||||||
o._backend = _bc_backend:init(cb_handler)
|
|
||||||
-- Fetch list of nouns and verbs from master
|
|
||||||
local list = o._backend:fetch_list(master_id)
|
|
||||||
o._nouns = list.nouns
|
|
||||||
o._verbs = list.verbs
|
|
||||||
-- Initialize own nouns
|
|
||||||
o._mynouns = {}
|
|
||||||
for noun, addr in pairs(o._nouns) do
|
|
||||||
if addr == o._backend.local_addr then
|
|
||||||
o._mynouns[noun] = 0
|
|
||||||
end
|
|
||||||
end
|
|
||||||
list = nil
|
|
||||||
return o
|
return o
|
||||||
end
|
end
|
||||||
|
|
||||||
function bc:get_noun(n)
|
function Queue:insert(value)
|
||||||
if self._nouns[n] == self._backend.local_addr then
|
local last = self.last + 1
|
||||||
return self._mynouns[n]
|
self.last = last
|
||||||
else
|
self[last] = value
|
||||||
return self._backend:request_noun(self._nouns[n], n)
|
end
|
||||||
|
|
||||||
|
function Queue:isempty()
|
||||||
|
return self.first > self.last
|
||||||
|
end
|
||||||
|
|
||||||
|
function Queue:remove()
|
||||||
|
local first = self.first
|
||||||
|
if self:isempty() then
|
||||||
|
error("Queue is empty")
|
||||||
end
|
end
|
||||||
|
local value = self[first]
|
||||||
|
self[first] = nil
|
||||||
|
self.first = first + 1
|
||||||
|
return value
|
||||||
end
|
end
|
||||||
|
|
||||||
function bc:set_noun(n, val)
|
message_type = {
|
||||||
if self._nouns[n] == self._backend.local_addr then
|
request_noun = 1,
|
||||||
-- Call listeners
|
call_verb = 2,
|
||||||
if type(self._listeners[n]) == "table" then
|
noun_response = 3,
|
||||||
for i, listener in ipairs(self._listeners[n]) do
|
hello = 4,
|
||||||
-- Check condition
|
goodbye = 5,
|
||||||
if type(listener) == "table" and
|
hello_response = 6,
|
||||||
( (listener.query == "onchange" and self._mynouns[n] ~= val)
|
}
|
||||||
or (listener.query == "onrising" and self._mynouns[n] < val)
|
|
||||||
or (listener.query == "onfalling" and self._mynouns[n] > val)
|
CFG_PORT = 1234
|
||||||
or (listener.query == "onabove" and val > listener.qparam)
|
|
||||||
or (listener.query == "onbelow" and val < listener.qparam)
|
bc = {}
|
||||||
or (listener.query == "onvalue" and val == listener.qparam)
|
|
||||||
) then
|
function bc:init(my_address, local_nouns, local_verbs)
|
||||||
self._backend:call_listener(listener.addr, n, i, val)
|
local o = {}
|
||||||
|
setmetatable(o, self)
|
||||||
|
self.__index = self
|
||||||
|
o.queue = Queue:new()
|
||||||
|
o.remote_verbs = {}
|
||||||
|
o.remote_nouns = {}
|
||||||
|
-- Modem listener
|
||||||
|
function modem_listener(localAddress, remoteAddress, port, dist, message)
|
||||||
|
dbg = message
|
||||||
|
message = serializer.unserialize(message)
|
||||||
|
if port == CFG_PORT then
|
||||||
|
if message.ty == message_type.request_noun then
|
||||||
|
msg = {ty=message_type.noun_response,
|
||||||
|
noun=message.noun,
|
||||||
|
value=o.local_nouns[message.noun]
|
||||||
|
}
|
||||||
|
o.modem:send(remoteAddress, port, serializer.serialize(msg))
|
||||||
|
elseif message.ty == message_type.noun_response then
|
||||||
|
-- TODO: Make this a bit better, this is dumb
|
||||||
|
o._nr = message.value
|
||||||
|
elseif message.ty == message_type.call_verb then
|
||||||
|
o.local_verbs[message.verb](o, message.param)
|
||||||
|
elseif message.ty == message_type.hello then
|
||||||
|
for i,verb in ipairs(message.verbs) do
|
||||||
|
o.remote_verbs[verb] = remoteAddress
|
||||||
|
end
|
||||||
|
for i,noun in ipairs(message.nouns) do
|
||||||
|
o.remote_nouns[noun] = remoteAddress
|
||||||
|
end
|
||||||
|
-- Respond with own verbs and nouns
|
||||||
|
local mynouns = {}
|
||||||
|
for noun in pairs(o.local_nouns) do
|
||||||
|
table.insert(mynouns, noun)
|
||||||
|
end
|
||||||
|
local myverbs = {}
|
||||||
|
for verb in pairs(o.local_verbs) do
|
||||||
|
table.insert(myverbs, verb)
|
||||||
|
end
|
||||||
|
o.modem:send(remoteAddress, port, serializer.serialize({ty=message_type.hello_response, verbs=myverbs, nouns=mynouns}))
|
||||||
|
elseif message.ty == message_type.hello_response then
|
||||||
|
for i,verb in ipairs(message.verbs) do
|
||||||
|
o.remote_verbs[verb] = remoteAddress
|
||||||
|
end
|
||||||
|
for i,noun in ipairs(message.nouns) do
|
||||||
|
o.remote_nouns[noun] = remoteAddress
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
-- Set value
|
|
||||||
self._mynouns[n] = val
|
|
||||||
return true
|
|
||||||
end
|
end
|
||||||
return false
|
-- Change this for deployment VVV
|
||||||
end
|
o.modem = modem:init(my_address, modem_listener)
|
||||||
|
-- Init local stuff
|
||||||
function bc:on_verb(v, handler)
|
o.local_verbs = local_verbs
|
||||||
self._verb_handlers[v] = handler
|
o.local_nouns = local_nouns
|
||||||
|
-- Send own nouns and verbs and request all remotes to send theirs
|
||||||
|
local mynouns = {}
|
||||||
|
for noun in pairs(local_nouns) do
|
||||||
|
table.insert(mynouns, noun)
|
||||||
|
end
|
||||||
|
local myverbs = {}
|
||||||
|
for verb in pairs(local_verbs) do
|
||||||
|
table.insert(myverbs, verb)
|
||||||
|
end
|
||||||
|
o.modem:broadcast(CFG_PORT, serializer.serialize({ty=message_type.hello, verbs=myverbs, nouns=mynouns}))
|
||||||
|
return o
|
||||||
end
|
end
|
||||||
|
|
||||||
function bc:call_verb(v, param)
|
function bc:get_noun(n)
|
||||||
self._backend:call_verb(self._verbs[v], v, param)
|
if self.local_nouns[n] ~= nil then -- This noun is local, makes it easy
|
||||||
|
return self.local_nouns[n]
|
||||||
|
elseif self.remote_nouns[n] ~= nil then -- This noun is remote
|
||||||
|
self.modem:send(self.remote_nouns[n], CFG_PORT, serializer.serialize({ty=message_type.request_noun, noun=n}))
|
||||||
|
while self._nr == nil do
|
||||||
|
-- Nothing, wait for response, TODO: Research performance impact
|
||||||
|
end
|
||||||
|
local value = self._nr
|
||||||
|
self._nr = nil -- Reset
|
||||||
|
return value
|
||||||
|
else
|
||||||
|
error("Noun not found, Node might be offline")
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
function bc:listen_noun(n, query, qparam, cb)
|
function bc:set_noun(n, value)
|
||||||
local id = self._backend:register_callback(self._nouns[n], n, query, qparam)
|
if self.local_nouns[n] ~= nil then
|
||||||
self._listen_callbacks[id] = cb
|
self.local_nouns[n] = value
|
||||||
return id
|
else
|
||||||
|
error("Can't set remote nouns")
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
function bc:listen_cancel(n, id)
|
function bc:call_verb(v, param)
|
||||||
self._backend:cancel_callback(n, id)
|
if self.local_verbs[v] ~= nil then
|
||||||
|
self.local_verbs[v](self, param)
|
||||||
|
elseif self.remote_verbs[v] ~= nil then
|
||||||
|
self.modem:send(self.remote_verbs[v], CFG_PORT, serializer.serialize({ty=message_type.call_verb, verb=v, param=param}))
|
||||||
|
else
|
||||||
|
error("Verb not found, Node might be offline")
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
return bc
|
return bc
|
||||||
|
|||||||
Loading…
Reference in new issue