Gilles Soulier
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
package transport
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
|
|
"github.com/user/nanometrics/server/models"
|
|
)
|
|
|
|
func StartUDP(addr string, handler func(*models.AgentMetrics)) error {
|
|
conn, err := net.ListenPacket("udp", addr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Printf("[udp] écoute sur %s", addr)
|
|
go func() {
|
|
buf := make([]byte, 65535)
|
|
for {
|
|
n, src, err := conn.ReadFrom(buf)
|
|
if err != nil {
|
|
log.Printf("[udp] erreur lecture: %v", err)
|
|
continue
|
|
}
|
|
data := make([]byte, n)
|
|
copy(data, buf[:n])
|
|
go processUDP(data, src.String(), handler)
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func processUDP(data []byte, src string, handler func(*models.AgentMetrics)) {
|
|
var m models.AgentMetrics
|
|
if err := json.Unmarshal(data, &m); err != nil {
|
|
preview := ""
|
|
if len(data) > 0 {
|
|
end := len(data)
|
|
if end > 32 {
|
|
end = 32
|
|
}
|
|
preview = fmt.Sprintf(" | src=%s | premiers octets: %x | texte: %q", src, data[:end], data[:end])
|
|
}
|
|
log.Printf("[udp] JSON invalide: %v%s", err, preview)
|
|
return
|
|
}
|
|
if m.Hostname == "" {
|
|
return
|
|
}
|
|
handler(&m)
|
|
}
|