104 lines
2.0 KiB
Go
104 lines
2.0 KiB
Go
package respond
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
//Collector for a specificle respond messages
|
|
type Collector struct {
|
|
CollectType string
|
|
connection *net.UDPConn // UDP socket
|
|
queue chan *Response // received responses
|
|
parse func(coll *Collector, res *Response)
|
|
}
|
|
|
|
//NewCollector creates a Collector struct
|
|
func NewCollector(CollectType string, parseFunc func(coll *Collector, res *Response)) *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, 400),
|
|
parse: parseFunc,
|
|
}
|
|
|
|
go collector.receiver()
|
|
go collector.parser()
|
|
|
|
collector.sendOnce()
|
|
|
|
return collector
|
|
}
|
|
|
|
//Close Collector
|
|
func (coll *Collector) Close() {
|
|
coll.connection.Close()
|
|
close(coll.queue)
|
|
}
|
|
|
|
func (coll *Collector) sendOnce() {
|
|
coll.sendPacket(net.JoinHostPort(multiCastGroup, port))
|
|
log.Println("send request")
|
|
}
|
|
|
|
func (coll *Collector) sendPacket(address string) {
|
|
addr, err := net.ResolveUDPAddr("udp", address)
|
|
if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
if _, err := coll.connection.WriteToUDP([]byte(coll.CollectType), addr); err != nil {
|
|
log.Println("WriteToUDP failed:", err)
|
|
}
|
|
}
|
|
|
|
func (coll *Collector) sender(collectInterval time.Duration) {
|
|
c := time.Tick(collectInterval)
|
|
|
|
for range c {
|
|
// TODO break condition
|
|
coll.sendOnce()
|
|
}
|
|
}
|
|
|
|
func (coll *Collector) parser() {
|
|
for obj := range coll.queue {
|
|
coll.parse(coll, obj)
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|