140 lines
2.5 KiB
Go
140 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
// default multicast group used by announced
|
|
MultiCastGroup string = "ff02:0:0:0:0:0:2:1001"
|
|
|
|
// default udp port used by announced
|
|
Port int = 1001
|
|
|
|
// maximum receivable size
|
|
MaxDataGramSize int = 8192
|
|
)
|
|
|
|
type Response struct {
|
|
Address net.UDPAddr
|
|
Raw []byte
|
|
}
|
|
|
|
type Collector struct {
|
|
collectType string
|
|
connection *net.UDPConn // UDP socket
|
|
queue chan *Response // received responses
|
|
}
|
|
|
|
func NewCollector(collectType string) *Collector {
|
|
// Parse address
|
|
addr, err := net.ResolveUDPAddr("udp", "[::]:0")
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
// Open socket
|
|
conn, err := net.ListenUDP("udp", addr)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
conn.SetReadBuffer(MaxDataGramSize)
|
|
|
|
collector := &Collector{
|
|
collectType: collectType,
|
|
connection: conn,
|
|
queue: make(chan *Response, 100),
|
|
}
|
|
|
|
go collector.sendOnce()
|
|
go collector.sender()
|
|
|
|
go collector.receiver()
|
|
go collector.parser()
|
|
|
|
return collector
|
|
}
|
|
|
|
func (coll *Collector) Close() {
|
|
coll.connection.Close()
|
|
close(coll.queue)
|
|
}
|
|
|
|
func (coll *Collector) sendOnce() {
|
|
// TODO
|
|
coll.sendPacket("[2a06:8782:ffbb:1337:c24a:ff:fe2c:c7ac]:1001")
|
|
coll.sendPacket("[2001:bf7:540:0:32b5:c2ff:fe6e:99d5]:1001")
|
|
}
|
|
|
|
func (coll *Collector) sendPacket(address string) {
|
|
addr, err := net.ResolveUDPAddr("udp", address)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
coll.connection.WriteToUDP([]byte(coll.collectType), addr)
|
|
}
|
|
|
|
func (coll *Collector) sender() {
|
|
c := time.Tick(collectInterval)
|
|
|
|
for range c {
|
|
// TODO break condition
|
|
coll.sendOnce()
|
|
}
|
|
}
|
|
|
|
func (coll *Collector) parser() {
|
|
for obj := range coll.queue {
|
|
coll.parse(obj)
|
|
}
|
|
}
|
|
|
|
// Parses a response
|
|
func (coll *Collector) parse(res *Response) {
|
|
var result map[string]interface{}
|
|
json.Unmarshal(res.Raw, &result)
|
|
|
|
nodeId, _ := result["node_id"].(string)
|
|
|
|
if nodeId == "" {
|
|
log.Println("unable to parse node_id")
|
|
return
|
|
}
|
|
|
|
node := nodes.get(nodeId)
|
|
|
|
// Set result
|
|
elem := reflect.ValueOf(node).Elem()
|
|
field := elem.FieldByName(strings.Title(coll.collectType))
|
|
field.Set(reflect.ValueOf(result))
|
|
|
|
nodeserver.SendAll(node)
|
|
}
|
|
|
|
func (coll *Collector) receiver() {
|
|
buf := make([]byte, MaxDataGramSize)
|
|
for {
|
|
n, src, err := coll.connection.ReadFromUDP(buf)
|
|
|
|
if err != nil {
|
|
log.Println("ReadFromUDP failed:", err)
|
|
return
|
|
}
|
|
|
|
raw := make([]byte, n)
|
|
copy(raw, buf)
|
|
|
|
coll.queue <- &Response{
|
|
Address: *src,
|
|
Raw: raw,
|
|
}
|
|
log.Println("received", coll.collectType, "from", src)
|
|
}
|
|
}
|