86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
// Production build: go build -ldflags="-w" -o gelf-reader_release gelf-reader
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/valyala/fasttemplate"
|
|
"gopkg.in/Graylog2/go-gelf.v2/gelf"
|
|
)
|
|
|
|
const (
|
|
VERSION = "0.2.0"
|
|
)
|
|
|
|
func main() {
|
|
var (
|
|
listenAddr string
|
|
outputFile string
|
|
stemplate string
|
|
showVersion bool
|
|
outWriter io.Writer
|
|
template *fasttemplate.Template
|
|
)
|
|
|
|
flag.StringVar(&listenAddr, "listen", "0.0.0.0:12201", "listen address")
|
|
flag.StringVar(&outputFile, "output", "", "Output file (stdout if empty)")
|
|
flag.StringVar(&stemplate, "template", "{{TimeUnix}} {{Host}}: {{Short}}",
|
|
"Output template")
|
|
flag.BoolVar(&showVersion, "version", false, "Show version and exit")
|
|
flag.Parse()
|
|
|
|
fmt.Printf("GELF reader server %v\n", VERSION)
|
|
if showVersion {
|
|
return
|
|
}
|
|
|
|
template, tplerr := fasttemplate.NewTemplate(stemplate, "{{", "}}")
|
|
if tplerr != nil {
|
|
log.Fatalf("Error parsing template\n%s", tplerr)
|
|
}
|
|
|
|
if len(outputFile) > 0 {
|
|
outFile, err := os.OpenFile(outputFile, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0755)
|
|
if err != nil {
|
|
log.Fatalf("Error opening output file\n%s", err)
|
|
}
|
|
|
|
outWriter = outFile
|
|
defer outFile.Close()
|
|
} else {
|
|
outWriter = os.Stdout
|
|
}
|
|
|
|
gelfReader, err := gelf.NewReader(listenAddr)
|
|
if err != nil {
|
|
log.Fatalf("Error creating GELF network reader\n%s", err)
|
|
}
|
|
|
|
fmt.Printf("Listening on %v\n", gelfReader.Addr())
|
|
|
|
msgData := make(map[string]interface{}, 8)
|
|
|
|
for {
|
|
msg, err := gelfReader.ReadMessage()
|
|
if err != nil {
|
|
log.Printf("Error reading message\n%s\n", err)
|
|
} else {
|
|
msgData["Facility"] = msg.Facility
|
|
msgData["Full"] = msg.Full
|
|
msgData["Host"] = msg.Host
|
|
msgData["Level"] = string(msg.Level)
|
|
msgData["RawExtra"] = string(msg.RawExtra)
|
|
msgData["Short"] = msg.Short
|
|
msgData["TimeUnix"] = fmt.Sprintf("%f", msg.TimeUnix)
|
|
msgData["Version"] = msg.Version
|
|
|
|
template.Execute(outWriter, msgData)
|
|
outWriter.Write([]byte("\n"))
|
|
}
|
|
}
|
|
}
|