103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package command
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/hashicorp/logutils"
|
|
"github.com/mitchellh/cli"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
// MonitorCommand is a Command implementation that queries a running
|
|
// Consul agent what members are part of the cluster currently.
|
|
type MonitorCommand struct {
|
|
ShutdownCh <-chan struct{}
|
|
Ui cli.Ui
|
|
|
|
lock sync.Mutex
|
|
quitting bool
|
|
}
|
|
|
|
func (c *MonitorCommand) Help() string {
|
|
helpText := `
|
|
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.
|
|
|
|
Options:
|
|
|
|
-log-level=info Log level of the agent.
|
|
-rpc-addr=127.0.0.1:8400 RPC address of the Consul agent.
|
|
`
|
|
return strings.TrimSpace(helpText)
|
|
}
|
|
|
|
func (c *MonitorCommand) Run(args []string) int {
|
|
var logLevel string
|
|
cmdFlags := flag.NewFlagSet("monitor", flag.ContinueOnError)
|
|
cmdFlags.Usage = func() { c.Ui.Output(c.Help()) }
|
|
cmdFlags.StringVar(&logLevel, "log-level", "INFO", "log level")
|
|
rpcAddr := RPCAddrFlag(cmdFlags)
|
|
if err := cmdFlags.Parse(args); err != nil {
|
|
return 1
|
|
}
|
|
|
|
client, err := RPCClient(*rpcAddr)
|
|
if err != nil {
|
|
c.Ui.Error(fmt.Sprintf("Error connecting to Consul agent: %s", err))
|
|
return 1
|
|
}
|
|
defer client.Close()
|
|
|
|
logCh := make(chan string, 1024)
|
|
monHandle, err := client.Monitor(logutils.LogLevel(logLevel), logCh)
|
|
if err != nil {
|
|
c.Ui.Error(fmt.Sprintf("Error starting monitor: %s", err))
|
|
return 1
|
|
}
|
|
defer client.Stop(monHandle)
|
|
|
|
eventDoneCh := make(chan struct{})
|
|
go func() {
|
|
defer close(eventDoneCh)
|
|
OUTER:
|
|
for {
|
|
select {
|
|
case log := <-logCh:
|
|
if log == "" {
|
|
break OUTER
|
|
}
|
|
c.Ui.Info(log)
|
|
}
|
|
}
|
|
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
if !c.quitting {
|
|
c.Ui.Info("")
|
|
c.Ui.Output("Remote side ended the monitor! This usually means that the\n" +
|
|
"remote side has exited or crashed.")
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case <-eventDoneCh:
|
|
return 1
|
|
case <-c.ShutdownCh:
|
|
c.lock.Lock()
|
|
c.quitting = true
|
|
c.lock.Unlock()
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func (c *MonitorCommand) Synopsis() string {
|
|
return "Stream logs from a Consul agent"
|
|
}
|