From a735e666cff14e4e485110d2a59d401b4be61b9b Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 29 八月 2019 17:50:09 +0800
Subject: [PATCH] add recover

---
 camera/camera.go |   25 +++++++++++++++++++------
 1 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/camera/camera.go b/camera/camera.go
index 8bc26d0..727e017 100644
--- a/camera/camera.go
+++ b/camera/camera.go
@@ -5,10 +5,11 @@
 	//"sync"
 	"taskpubsub/logger"
 
-	"basic.com/pubsub/protomsg.git"
-	"github.com/gogo/protobuf/proto"
 	"taskpubsub/sdk"
 	"taskpubsub/util"
+
+	"basic.com/pubsub/protomsg.git"
+	"github.com/gogo/protobuf/proto"
 
 	// "golang.org/x/sys/unix"
 
@@ -61,14 +62,13 @@
 
 		url := fmt.Sprintf("ipc:///tmp/%s.ipc", id)
 		m := deliver.PushPull
-		if shm{
+		if shm {
 			m = deliver.Shm
 			url = id
 
-			// unix.Unlink("/dev/shm/" + url)
 		}
-		
-		fmt.Println("ipc url: ", url)
+
+		fmt.Println("CAMERA URL : ", url)
 		socket, err := util.NewSocketListen(int(m), url, shm)
 		if err != nil {
 			logger.Error("create socket error")
@@ -124,6 +124,8 @@
 
 func Recv(socket util.SocketContext) {
 
+	tryCount := 0
+
 	var recvmessage []byte
 	var imagemsg protomsg.Image
 	var err error
@@ -135,6 +137,17 @@
 		default:
 			if recvmessage, err = socket.Sock.Recv(); err != nil {
 				//logger.Error("[camera] err is: ", err)
+				if socket.UseSHM {
+					if tryCount > util.SHMMaxTryCount {
+						socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
+						fmt.Println("CAMERA SHM TRY :", tryCount, " RESTART IT")
+
+						tryCount = 0
+
+						continue
+					}
+					tryCount++
+				}
 				continue
 			}
 

--
Gitblit v1.8.0