forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-07-28 07:36:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fixes go test

Browse source
This commit is contained in:
rafaelss 2024-11-06 10:39:49 -03:00 committed by Chen Kai
parent 77e6dc181c
commit 80e20e6386
1 changed files with 11 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
p2p/discover/portal_protocol.go
View file

@ -1151,27 +1151,25 @@ func (p *PortalProtocol) handleFindContent(id enode.ID, addr *net.UDPAddr, reque
var conn *utp.Conn
var connectCtx context.Context
var cancel context.CancelFunc
defer func(clsConn *utp.Conn) {
defer func() {
p.connIdGen.Remove(connId)
if clsConn == nil {
if conn == nil {
return
}
err := clsConn.Close()
err := conn.Close()
if err != nil {
p.Log.Error("failed to close connection", "err", err)
}
}(conn)
connectCtx, cancel = context.WithTimeout(bctx, defaultUTPConnectTimeout)
defer func(cancelFunc context.CancelFunc) {
cancelFunc()
}(cancel)
}()
for {
select {
case <-bctx.Done():
return
default:
p.Log.Debug("will accept find content conn from: ", "source", addr, "connId", connId)
connectCtx, cancel = context.WithTimeout(bctx, defaultUTPConnectTimeout)
conn, err = p.utp.AcceptUTPContext(connectCtx, connectionId.SendId())
cancel()
if err != nil {
if metrics.Enabled {
p.portalMetrics.utpOutFailConn.Inc(1)
@ -1285,26 +1283,23 @@ func (p *PortalProtocol) handleOffer(id enode.ID, addr *net.UDPAddr, request *po
var conn *utp.Conn
var connectCtx context.Context
var cancel context.CancelFunc
defer func(clsConn *utp.Conn) {
defer func() {
p.connIdGen.Remove(connId)
if clsConn == nil {
if conn == nil {
return
}
err := clsConn.Close()
err := conn.Close()
if err != nil {
p.Log.Error("failed to close connection", "err", err)
}
}(conn)
connectCtx, cancel = context.WithTimeout(bctx, defaultUTPConnectTimeout)
defer func(cancelFunc context.CancelFunc) {
cancelFunc()
}(cancel)
}()
for {
select {
case <-bctx.Done():
return
default:
p.Log.Debug("will accept offer conn from: ", "source", addr, "connId", connId)
connectCtx, cancel = context.WithTimeout(bctx, defaultUTPConnectTimeout)
conn, err = p.utp.AcceptUTPContext(connectCtx, connectionId.SendId())
cancel()
if err != nil {