63 lines
1.2 KiB
Go
63 lines
1.2 KiB
Go
|
package journald_json
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"github.com/genofire/logmania/lib"
|
||
|
|
"github.com/genofire/logmania/log"
|
||
|
|
"github.com/genofire/logmania/receive"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Receiver struct {
|
||
|
|
receive.Receiver
|
||
|
|
exportChannel chan *log.Entry
|
||
|
|
serverSocket *net.UDPConn
|
||
|
|
}
|
||
|
|
|
||
|
|
func Init(config *lib.ReceiveConfig, exportChannel chan *log.Entry) receive.Receiver {
|
||
|
|
addr, err := net.ResolveUDPAddr(config.JournaldJSON.Type, config.JournaldJSON.Address)
|
||
|
|
ln, err := net.ListenUDP(config.JournaldJSON.Type, addr)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Error("journald-json init ", err)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
recv := &Receiver{
|
||
|
|
serverSocket: ln,
|
||
|
|
exportChannel: exportChannel,
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Info("journald-json init")
|
||
|
|
|
||
|
|
return recv
|
||
|
|
}
|
||
|
|
|
||
|
|
const maxDataGramSize = 8192
|
||
|
|
|
||
|
|
func (rc *Receiver) Listen() {
|
||
|
|
log.Info("journald-json listen")
|
||
|
|
for {
|
||
|
|
buf := make([]byte, maxDataGramSize)
|
||
|
|
n, src, err := rc.serverSocket.ReadFromUDP(buf)
|
||
|
|
if err != nil {
|
||
|
|
log.Warn("failed to accept connection", err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
raw := make([]byte, n)
|
||
|
|
copy(raw, buf)
|
||
|
|
entry := toLogEntry(raw, src.IP.String())
|
||
|
|
if entry != nil {
|
||
|
|
rc.exportChannel <- entry
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (rc *Receiver) Close() {
|
||
|
|
rc.serverSocket.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
receive.AddReceiver("journald_json", Init)
|
||
|
|
}
|