mirror of
https://github.com/kemko/nomad.git
synced 2026-01-01 16:05:42 +03:00
monitor command takes no args
rm extra new line fix lint errors return after close fix, simplify test
This commit is contained in:
@@ -244,7 +244,6 @@ type MonitorFrame struct {
|
||||
// Monitor returns a channel which will receive streaming logs from the agent
|
||||
// Providing a non-nil stopCh can be used to close the connection and stop log streaming
|
||||
func (a *Agent) Monitor(stopCh <-chan struct{}, q *QueryOptions) (<-chan *StreamFrame, <-chan error) {
|
||||
|
||||
errCh := make(chan error, 1)
|
||||
r, err := a.client.newRequest("GET", "/v1/agent/monitor")
|
||||
if err != nil {
|
||||
|
||||
@@ -1,7 +1,6 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"sort"
|
||||
@@ -320,7 +319,6 @@ func TestAgent_MonitorWithNode(t *testing.T) {
|
||||
require.NoError(t, err)
|
||||
|
||||
// Wait for a log message
|
||||
var result bytes.Buffer
|
||||
OUTER:
|
||||
for {
|
||||
select {
|
||||
|
||||
@@ -29,10 +29,6 @@ func NewAgentEndpoint(c *Client) *Agent {
|
||||
return m
|
||||
}
|
||||
|
||||
type monitorFrame struct {
|
||||
Data []byte `json:",omitempty"`
|
||||
}
|
||||
|
||||
func (m *Agent) monitor(conn io.ReadWriteCloser) {
|
||||
defer metrics.MeasureSince([]string{"client", "monitor", "monitor"}, time.Now())
|
||||
defer conn.Close()
|
||||
|
||||
@@ -22,20 +22,23 @@ func TestMonitor_Start(t *testing.T) {
|
||||
})
|
||||
|
||||
closeCh := make(chan struct{})
|
||||
defer close(closeCh)
|
||||
|
||||
logCh := m.Start(closeCh)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case log := <-logCh:
|
||||
require.Contains(t, string(log), "[DEBUG] test log")
|
||||
case <-time.After(1 * time.Second):
|
||||
t.Fatal("Expected to receive from log channel")
|
||||
}
|
||||
}
|
||||
logger.Debug("test log")
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
}()
|
||||
logger.Debug("test log")
|
||||
|
||||
for {
|
||||
select {
|
||||
case log := <-logCh:
|
||||
require.Contains(t, string(log), "[DEBUG] test log")
|
||||
return
|
||||
case <-time.After(3 * time.Second):
|
||||
t.Fatal("Expected to receive from log channel")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Ensure number of dropped messages are logged
|
||||
@@ -61,7 +64,6 @@ func TestMonitor_DroppedMessages(t *testing.T) {
|
||||
}
|
||||
|
||||
received := ""
|
||||
|
||||
passed := make(chan struct{})
|
||||
go func() {
|
||||
for {
|
||||
|
||||
@@ -74,6 +74,13 @@ func (c *MonitorCommand) Run(args []string) int {
|
||||
return 1
|
||||
}
|
||||
|
||||
args = flags.Args()
|
||||
if l := len(args); l != 0 {
|
||||
c.Ui.Error("This command takes no arguments")
|
||||
c.Ui.Error(commandErrorText(c))
|
||||
return 1
|
||||
}
|
||||
|
||||
client, err := c.Meta.Client()
|
||||
if err != nil {
|
||||
c.Ui.Error(fmt.Sprintf("Error initializing client: %s", err))
|
||||
|
||||
Reference in New Issue
Block a user