forked from boyska/circolog
154 lines
3.4 KiB
Go
154 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"time"
|
|
|
|
"git.lattuga.net/boyska/circolog/filtering"
|
|
"git.lattuga.net/boyska/circolog/formatter"
|
|
"github.com/gorilla/websocket"
|
|
isatty "github.com/mattn/go-isatty"
|
|
"github.com/mgutz/ansi"
|
|
"gopkg.in/mcuadros/go-syslog.v2/format"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
type BoolAuto uint
|
|
|
|
const (
|
|
BoolAuto_NO BoolAuto = iota
|
|
BoolAuto_YES BoolAuto = iota
|
|
BoolAuto_AUTO BoolAuto = iota
|
|
)
|
|
|
|
func (b *BoolAuto) String() string {
|
|
switch *b {
|
|
case BoolAuto_NO:
|
|
return "no"
|
|
case BoolAuto_YES:
|
|
return "always"
|
|
case BoolAuto_AUTO:
|
|
return "auto"
|
|
}
|
|
return ""
|
|
}
|
|
func (b *BoolAuto) Set(s string) error {
|
|
switch s {
|
|
case "auto":
|
|
*b = BoolAuto_AUTO
|
|
case "always":
|
|
*b = BoolAuto_YES
|
|
case "no":
|
|
*b = BoolAuto_NO
|
|
default:
|
|
return fmt.Errorf("Invalid value %s", s)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
addr := flag.String("addr", "localhost:9080", "http service address")
|
|
querySocket := flag.String("socket", "", "Path to a unix domain socket for the HTTP server")
|
|
backlogLimit := flag.Int("n", -1, "Limit the backlog length, defaults to no limit (-1)")
|
|
var filter filtering.ExprValue
|
|
flag.Var(&filter, "where", "sql-like query to filter logs")
|
|
// TODO: change to color-mode=auto/no/always
|
|
hasColor := BoolAuto_AUTO
|
|
flag.Var(&hasColor, "color", "dis/enable colors")
|
|
flag.Parse()
|
|
|
|
if hasColor == BoolAuto_NO || (!isatty.IsTerminal(os.Stdout.Fd()) && hasColor != BoolAuto_YES) {
|
|
ansi.DisableColors(true)
|
|
}
|
|
|
|
interrupt := make(chan os.Signal, 1)
|
|
signal.Notify(interrupt, os.Interrupt)
|
|
var d *websocket.Dialer
|
|
u := url.URL{Scheme: "ws",
|
|
Host: *addr, // ignored in case of -socket; see the Dialer below
|
|
Path: "/ws",
|
|
}
|
|
q := u.Query()
|
|
q.Set("fmt", "bson")
|
|
if *backlogLimit >= 0 {
|
|
q.Set("l", strconv.Itoa(*backlogLimit))
|
|
}
|
|
u.RawQuery = q.Encode()
|
|
if *querySocket != "" {
|
|
d = &websocket.Dialer{
|
|
NetDial: func(network, addr string) (net.Conn, error) {
|
|
return net.Dial("unix", *querySocket)
|
|
},
|
|
HandshakeTimeout: 45 * time.Second, // same as DefaultDialer
|
|
}
|
|
|
|
log.Printf("connecting to %s", *querySocket)
|
|
} else {
|
|
d = websocket.DefaultDialer
|
|
log.Printf("connecting to %s", *addr)
|
|
}
|
|
|
|
c, _, err := d.Dial(u.String(), nil)
|
|
if err != nil {
|
|
log.Fatal("dial:", err)
|
|
}
|
|
defer c.Close()
|
|
log.Println("connected!", u.String())
|
|
|
|
done := make(chan struct{})
|
|
|
|
go func() {
|
|
defer close(done)
|
|
for {
|
|
_, serialized, err := c.ReadMessage()
|
|
if err != nil {
|
|
log.Println("close:", err)
|
|
return
|
|
}
|
|
var parsed format.LogParts
|
|
if err := bson.Unmarshal(serialized, &parsed); err != nil {
|
|
log.Println("invalid BSON", err)
|
|
continue
|
|
}
|
|
if !filter.Validate(parsed) {
|
|
continue
|
|
}
|
|
if err := formatter.WriteFormatted(os.Stdout, formatter.FormatSyslog, parsed); err != nil {
|
|
log.Println("error printing", err)
|
|
}
|
|
fmt.Println()
|
|
}
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-done:
|
|
return
|
|
case <-interrupt:
|
|
log.Println("interrupt")
|
|
|
|
// Cleanly close the connection by sending a close message and then waiting (with timeout) for the
|
|
// server to close the connection.
|
|
err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
|
|
if err != nil {
|
|
log.Println("write close:", err)
|
|
return
|
|
}
|
|
select {
|
|
case <-done:
|
|
log.Println("Successfully close")
|
|
case <-time.After(1 * time.Second):
|
|
log.Println("Forced close")
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
}
|