consul: shrink yamux recv buffer on idle streams

This commit is contained in:
Armon Dadgar 2015-11-27 17:20:57 -08:00
parent 1f57584d39
commit 847343d801
1 changed files with 6 additions and 0 deletions

View File

@ -103,6 +103,12 @@ func (c *Conn) returnClient(client *StreamClient) {
if c.clients.Len() < c.pool.maxStreams && atomic.LoadInt32(&c.shouldClose) == 0 { if c.clients.Len() < c.pool.maxStreams && atomic.LoadInt32(&c.shouldClose) == 0 {
c.clients.PushFront(client) c.clients.PushFront(client)
didSave = true didSave = true
// If this is a Yamux stream, shrink the internal buffers so that
// we can GC the idle memory
if ys, ok := client.stream.(*yamux.Stream); ok {
ys.Shrink()
}
} }
c.clientLock.Unlock() c.clientLock.Unlock()
if !didSave { if !didSave {