shkong/MiraiGo
1
0
Fork 0
mirror of https://github.com/Mrs4s/MiraiGo.git synced 2025-05-04 11:07:40 +08:00
Code Issues Projects Releases Wiki Activity

client/internal/highway: shorten lock time

Browse Source
This commit is contained in:
wdvxdr 2023-02-19 19:59:54 +08:00
parent 4716f69c1a
commit 3b97ce341b
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
client/internal/highway/highway.go
View File

@ -40,11 +40,12 @@ type Session struct {
SigSession []byte
SessionKey []byte
seq int32
addrMu sync.Mutex
idx int
SsoAddr []Addr
seq int32
idleMu sync.Mutex
idleCount int
idle *idle
@ -209,6 +210,14 @@ func (s *Session) connect(addr Addr) (persistConn, error) {
return pc, nil
}
func (s *Session) nextAddr() Addr {
s.addrMu.Lock()
defer s.addrMu.Unlock()
addr := s.SsoAddr[s.idx]
s.idx = (s.idx + 1) % len(s.SsoAddr)
return addr
}
func (s *Session) selectConn() (pc persistConn, err error) {
for { // select from idle pc
pc = s.getIdleConn()
@ -222,13 +231,9 @@ func (s *Session) selectConn() (pc persistConn, err error) {
}
}
s.addrMu.Lock()
defer s.addrMu.Unlock()
try := 0
idx := s.idx
for {
addr := s.SsoAddr[idx]
idx = (idx + 1) % len(s.SsoAddr) // move next
addr := s.nextAddr()
pc, err = s.connect(addr)
if err == nil {
break
@ -238,6 +243,5 @@ func (s *Session) selectConn() (pc persistConn, err error) {
break
}
}
s.idx = idx
return
}