fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-04 23:40:26 +08:00
Code Activity

fix(tcp): close recv on io.EOF

Browse Source
This commit is contained in:
源文雨
2025-02-21 16:44:58 +09:00
parent 956199cf19
commit fae1b768f2
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
gold/p2p/tcp/tcp.go
View File

@@ -230,10 +230,7 @@ func (conn *Conn) receive(tcpconn *net.TCPConn, hasvalidated bool) {
}
r.conn = tcpconn
stopch := make(chan struct{})
t := time.AfterFunc(peerstimeout, func() {
stopch <- struct{}{}
})
t := time.NewTimer(peerstimeout)
var err error
copych := make(chan struct{})
@@ -243,7 +240,7 @@ func (conn *Conn) receive(tcpconn *net.TCPConn, hasvalidated bool) {
}()
select {
case <-stopch:
case <-t.C:
if config.ShowDebugLog {
logrus.Debugln("[tcp] recv from", ep, "timeout")
}
@@ -261,7 +258,7 @@ func (conn *Conn) receive(tcpconn *net.TCPConn, hasvalidated bool) {
if config.ShowDebugLog {
logrus.Debugln("[tcp] recv from", ep, "err:", err)
}
if errors.Is(err, net.ErrClosed) || errors.Is(err, io.ErrClosedPipe) {
if errors.Is(err, net.ErrClosed) || errors.Is(err, io.ErrClosedPipe) || errors.Is(err, io.EOF) {
_ = tcpconn.CloseRead()
return
}