circolog/main.go

package main

import (
	"container/ring"
	"flag"
	"fmt"
	"net/http"
	"os"

	syslog "gopkg.in/mcuadros/go-syslog.v2"
	"gopkg.in/mcuadros/go-syslog.v2/format"
)

func getSyslogdHandler(circbuf *ring.Ring, hub Hub) func(channel syslog.LogPartsChannel) {
	return func(channel syslog.LogPartsChannel) {
		for logParts := range channel {
			hub.logMessages <- logParts
			fmt.Println(logParts)
			circbuf.Value = logParts
			circbuf = circbuf.Next()
		}
	}
}

func main() {
	var err error
	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")
	syslogAddr := flag.String("syslog-addr", "127.0.0.1:9514", "Address:port where to listen for syslog messages")
	queryAddr := flag.String("query-addr", "127.0.0.1:9080", "Address:port where to bind the query service")
	flag.Parse()

	var hub Hub
	hub = NewHub()
	channel := make(chan format.LogParts)
	handler := syslog.NewChannelHandler(channel)

	server := syslog.NewServer()
	server.SetFormat(syslog.RFC5424)
	server.SetHandler(handler)
	if *syslogSocketPath != "" {
		if err = server.ListenUnixgram(*syslogSocketPath); err != nil {
			fmt.Fprintln(os.Stderr, "argh", err)
			os.Exit(1)
		}
		fmt.Printf("Binding socket `%s` [syslog]\n", *syslogSocketPath)
	} else {
		fmt.Printf("Binding address `%s` [syslog]\n", *syslogAddr)
		if err = server.ListenUDP(*syslogAddr); err != nil {
			fmt.Fprintln(os.Stderr, "argh", err)
			os.Exit(1)
		}
	}
	circbuf := ring.New(*bufsize)
	if err = server.Boot(); err != nil {
		fmt.Fprintln(os.Stderr, "argh", err)
		os.Exit(1)
	}
	go hub.Run()
	go getSyslogdHandler(circbuf, hub)(channel)

	setupHttp(circbuf, hub)
	fmt.Printf("Binding address `%s` [http]\n", *queryAddr)
	http.ListenAndServe(*queryAddr, nil)
	server.Wait()
}