open-consul/command/monitor/monitor.go

131 lines
2.6 KiB
Go
Raw Normal View History

package monitor
2013-12-31 00:09:39 +00:00
import (
"flag"
2013-12-31 00:09:39 +00:00
"fmt"
"sync"
"github.com/hashicorp/consul/api"
"github.com/hashicorp/consul/command/flags"
"github.com/mitchellh/cli"
2013-12-31 00:09:39 +00:00
)
// cmd is a Command implementation that queries a running
2013-12-31 00:09:39 +00:00
// Consul agent what members are part of the cluster currently.
type cmd struct {
UI cli.Ui
help string
flags *flag.FlagSet
http *flags.HTTPFlags
shutdownCh <-chan struct{}
2013-12-31 00:09:39 +00:00
lock sync.Mutex
quitting bool
// flags
logLevel string
logJSON bool
}
func New(ui cli.Ui, shutdownCh <-chan struct{}) *cmd {
c := &cmd{UI: ui, shutdownCh: shutdownCh}
c.init()
return c
2013-12-31 00:09:39 +00:00
}
func (c *cmd) init() {
c.flags = flag.NewFlagSet("", flag.ContinueOnError)
c.flags.StringVar(&c.logLevel, "log-level", "INFO",
"Log level of the agent.")
c.flags.BoolVar(&c.logJSON, "log-json", false,
"Output logs in JSON format.")
2017-10-18 00:38:26 +00:00
c.http = &flags.HTTPFlags{}
flags.Merge(c.flags, c.http.ClientFlags())
c.help = flags.Usage(help, c.flags)
2013-12-31 00:09:39 +00:00
}
func (c *cmd) Run(args []string) int {
var logCh chan string
var err error
var client *api.Client
if err = c.flags.Parse(args); err != nil {
2013-12-31 00:09:39 +00:00
return 1
}
client, err = c.http.APIClient()
2013-12-31 00:09:39 +00:00
if err != nil {
2017-04-21 00:02:42 +00:00
c.UI.Error(fmt.Sprintf("Error connecting to Consul agent: %s", err))
2013-12-31 00:09:39 +00:00
return 1
}
eventDoneCh := make(chan struct{})
if c.logJSON {
logCh, err = client.Agent().MonitorJSON(c.logLevel, eventDoneCh, nil)
2020-02-11 09:57:50 +00:00
if err != nil {
c.UI.Error(fmt.Sprintf("Error starting JSON monitor: %s", err))
return 1
}
} else {
logCh, err = client.Agent().Monitor(c.logLevel, eventDoneCh, nil)
if err != nil {
c.UI.Error(fmt.Sprintf("Error starting monitor: %s", err))
return 1
}
2013-12-31 00:09:39 +00:00
}
go func() {
defer close(eventDoneCh)
OUTER:
for {
select {
case log := <-logCh:
if log == "" {
break OUTER
}
2017-04-21 00:02:42 +00:00
c.UI.Info(log)
2013-12-31 00:09:39 +00:00
}
}
c.lock.Lock()
defer c.lock.Unlock()
if !c.quitting {
2017-04-21 00:02:42 +00:00
c.UI.Info("")
c.UI.Output("Remote side ended the monitor! This usually means that the\n" +
2013-12-31 00:09:39 +00:00
"remote side has exited or crashed.")
}
}()
select {
case <-eventDoneCh:
return 1
case <-c.shutdownCh:
2013-12-31 00:09:39 +00:00
c.lock.Lock()
c.quitting = true
c.lock.Unlock()
}
return 0
}
func (c *cmd) Synopsis() string {
return synopsis
2013-12-31 00:09:39 +00:00
}
func (c *cmd) Help() string {
return c.help
}
const synopsis = "Stream logs from a Consul agent"
const help = `
Usage: consul monitor [options]
Shows recent log messages of a Consul agent, and attaches to the agent,
outputting log messages as they occur in real time. The monitor lets you
listen for log levels that may be filtered out of the Consul agent. For
example your agent may only be logging at INFO level, but with the monitor
you can see the DEBUG level logs.
`