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

feat(tcp): add write to peer lock

Browse Source
This commit is contained in:
源文雨
2024-07-17 14:29:15 +09:00
parent 58cb7e09a8
commit 06853c6552
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -6,6 +6,7 @@ import (
"net" "net"
"reflect" "reflect"
"strconv" "strconv"
"sync"
"time" "time"
"github.com/FloatTech/ttl" "github.com/FloatTech/ttl"
@@ -71,6 +72,7 @@ func (ep *EndPoint) Listen() (p2p.Conn, error) {
}, nil, }, nil,
}), }),
recv: make(chan *connrecv, chansz), recv: make(chan *connrecv, chansz),
cplk: &sync.Mutex{},
} }
go conn.accept() go conn.accept()
return conn, nil return conn, nil
@@ -88,6 +90,7 @@ type Conn struct {
lstn *net.TCPListener lstn *net.TCPListener
peers *ttl.Cache[string, *net.TCPConn] peers *ttl.Cache[string, *net.TCPConn]
recv chan *connrecv recv chan *connrecv
cplk *sync.Mutex
} }
func (conn *Conn) accept() { func (conn *Conn) accept() {
@@ -236,6 +239,8 @@ func (conn *Conn) WriteToPeer(b []byte, ep p2p.EndPoint) (n int, err error) {
return 0, errors.New("data size " + strconv.Itoa(blen) + " is too large") return 0, errors.New("data size " + strconv.Itoa(blen) + " is too large")
} }
retried := false retried := false
conn.cplk.Lock()
defer conn.cplk.Unlock()
tcpconn := conn.peers.Get(tcpep.String()) tcpconn := conn.peers.Get(tcpep.String())
RECONNECT: RECONNECT:
if tcpconn == nil { if tcpconn == nil {