opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change quota version to uint32 instead on uint64 (#1517)

Browse Source
This commit is contained in:
mmukhi
2017-09-14 17:40:38 -07:00
committed by GitHub
parent 35170916ff
commit bb78878767
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
transport/control.go
View File

@ -126,7 +126,7 @@ type quotaPool struct {
c chan int
mu sync.Mutex
version uint64
version uint32
quota int
}
@ -183,17 +183,17 @@ func (qb *quotaPool) addAndUpdate(v int) {
// compareAndExecute is processing, this function doesn't
// get executed partially (quota gets updated but the version
// doesn't).
atomic.AddUint64(&(qb.version), 1)
atomic.AddUint32(&(qb.version), 1)
}
func (qb *quotaPool) acquireWithVersion() (<-chan int, uint64) {
return qb.c, atomic.LoadUint64(&(qb.version))
func (qb *quotaPool) acquireWithVersion() (<-chan int, uint32) {
return qb.c, atomic.LoadUint32(&(qb.version))
}
func (qb *quotaPool) compareAndExecute(version uint64, success, failure func()) bool {
func (qb *quotaPool) compareAndExecute(version uint32, success, failure func()) bool {
qb.mu.Lock()
defer qb.mu.Unlock()
if version == atomic.LoadUint64(&(qb.version)) {
if version == atomic.LoadUint32(&(qb.version)) {
success()
return true
}