From 47a19e070f57d92aadf52c62361da6ab5397bd4c Mon Sep 17 00:00:00 2001 From: chenshijun <chenshijun@aiotlink.com> Date: 星期四, 04 六月 2020 16:13:59 +0800 Subject: [PATCH] 将git的shm库变成本地 --- shmqueue.go | 35 ++++++++++++++--------------------- 1 files changed, 14 insertions(+), 21 deletions(-) diff --git a/shmqueue.go b/shmqueue.go index 2cc6908..33b61e5 100644 --- a/shmqueue.go +++ b/shmqueue.go @@ -3,7 +3,7 @@ import ( "context" "fmt" - "github.com/gen2brain/shm" + shm "basic.com/valib/goshm.git" "reflect" "runtime" "sync/atomic" @@ -12,7 +12,7 @@ ) const ( - TimePeriodPutOrGet = time.Duration(1) * time.Microsecond + TimePeriodPutOrGet = time.Duration(5)*time.Millisecond //ms ) //Element info @@ -139,25 +139,21 @@ // Capacity queue 瀹归噺锛屾�诲ぇ灏� func (eqi *EsQueueInfo) QueueCapacity() int { - var capMod uint32 - capMod = atomic.LoadUint32(&eqi.Queue.capMod) - return int(capMod) + return int(eqi.Queue.capMod) } // Size queue 瀹為檯澶у皬 func (eqi *EsQueueInfo) QueueSize() int { - var putPos, getPos, capMod uint32 + var putPos, getPos uint32 var quantity uint32 - capMod = atomic.LoadUint32(&eqi.Queue.capMod) getPos = atomic.LoadUint32(&eqi.Queue.getPos) putPos = atomic.LoadUint32(&eqi.Queue.putPos) if putPos >= getPos { quantity = putPos - getPos } else { - //quantity = q.capMod + (putPos - getPos) - quantity = (capMod + (putPos - getPos)) % capMod + quantity = eqi.Queue.capMod + (putPos - getPos) } return int(quantity) @@ -179,19 +175,17 @@ // Put 鍗曟鍐欏叆锛屽彲鑳藉け璐� func (eqi *EsQueueInfo) Put(val ElemInfo) (bool, int) { - var putPos, putPosNew, getPos, posCnt, capMod ,capacity uint32 + var putPos, putPosNew, getPos, posCnt uint32 var cache *esCache + capMod := eqi.Queue.capMod - capacity = atomic.LoadUint32(&eqi.Queue.capacity) - capMod = atomic.LoadUint32(&eqi.Queue.capMod) getPos = atomic.LoadUint32(&eqi.Queue.getPos) putPos = atomic.LoadUint32(&eqi.Queue.putPos) if putPos >= getPos { posCnt = putPos - getPos } else { - //posCnt = capMod + (putPos - getPos) - posCnt = (capMod + (putPos - getPos)) % capMod + posCnt = capMod + (putPos - getPos) } //todo @@ -215,7 +209,7 @@ putNo := atomic.LoadUint32(&cache.putNo) if putPosNew == putNo && getNo == putNo { cache.value = val - atomic.AddUint32(&cache.putNo, capacity) + atomic.AddUint32(&cache.putNo, eqi.Queue.capacity) return true, int(posCnt + 1) } else { runtime.Gosched() @@ -275,11 +269,10 @@ // Get 鍗曟鑾峰彇锛屽彲鑳藉け璐� func (eqi *EsQueueInfo) Get() (ElemInfo, bool, int) { - var putPos, getPos, getPosNew, posCnt, capMod, capacity uint32 + var putPos, getPos, getPosNew, posCnt uint32 var cache *esCache + capMod := eqi.Queue.capMod - capacity = atomic.LoadUint32(&eqi.Queue.capacity) - capMod = atomic.LoadUint32(&eqi.Queue.capMod) putPos = atomic.LoadUint32(&eqi.Queue.putPos) getPos = atomic.LoadUint32(&eqi.Queue.getPos) @@ -305,10 +298,10 @@ for { getNo := atomic.LoadUint32(&cache.getNo) putNo := atomic.LoadUint32(&cache.putNo) - if getPosNew == getNo && getNo == putNo-capacity { + if getPosNew == getNo && getNo == putNo-eqi.Queue.capacity { val := cache.value - cache.value = ElemInfo{PicId: 0} - atomic.AddUint32(&cache.getNo, capacity) + cache.value = ElemInfo{PicId: 0, InfoId:0} + atomic.AddUint32(&cache.getNo, eqi.Queue.capacity) return val, true, int(posCnt - 1) } else { runtime.Gosched() -- Gitblit v1.8.0