From bb78878767b96d411e740439ac820f118e95ae2f Mon Sep 17 00:00:00 2001 From: mmukhi Date: Thu, 14 Sep 2017 17:40:38 -0700 Subject: [PATCH] Change quota version to uint32 instead on uint64 (#1517) --- transport/control.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/transport/control.go b/transport/control.go index 8b5c01e5..77914de1 100644 --- a/transport/control.go +++ b/transport/control.go @@ -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 }