From e8bfcf3115506b2c4109d912dd1bffc7e8c44d9b Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 27 十一月 2019 16:40:50 +0800
Subject: [PATCH] update

---
 sdk/sdk.go |   46 ++++++++++++++++++++++++----------------------
 1 files changed, 24 insertions(+), 22 deletions(-)

diff --git a/sdk/sdk.go b/sdk/sdk.go
index ac4788e..8e5c723 100644
--- a/sdk/sdk.go
+++ b/sdk/sdk.go
@@ -220,18 +220,19 @@
 			return
 		default:
 			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")
+				tryCount++
+				socket = util.MaybeRestartSocket(socket, &tryCount)
 
-						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
 			} else {
 				err = proto.Unmarshal(msg, &repsdkmsg)
@@ -273,18 +274,19 @@
 						logger.Error("failed send:sdkid=", sdkid)
 					}
 
-					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++
+					socket = util.MaybeRestartSocket(socket, &tryCount)
 
-							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
 				}

--
Gitblit v1.8.0