From 60fc03f4494902216a019c57ec42bfe6b457aa6a Mon Sep 17 00:00:00 2001
From: chenshijun <chenshijun@aiotlink.com>
Date: 星期四, 04 六月 2020 16:14:31 +0800
Subject: [PATCH] Merge branch 'master' of ssh://192.168.5.5:29418/valib/shmqueue

---
 shmqueue.go |   26 ++++++++++----------------
 1 files changed, 10 insertions(+), 16 deletions(-)

diff --git a/shmqueue.go b/shmqueue.go
index bf6dcfd..16d8b2e 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(5) * time.Microsecond
+	TimePeriodPutOrGet = time.Duration(5)*time.Millisecond //ms
 )
 
 //Element info
@@ -72,6 +72,9 @@
 	shmLen = shmblocks*uint32(unsafe.Sizeof(datainfo)) + uint32(unsafe.Sizeof(shmstruct))
 
 	data, shmid := CreateRawShm(ctx, int(shmLen), key)
+	if shmid == -1 {
+		return nil
+	}
 	q := bytes2shmEsQueue(data)
 	//init parameters
 	q.capacity = shmblocks
@@ -104,6 +107,9 @@
 
 	var eqi EsQueueInfo
 	data, shmid := AttachRawShm(ctx, key)
+	if shmid == -1 {
+		return nil
+	}
 	shmdata := bytes2shmEsQueue(data)
 	eqi.EsCaches = ptr2esCache(unsafe.Pointer(&shmdata.cache), int(shmdata.capacity))
 	//fmt.Println("AttachQueue EsCaches:", eqi.EsCaches)
@@ -189,8 +195,8 @@
 	}
 
 	//todo
-	if posCnt >= capMod-1 {
-	//if posCnt >= capMod {
+	//if posCnt >= capMod-1 {
+	if posCnt >= capMod {
 		runtime.Gosched()
 		return false, int(posCnt)
 	}
@@ -204,8 +210,6 @@
 
 	cache = &(eqi.EsCaches[putPosNew&capMod])
 
-	//tryMax := 100
-	//tryCount := 0
 	for {
 		getNo := atomic.LoadUint32(&cache.getNo)
 		putNo := atomic.LoadUint32(&cache.putNo)
@@ -216,10 +220,6 @@
 		} else {
 			runtime.Gosched()
 		}
-		//tryCount++
-		//if tryCount >= tryMax {
-		//	return false, int(posCnt)
-		//}
 	}
 }
 
@@ -301,8 +301,6 @@
 
 	cache = &(eqi.EsCaches[getPosNew&capMod])
 
-	//tryMax := 100
-	//tryCount := 0
 	for {
 		getNo := atomic.LoadUint32(&cache.getNo)
 		putNo := atomic.LoadUint32(&cache.putNo)
@@ -314,10 +312,6 @@
 		} else {
 			runtime.Gosched()
 		}
-		//tryCount++
-		//if tryCount >= tryMax {
-		//	return ElemInfo{}, false, int(posCnt)
-		//}
 	}
 }
 

--
Gitblit v1.8.0