package command
|
|
import (
|
"flag"
|
"fmt"
|
"strings"
|
"sync"
|
|
"github.com/hashicorp/logutils"
|
"github.com/mitchellh/cli"
|
)
|
|
// MonitorCommand is a Command implementation that queries a running
|
// Serf 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: serf monitor [options]
|
|
Shows recent log messages of a Serf 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 Serf 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:7373 RPC address of the Serf agent.
|
-rpc-auth="" RPC auth token of the Serf 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)
|
rpcAuth := RPCAuthFlag(cmdFlags)
|
if err := cmdFlags.Parse(args); err != nil {
|
return 1
|
}
|
|
client, err := RPCClient(*rpcAddr, *rpcAuth)
|
if err != nil {
|
c.Ui.Error(fmt.Sprintf("Error connecting to Serf agent: %s", err))
|
return 1
|
}
|
defer client.Close()
|
|
eventCh := make(chan map[string]interface{}, 1024)
|
streamHandle, err := client.Stream("*", eventCh)
|
if err != nil {
|
c.Ui.Error(fmt.Sprintf("Error starting stream: %s", err))
|
return 1
|
}
|
defer client.Stop(streamHandle)
|
|
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)
|
case event := <-eventCh:
|
if event == nil {
|
break OUTER
|
}
|
c.Ui.Info("Event Info:")
|
for key, val := range event {
|
c.Ui.Info(fmt.Sprintf("\t%s: %#v", key, val))
|
}
|
}
|
}
|
|
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 Serf agent"
|
}
|
