From 93c3c5e031eb1ca0960c2cf69d81423875b553c7 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期一, 16 十二月 2019 17:46:00 +0800
Subject: [PATCH] update

---
 camera/camera.go |   72 +++++++++++++++++++++++++++++-------
 1 files changed, 58 insertions(+), 14 deletions(-)

diff --git a/camera/camera.go b/camera/camera.go
index 81bc720..3532af1 100644
--- a/camera/camera.go
+++ b/camera/camera.go
@@ -1,6 +1,7 @@
 package camera
 
 import (
+	"strings"
 	"taskpubsub/sdk"
 	"taskpubsub/util"
 
@@ -43,10 +44,20 @@
 		createCameraRecvServerAndListen(cam.Id)
 	}
 
+	//鏂囦欢鍒濆鍖�
+	for _, f := range util.FileArr {
+		logger.Info(f)
+	}
+	for _, f := range util.FileArr {
+		createCameraRecvServerAndListen(f.Id)
+	}
+
 	// 鎵嬪姩杈撳叆鐨勪富棰�
 	initInnerTopic()
 
 	go autoUpdateCamera(util.Cameraflag)
+
+	go autoUpdateFiles(util.Fileflag)
 }
 
 // camera 鎺ュ彈鏁版嵁
@@ -89,7 +100,9 @@
 		var oldcameras []string
 
 		for key := range SocketManage {
-			oldcameras = append(oldcameras, key)
+			if !strings.HasPrefix(key, util.File_Video_Id_Pre) && !strings.HasPrefix(key, util.File_Audio_Id_Pre) && !strings.HasPrefix(key, util.File_Img_Id_Pre) {
+				oldcameras = append(oldcameras, key)
+			}
 		}
 
 		var newcameras []string
@@ -115,6 +128,37 @@
 	}
 }
 
+//鍔ㄦ�佸鐞嗘枃浠�
+func autoUpdateFiles(fileflag chan bool) {
+
+	for _ = range fileflag {
+		logger.Info("do auto del fileflag")
+		var oldIds []string
+
+		for key := range SocketManage {
+			if strings.HasPrefix(key, util.File_Video_Id_Pre) || strings.HasPrefix(key, util.File_Audio_Id_Pre) || strings.HasPrefix(key, util.File_Img_Id_Pre) {
+				oldIds = append(oldIds, key)
+			}
+		}
+
+		var newIds []string
+		for _, f := range util.FileArr {
+			newIds = append(newIds, f.Id)
+		}
+
+		fileListUpdate := util.Difference(oldIds, newIds)
+		logger.Info(fileListUpdate)
+
+		for key, op := range fileListUpdate {
+			if op == "add" {
+				createCameraRecvServerAndListen(key)
+			} else {
+				deleteCameraRecvServer(key)
+			}
+		}
+	}
+}
+
 func Recv(socket util.SocketContext) {
 
 	tryCount := 0
@@ -125,24 +169,25 @@
 	for {
 		select {
 		case <-socket.Context.Done():
+			socket.Sock.Close()
 			logger.Error("listen recv quit")
 			return
 		default:
 			if recvmessage, err = socket.Sock.Recv(); err != nil {
 
-				tryCount++
-				socket = util.MaybeRestartSocket(socket, &tryCount)
+				// tryCount++
+				// socket = util.MaybeRestartSocket(socket, &tryCount)
 
-				// if socket.UseSHM {
-				// 	if tryCount > util.SHMMaxTryCount {
-				// 		socket.Sock.Close()
-				// 		socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
-				// 		logger.Info("SDK SEND SHM TRY :", util.SHMMaxTryCount, " RESTART IT")
-				// 		tryCount = 0
-				// 		continue
-				// 	}
-				// 	tryCount++
-				// }
+				if socket.UseSHM {
+					if tryCount > util.ShmMaxTryCount {
+						socket.Sock.Close()
+						socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
+						logger.Info("SDK SEND SHM TRY :", util.ShmMaxTryCount, " RESTART IT")
+						tryCount = 0
+						continue
+					}
+					tryCount++
+				}
 				continue
 			}
 
@@ -162,7 +207,6 @@
 			} else {
 				taskIDs := GetAllTaskByID(imagemsg.Cid)
 				for _, taskID := range taskIDs {
-					logger.Debug("id: ", imagemsg.Cid, " taskid: ", taskID)
 					doTaskList(imagemsg.Cid, "", taskID, recvmessage)
 				}
 			}

--
Gitblit v1.8.0