yanic/respond/collector.go

311 lines
7.1 KiB
Go
Raw Normal View History

2016-02-26 09:28:31 +01:00
package respond
2015-12-29 04:08:03 +01:00
import (
2016-03-19 01:50:23 +01:00
"bytes"
"compress/flate"
2016-03-12 00:59:36 +01:00
"encoding/json"
"fmt"
2015-12-29 04:08:03 +01:00
"log"
"net"
"time"
2017-03-03 16:19:35 +01:00
"github.com/FreifunkBremen/yanic/data"
"github.com/FreifunkBremen/yanic/database"
"github.com/FreifunkBremen/yanic/lib/jsontime"
"github.com/FreifunkBremen/yanic/runtime"
2015-12-29 04:08:03 +01:00
)
// Collector for a specificle respond messages
2015-12-29 04:08:03 +01:00
type Collector struct {
connections []*net.UDPConn // UDP sockets
ifaceToConn map[string]*net.UDPConn // map from interface name to UDP socket
2017-10-26 15:24:04 +02:00
port int
queue chan *Response // received responses
db database.Connection
nodes *runtime.Nodes
sites []string
interval time.Duration // Interval for multicast packets
stop chan interface{}
2015-12-29 04:08:03 +01:00
}
// NewCollector creates a Collector struct
func NewCollector(db database.Connection, nodes *runtime.Nodes, sites []string, ifaces []string, port int) *Collector {
coll := &Collector{
db: db,
nodes: nodes,
sites: sites,
2017-10-26 15:24:04 +02:00
port: port,
queue: make(chan *Response, 400),
stop: make(chan interface{}),
ifaceToConn: make(map[string]*net.UDPConn),
}
for _, iface := range ifaces {
coll.listenUDP(iface)
}
go coll.parser()
if coll.db != nil {
go coll.globalStatsWorker()
}
return coll
}
func (coll *Collector) listenUDP(iface string) {
if _, found := coll.ifaceToConn[iface]; found {
log.Panicf("can not listen twice on %s", iface)
}
linkLocalAddr, err := getLinkLocalAddr(iface)
2015-12-29 04:08:03 +01:00
if err != nil {
log.Panic(err)
}
// Open socket
conn, err := net.ListenUDP("udp", &net.UDPAddr{
IP: linkLocalAddr,
2017-10-26 15:24:04 +02:00
Port: coll.port,
Zone: iface,
})
2015-12-29 04:08:03 +01:00
if err != nil {
log.Panic(err)
}
conn.SetReadBuffer(maxDataGramSize)
2015-12-29 04:08:03 +01:00
coll.ifaceToConn[iface] = conn
coll.connections = append(coll.connections, conn)
2016-03-12 01:04:22 +01:00
// Start receiver
go coll.receiver(conn)
}
// Returns the first link local unicast address for the given interface name
func getLinkLocalAddr(ifname string) (net.IP, error) {
iface, err := net.InterfaceByName(ifname)
if err != nil {
return nil, err
}
addresses, err := iface.Addrs()
if err != nil {
return nil, err
}
for _, addr := range addresses {
if ipnet := addr.(*net.IPNet); ipnet.IP.IsLinkLocalUnicast() {
return ipnet.IP, nil
}
}
return nil, fmt.Errorf("unable to find link local unicast address for %s", ifname)
}
// Start Collector
func (coll *Collector) Start(interval time.Duration) {
2017-01-29 22:26:16 +01:00
if coll.interval != 0 {
panic("already started")
}
2017-01-29 22:26:16 +01:00
if interval <= 0 {
panic("invalid collector interval")
}
coll.interval = interval
2016-03-12 18:26:51 +01:00
go func() {
coll.sendOnce() // immediately
coll.sender() // periodically
2016-03-12 18:26:51 +01:00
}()
2015-12-29 04:08:03 +01:00
}
2016-03-11 23:56:23 +01:00
// Close Collector
2015-12-29 04:08:03 +01:00
func (coll *Collector) Close() {
2017-01-29 22:26:16 +01:00
close(coll.stop)
for _, conn := range coll.connections {
conn.Close()
}
2015-12-29 04:08:03 +01:00
close(coll.queue)
}
2015-12-29 14:05:47 +01:00
func (coll *Collector) sendOnce() {
now := jsontime.Now()
coll.sendMulticast()
// Wait for the multicast responses to be processed and send unicasts
time.Sleep(coll.interval / 2)
coll.sendUnicasts(now)
2015-12-29 14:05:47 +01:00
}
func (coll *Collector) sendMulticast() {
log.Println("sending multicasts")
for _, conn := range coll.connections {
coll.sendPacket(conn, multiCastGroup)
}
}
// Send unicast packets to nodes that did not answer the multicast
func (coll *Collector) sendUnicasts(seenBefore jsontime.Time) {
seenAfter := seenBefore.Add(-time.Minute * 10)
// Select online nodes that has not been seen recently
nodes := coll.nodes.Select(func(n *runtime.Node) bool {
return n.Lastseen.After(seenAfter) && n.Lastseen.Before(seenBefore) && n.Address != nil
})
// Send unicast packets
log.Printf("sending unicast to %d nodes", len(nodes))
for _, node := range nodes {
conn := coll.ifaceToConn[node.Address.Zone]
if conn == nil {
log.Printf("unable to find connection for %s", node.Address.Zone)
continue
}
coll.sendPacket(conn, node.Address.IP)
time.Sleep(10 * time.Millisecond)
}
}
// SendPacket sends a UDP request to the given unicast or multicast address on the first UDP socket
func (coll *Collector) SendPacket(destination net.IP) {
coll.sendPacket(coll.connections[0], destination)
}
// sendPacket sends a UDP request to the given unicast or multicast address on the given UDP socket
func (coll *Collector) sendPacket(conn *net.UDPConn, destination net.IP) {
addr := net.UDPAddr{
IP: destination,
Port: port,
Zone: conn.LocalAddr().(*net.UDPAddr).Zone,
2016-02-19 11:13:30 +01:00
}
2015-12-29 14:05:47 +01:00
if _, err := conn.WriteToUDP([]byte("GET nodeinfo statistics neighbours"), &addr); err != nil {
2016-03-07 01:37:07 +01:00
log.Println("WriteToUDP failed:", err)
}
2015-12-29 04:08:03 +01:00
}
2016-03-11 23:56:23 +01:00
// send packets continously
func (coll *Collector) sender() {
2017-01-29 22:26:16 +01:00
ticker := time.NewTicker(coll.interval)
2016-03-11 23:56:23 +01:00
for {
select {
case <-coll.stop:
2017-01-29 22:26:16 +01:00
ticker.Stop()
2016-03-11 23:56:23 +01:00
return
2017-01-29 22:26:16 +01:00
case <-ticker.C:
2016-10-04 01:05:18 +02:00
// send the multicast packet to request per-node statistics
2016-03-11 23:56:23 +01:00
coll.sendOnce()
}
2015-12-29 04:08:03 +01:00
}
}
func (coll *Collector) parser() {
2016-01-04 02:07:09 +01:00
for obj := range coll.queue {
2016-10-03 19:55:37 +02:00
if data, err := obj.parse(); err != nil {
log.Println("unable to decode response from", obj.Address.String(), err, "\n", string(obj.Raw))
2016-10-03 19:55:37 +02:00
} else {
coll.saveResponse(obj.Address, data)
2016-03-12 00:59:36 +01:00
}
2015-12-29 04:08:03 +01:00
}
}
2016-10-03 19:55:37 +02:00
func (res *Response) parse() (*data.ResponseData, error) {
2016-03-19 23:27:07 +01:00
// Deflate
2016-10-03 19:55:37 +02:00
deflater := flate.NewReader(bytes.NewReader(res.Raw))
2016-03-19 23:27:07 +01:00
defer deflater.Close()
2016-03-19 15:07:44 +01:00
2016-03-19 23:27:07 +01:00
// Unmarshal
2016-10-03 19:55:37 +02:00
rdata := &data.ResponseData{}
err := json.NewDecoder(deflater).Decode(rdata)
return rdata, err
}
func (coll *Collector) saveResponse(addr *net.UDPAddr, res *data.ResponseData) {
2016-10-03 19:55:37 +02:00
// Search for NodeID
var nodeID string
2016-10-03 19:55:37 +02:00
if val := res.NodeInfo; val != nil {
nodeID = val.NodeID
2016-10-03 19:55:37 +02:00
} else if val := res.Neighbours; val != nil {
nodeID = val.NodeID
2016-10-03 19:55:37 +02:00
} else if val := res.Statistics; val != nil {
nodeID = val.NodeID
2016-10-03 19:55:37 +02:00
}
2017-01-29 22:26:16 +01:00
// Check length of nodeID
if len(nodeID) != 12 {
log.Printf("invalid NodeID '%s' from %s", nodeID, addr.String())
2016-10-03 19:55:37 +02:00
return
2016-03-19 15:07:44 +01:00
}
2017-01-29 22:26:16 +01:00
// Set fields to nil if nodeID is inconsistent
if res.Statistics != nil && res.Statistics.NodeID != nodeID {
res.Statistics = nil
}
if res.Neighbours != nil && res.Neighbours.NodeID != nodeID {
res.Neighbours = nil
}
if res.NodeInfo != nil && res.NodeInfo.NodeID != nodeID {
res.NodeInfo = nil
}
// Process the data and update IP address
node := coll.nodes.Update(nodeID, res)
node.Address = addr
2016-03-19 15:07:44 +01:00
2017-04-18 01:48:38 +02:00
// Store statistics in database
if db := coll.db; db != nil {
db.InsertNode(node)
// Store link data
if neighbours := node.Neighbours; neighbours != nil {
coll.nodes.RLock()
for _, link := range coll.nodes.NodeLinks(node) {
db.InsertLink(&link, node.Lastseen.GetTime())
}
coll.nodes.RUnlock()
}
2016-10-03 19:55:37 +02:00
}
2016-03-19 01:50:23 +01:00
}
func (coll *Collector) receiver(conn *net.UDPConn) {
buf := make([]byte, maxDataGramSize)
2015-12-29 04:08:03 +01:00
for {
n, src, err := conn.ReadFromUDP(buf)
2015-12-29 04:08:03 +01:00
if err != nil {
log.Println("ReadFromUDP failed:", err)
2015-12-29 14:05:47 +01:00
return
2015-12-29 04:08:03 +01:00
}
2016-01-04 02:07:09 +01:00
raw := make([]byte, n)
copy(raw, buf)
coll.queue <- &Response{
Address: src,
2016-01-04 02:07:09 +01:00
Raw: raw,
}
2015-12-29 04:08:03 +01:00
}
}
func (coll *Collector) globalStatsWorker() {
ticker := time.NewTicker(time.Minute)
for {
select {
case <-coll.stop:
2017-01-29 22:26:16 +01:00
ticker.Stop()
return
case <-ticker.C:
coll.saveGlobalStats()
}
}
}
// saves global statistics
func (coll *Collector) saveGlobalStats() {
stats := runtime.NewGlobalStats(coll.nodes, coll.sites)
for site, stat := range stats {
coll.db.InsertGlobals(stat, time.Now(), site)
}
}