123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- package main
- import (
- "flag"
- "fmt"
- "log"
- "net"
- "net/url"
- "os"
- "os/signal"
- "strconv"
- "time"
- "git.lattuga.net/boyska/circolog/data"
- "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/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 format formatter.Format
- format = formatter.FormatSyslog
- flag.Var(&format, "fmt", "Output format [syslog|json]")
- 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; yes/no/auto")
- 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 data.Message
- 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, format, parsed); err != nil {
- log.Println("error printing", err)
- }
- if format == formatter.FormatSyslog { // oops
- 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
- }
- }
- }
|