github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-10-18 19:23:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix mux send window size counting

Browse Source
This commit is contained in:
ffdfgdfg
2019-12-27 13:02:28 +08:00
parent 4de24ff13a
commit 5e1b0be81c
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
lib/mux/conn.go
View File

@@ -446,7 +446,20 @@ func (Self *SendWindow) allow() {
}
func (Self *SendWindow) sent(sentSize uint32) {
atomic.AddUint64(&Self.remainingWait, ^(uint64(sentSize)<<dequeueBits - 1))
var remaining, wait uint32
for {
ptrs := atomic.LoadUint64(&Self.remainingWait)
remaining, wait = Self.unpack(ptrs)
if remaining >= sentSize {
atomic.AddUint64(&Self.remainingWait, ^(uint64(sentSize)<<dequeueBits - 1))
break
} else {
if atomic.CompareAndSwapUint64(&Self.remainingWait, ptrs, Self.pack(0, wait)) {
// just keep the wait status, it will be wait in the next loop
break
}
}
}
}
func (Self *SendWindow) WriteTo() (p []byte, sendSize uint32, part bool, err error) {