Skip to content

Commit e413e32

Browse files
committed
add debug output when verbose > 0
1 parent 00b0014 commit e413e32

File tree

2 files changed

+16
-7
lines changed

2 files changed

+16
-7
lines changed

src/github.com/kentik/libkflow/send.go

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@ func (s *Sender) Start(agg *agg.Agg, client *api.Client, device *api.Device, n i
5151
}
5252
go s.monitor()
5353

54+
s.debug("sender started with %d workers", n)
55+
5456
return nil
5557
}
5658

@@ -59,6 +61,7 @@ func (s *Sender) Segment() *capnp.Segment {
5961
}
6062

6163
func (s *Sender) Send(flow *chf.CHF) {
64+
s.debug("sending flow to aggregator")
6265
s.Agg.Add(flow)
6366
}
6467

@@ -78,20 +81,20 @@ func (s *Sender) dispatch() {
7881
url := s.URL.String()
7982

8083
for msg := range s.Agg.Output() {
84+
s.debug("dispatching aggregated flow")
85+
8186
buf.Reset()
8287
buf.Write(cid[:])
8388

8489
err := capnp.NewPackedEncoder(buf).Encode(msg)
8590
if err != nil {
86-
// FIXME: check verbosity
87-
log.Print("NewPackedEncoder", err)
91+
log.Print(err)
8892
continue
8993
}
9094

9195
err = s.Client.SendFlow(url, buf)
9296
if err != nil {
93-
// FIXME: check verbosity
94-
log.Print("HTTP", err)
97+
log.Print(err)
9598
continue
9699
}
97100
}
@@ -102,12 +105,18 @@ func (s *Sender) monitor() {
102105
for {
103106
select {
104107
case err := <-s.Agg.Errors():
105-
// FIXME: check verbosity
106-
log.Print("agg error", err)
108+
log.Print(err)
107109
case <-s.Agg.Done():
108110
s.workers.Wait()
109111
s.Exit <- struct{}{}
112+
s.debug("sender stopped")
110113
return
111114
}
112115
}
113116
}
117+
118+
func (s *Sender) debug(fmt string, v ...interface{}) {
119+
if s.Verbose > 0 {
120+
log.Printf(fmt, v...)
121+
}
122+
}

src/github.com/kentik/libkflow/send_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ func setup(t *testing.T) (*Sender, *test.Server, *assert.Assertions) {
6161
server.Log.SetOutput(ioutil.Discard)
6262

6363
url, _ := url.Parse(server.URL() + "/chf")
64-
sender := NewSender(url, 1*time.Second, 1)
64+
sender := NewSender(url, 1*time.Second, 0)
6565
sender.Start(agg, client, device, 1)
6666

6767
return sender, server, assert.New(t)

0 commit comments

Comments
 (0)