circolog/main.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()
}