circolog/main.go

63 lines
1.4 KiB
Go
Raw Normal View History

2018-08-22 18:47:00 +02:00
package main
import (
"container/ring"
"flag"
"fmt"
"net/http"
syslog "gopkg.in/mcuadros/go-syslog.v2"
"gopkg.in/mcuadros/go-syslog.v2/format"
)
var circbuf *ring.Ring
func syslogdHandler(channel syslog.LogPartsChannel) {
for logParts := range channel {
fmt.Println(logParts)
circbuf.Value = logParts
circbuf = circbuf.Next()
}
}
2018-08-22 20:56:43 +02:00
func httpHandler(w http.ResponseWriter, r *http.Request) {
circbuf.Do(func(x interface{}) {
if x == nil {
return
}
logmsg := x.(format.LogParts)
if logmsg["message"] == nil {
return
}
c := logmsg["message"].(string)
w.Write([]byte(c))
w.Write([]byte("\n"))
})
}
2018-08-22 18:47:00 +02:00
func main() {
syslogSocketPath := flag.String("syslogd-socket", "", "The socket to listen to syslog addresses")
// dumpSocketPath := flag.String("dump-socket", "/run/buffer.sock", "The socket that user will connect to in order to receive logs")
bufsize := flag.Int("buffer-size", 1000, "Number of messages to keep")
flag.Parse()
2018-08-22 20:56:43 +02:00
channel := make(chan format.LogParts)
2018-08-22 18:47:00 +02:00
handler := syslog.NewChannelHandler(channel)
server := syslog.NewServer()
server.SetFormat(syslog.RFC5424)
server.SetHandler(handler)
if *syslogSocketPath != "" {
server.ListenUnixgram(*syslogSocketPath)
2018-08-22 20:56:43 +02:00
} else {
server.ListenUDP("127.0.0.1:9514")
2018-08-22 18:47:00 +02:00
}
circbuf = ring.New(*bufsize)
server.Boot()
go syslogdHandler(channel)
2018-08-22 20:56:43 +02:00
http.HandleFunc("/", httpHandler)
2018-08-22 18:47:00 +02:00
http.ListenAndServe(":9080", nil)
server.Wait()
}