lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-09-19 15:47:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dynamically adjust connection timeout

Browse Source 
better fix for https://github.com/chrislusf/seaweedfs/issues/2541
This commit is contained in:
chrislu
2021-12-29 22:44:39 -08:00
parent 5788bf2270
commit fb434318e3
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
weed/util/net_timeout.go
View File

@@ -36,11 +36,13 @@ type Conn struct {
ReadTimeout time.Duration
WriteTimeout time.Duration
isClosed bool
bytesRead int64
bytesWritten int64
}
func (c *Conn) Read(b []byte) (count int, e error) {
if c.ReadTimeout != 0 {
err := c.Conn.SetReadDeadline(time.Now().Add(c.ReadTimeout))
err := c.Conn.SetReadDeadline(time.Now().Add(c.ReadTimeout * time.Duration(c.bytesRead/40000+1)))
if err != nil {
return 0, err
}
@@ -48,6 +50,7 @@ func (c *Conn) Read(b []byte) (count int, e error) {
count, e = c.Conn.Read(b)
if e == nil {
stats.BytesIn(int64(count))
c.bytesRead += int64(count)
}
return
}
@@ -55,7 +58,7 @@ func (c *Conn) Read(b []byte) (count int, e error) {
func (c *Conn) Write(b []byte) (count int, e error) {
if c.WriteTimeout != 0 {
// minimum 4KB/s
err := c.Conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout * time.Duration(len(b)/40000+1)))
err := c.Conn.SetWriteDeadline(time.Now().Add(c.WriteTimeout * time.Duration(c.bytesWritten/40000+1)))
if err != nil {
return 0, err
}
@@ -63,6 +66,7 @@ func (c *Conn) Write(b []byte) (count int, e error) {
count, e = c.Conn.Write(b)
if e == nil {
stats.BytesOut(int64(count))
c.bytesWritten += int64(count)
}
return
}