forked from boyska/circolog
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
Go
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()
|
|
}
|
|
}
|
|
|
|
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"))
|
|
})
|
|
}
|
|
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()
|
|
|
|
channel := make(chan format.LogParts)
|
|
handler := syslog.NewChannelHandler(channel)
|
|
|
|
server := syslog.NewServer()
|
|
server.SetFormat(syslog.RFC5424)
|
|
server.SetHandler(handler)
|
|
if *syslogSocketPath != "" {
|
|
server.ListenUnixgram(*syslogSocketPath)
|
|
fmt.Printf("Binding socket `%s` [syslog]\n", *syslogSocketPath)
|
|
} else {
|
|
addr := "127.0.0.1:9514"
|
|
fmt.Printf("Binding address `%s` [syslog]\n", addr)
|
|
server.ListenUDP(addr)
|
|
}
|
|
circbuf = ring.New(*bufsize)
|
|
server.Boot()
|
|
go syslogdHandler(channel)
|
|
|
|
http.HandleFunc("/", httpHandler)
|
|
fmt.Printf("Binding address `%s` [http]\n", ":9080")
|
|
http.ListenAndServe(":9080", nil)
|
|
|
|
server.Wait()
|
|
}
|