From 467d125ef6ac819cc1e8bd1a8b5dcd00d491f624 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 30 十月 2019 09:13:39 +0800
Subject: [PATCH] bug fixed memvirt leak

---
 sdk/sdk.go |   21 ++++++++++++---------
 1 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/sdk/sdk.go b/sdk/sdk.go
index bc0021a..ac4788e 100644
--- a/sdk/sdk.go
+++ b/sdk/sdk.go
@@ -1,9 +1,10 @@
 package sdk
 
 import (
-	"github.com/gogo/protobuf/proto"
 	"taskpubsub/tasktag"
 	"taskpubsub/util"
+
+	"github.com/gogo/protobuf/proto"
 
 	"basic.com/pubsub/protomsg.git"
 	"basic.com/valib/deliver.git"
@@ -30,13 +31,13 @@
 }
 
 func initInnerTopic() {
-	for _, sendTopic := range innerSendTopic {
-		createSdkSendServerAndListen(sendTopic)
-	}
+	// for _, sendTopic := range innerSendTopic {
+	// 	createSdkSendServerAndListen(sendTopic)
+	// }
 
-	for _, recvTopic := range innerRecvTopic {
-		createSdkRecvServerAndListen(recvTopic)
-	}
+	// for _, recvTopic := range innerRecvTopic {
+	// 	createSdkRecvServerAndListen(recvTopic)
+	// }
 }
 
 func Init(useShm bool) {
@@ -195,8 +196,8 @@
 		sendTopic = sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Ipcid
 		if "Yolo" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype &&
 			"FaceDetect" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype &&
-			"FaceCompare" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype{
-		//if sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype == "" {
+			"FaceCompare" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype {
+			//if sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype == "" {
 			logger.Info("----------Sdktype:yitusoutu")
 			logger.Info("鍒嗗彂鐨勪富棰�:", sendTopic, "!Sdktype锛�", sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype)
 		}
@@ -221,6 +222,7 @@
 			if msg, err := socket.Sock.Recv(); err != nil {
 				if socket.UseSHM {
 					if tryCount > util.SHMMaxTryCount {
+						socket.Sock.Close()
 						socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
 						logger.Info("SDK RECV SHM TRY :", tryCount, " RESTART IT")
 
@@ -273,6 +275,7 @@
 
 					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")
 

--
Gitblit v1.8.0