blob: 4d4bbfba0788f9ace0a90da1c56fe90cbd016129 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
--- @module etcd
local etcd = {}
-- @function update subtree configuration
local function update_subtree(tree)
if not tree then return end
for _, k in pairs(tree) do
if k.dir then
update_subtree(k.nodes)
else
local key,opt = k.key:gmatch('([^/]+)/([^/]+)$')()
if _G[key][opt] ~= k.value then
_G[key][opt] = k.value
end
end
end
end
-- @function reload whole configuration
function etcd.reload()
local res, err = etcd.cli:readdir('/', true)
if err then
error(err)
end
update_subtree(res.body.node.nodes)
end
function etcd.init()
etcd.Etcd = require('etcd.luasocket')
etcd.defaults = { prefix = '/knot-resolver' }
end
function etcd.deinit()
if etcd.ev then event.cancel(etcd.ev) end
end
function etcd.config(conf)
local options = etcd.defaults
if type(conf) == 'table' then
for k,v in pairs(conf) do options[k] = v end
end
-- create connection
local cli, err = etcd.Etcd.new(options)
if err then
error(err)
end
etcd.cli = cli
-- schedule recurrent polling
-- @todo: the etcd has watch() API, but this requires
-- coroutines on socket operations
if etcd.ev then event.cancel(etcd.ev) end
etcd.ev = event.recurrent(5 * sec, etcd.reload)
end
return etcd
|
