mirror of https://github.com/qwc/backive.git
62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package events
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"net"
|
|
)
|
|
|
|
var ls net.Listener
|
|
var done <-chan struct{}
|
|
var callbacks := make([]func(map[string]string), 3)
|
|
|
|
// Init initializes the unix socket.
|
|
func Init(socketPath string) {
|
|
ls, err = net.Listen(socketPath)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// Listen starts the event loop.
|
|
func Listen() {
|
|
for {
|
|
go func() {
|
|
process()
|
|
}()
|
|
}
|
|
}
|
|
|
|
// RegisterCallback adds a function to the list of callback functions for processing of events.
|
|
func RegisterCallback(cb func(map[string]string)){
|
|
append(callbacks, cb)
|
|
}
|
|
|
|
// process processes each and every unix socket event, Unmarshals the json data and calls the list of callbacks.
|
|
func process() {
|
|
client, err = ls.Accept()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
data := make([]byte, 2048)
|
|
for {
|
|
buf := make([]byte, 512)
|
|
nr, err := client.Read(buf)
|
|
if err != nil && err != io.EOF {
|
|
panic(err)
|
|
}
|
|
append(data, buf[0:nr])
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
}
|
|
env := map[string]string{}
|
|
err := json.Unmarshal(data, &env)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, v = range(callbacks) {
|
|
v(env)
|
|
}
|
|
}
|