1
0
mirror of https://github.com/golang/go synced 2024-11-22 03:04:41 -07:00

crypto/tls: shrink tls Conn's rawInput buffer when Read timeout

This will reduce the rawInput buffer memory cost for large number of idle tls connections
This commit is contained in:
Xargin 2021-09-08 00:22:00 +08:00
parent 6226020c2f
commit 51de0f831b

View File

@ -797,6 +797,14 @@ func (c *Conn) readFromUntil(r io.Reader, n int) error {
// "predict" closeNotify alerts.
c.rawInput.Grow(needs + bytes.MinRead)
_, err := c.rawInput.ReadFrom(&atLeastReader{r, int64(needs)})
// Read timeout, and we do not get any data, connection is idle,
// we should replace rawInput buffer with a small one
if e, ok := err.(net.Error); ok && e.Timeout() &&
c.rawInput.Len() == 0 && c.rawInput.Cap() > 4*bytes.MinRead {
c.rawInput = *bytes.NewBuffer(make([]byte, 0, bytes.MinRead))
}
return err
}