Daniel Barlow
c3f550698d
it was only working by accident, when it worked, which was by no means all of the time note that we unconditionally perform the action (restart or whatever) once we've started and got the initial state of the outputs. That's because we have no idea whether the outputs changed in the interval between the controlled service initially starting and watch-outputs starting, so updates in that interval could be lost
97 lines
3.3 KiB
Fennel
97 lines
3.3 KiB
Fennel
(local { : %% : system : assoc : split : table= : dig } (require :anoia))
|
|
(local svc (require :anoia.svc))
|
|
(local { : kill &as ll} (require :lualinux))
|
|
(import-macros { : define-tests : expect : expect= } :anoia.assert)
|
|
|
|
(local { : view } (require :fennel))
|
|
|
|
(fn output-refs [outputs]
|
|
(let [result {}]
|
|
(each [_ v (ipairs outputs)]
|
|
(let [[service path] (split ":" v)
|
|
paths (or (. result service) [])]
|
|
(table.insert paths (split "/" path ))
|
|
(tset result service paths)))
|
|
result))
|
|
|
|
(fn parse-args [args]
|
|
(match args
|
|
["-r" service & rest] (assoc (parse-args rest)
|
|
:controlled-service service
|
|
:action :restart)
|
|
["-R" service & rest] (assoc (parse-args rest)
|
|
:controlled-service service
|
|
:action :restart-all)
|
|
["-s" signal service & rest] (assoc (parse-args rest)
|
|
:controlled-service service
|
|
:action [:signal signal])
|
|
outputs { :output-references (output-refs outputs) } ))
|
|
|
|
(define-tests
|
|
(expect= (parse-args ["-r" "daemon"
|
|
"/nix/store/s1:out1"
|
|
"/nix/store/s2:out1" "/nix/store/s2:out2/ifname"])
|
|
{:action "restart"
|
|
:controlled-service "daemon"
|
|
:output-references
|
|
{"/nix/store/s1" [["out1"]]
|
|
"/nix/store/s2" [["out1"] ["out2" "ifname"]]}}
|
|
))
|
|
|
|
(fn changed? [paths old-tree new-tree]
|
|
(accumulate [changed? false
|
|
_ path (ipairs paths)]
|
|
(or changed? (not (table= (dig old-tree path) (dig new-tree path))))))
|
|
|
|
(define-tests
|
|
(expect (changed? [["ifname"]] {:ifname "true"} {:ifindex 2}))
|
|
(expect (changed? [["ifname"]] {:ifname "true"} {:ifname "false"}))
|
|
(expect (not (changed? [["ifname"]] {:ifname "true"} {:ifname "true"})))
|
|
(expect (not (changed? [["mtu"]] {:ifname "true"} {:ifname "false"})))
|
|
(expect (not (changed? [["mtu"]] {:ifname "true"} {:ifname "false"})))
|
|
)
|
|
|
|
(fn do-action [action service]
|
|
(case action
|
|
:restart (system (%% "s6-svc -r /run/service/%s" service))
|
|
:restart-all (system (%% "s6-rc -b -d %q; s6-rc-up-tree %q" service service))
|
|
[:signal n] (system (%% "s6-svc -s %q /run/service/%s" n service))))
|
|
|
|
(local POLLIN 0x0001)
|
|
(local POLLHUP 0x0010)
|
|
|
|
(fn wait-for-change [services]
|
|
(let [pollfds (collect [s _p (ipairs services)]
|
|
(bor (lshift (s:fileno) 32)
|
|
(lshift (bor POLLIN POLLHUP) 16)))]
|
|
(ll.poll pollfds)))
|
|
|
|
(fn open-services [output-references]
|
|
(collect [s p (pairs output-references)]
|
|
(values (svc.open s) p)))
|
|
|
|
(fn run []
|
|
(let [trees {}
|
|
{
|
|
: output-references
|
|
: controlled-service
|
|
: action
|
|
: watched-service
|
|
: paths } (parse-args arg)
|
|
services (open-services output-references)]
|
|
(while true
|
|
(when
|
|
(accumulate [changed false
|
|
service paths (pairs services)]
|
|
(let [new-tree (service:output ".")]
|
|
(if (changed? paths (or (. trees service) {}) new-tree)
|
|
(do (tset trees service new-tree) true)
|
|
changed)))
|
|
(do-action action controlled-service))
|
|
(wait-for-change services))))
|
|
|
|
|
|
|
|
|
|
{ : run }
|