diff options
Diffstat (limited to 'internal/clients/baseclient.go')
| -rw-r--r-- | internal/clients/baseclient.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/internal/clients/baseclient.go b/internal/clients/baseclient.go index ba18f95..008a01e 100644 --- a/internal/clients/baseclient.go +++ b/internal/clients/baseclient.go @@ -66,11 +66,11 @@ func (c *baseClient) makeConnections(maker maker) { c.stats = newTailStats(len(c.connections)) } -func (c *baseClient) Start(ctx context.Context) (status int) { +func (c *baseClient) Start(ctx context.Context, statsCh <-chan struct{}) (status int) { // Periodically check for unknown hosts, and ask the user whether to trust them or not. go c.hostKeyCallback.PromptAddHosts(ctx) - // Periodically print out connection stats to the client. - go c.stats.periodicLogStats(ctx, c.throttleCh) + // Print client stats every time something on statsCh is recieved. + go c.stats.Start(ctx, c.throttleCh, statsCh) // Keep count of active connections active := make(chan struct{}, len(c.connections)) |