zhangmeng
2019-12-05 e87bc90683436ee856c1d14ecab1f16940a20471
sdk/sdk.go
@@ -216,18 +216,22 @@
   for {
      select {
      case <-socket.Context.Done():
         socket.Sock.Close()
         logger.Info("socket close")
         return
      default:
         if msg, err := socket.Sock.Recv(); err != nil {
            // tryCount++
            // socket = util.MaybeRestartSocket(socket, &tryCount)
            if socket.UseSHM {
               if tryCount > util.SHMMaxTryCount {
               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")
                  logger.Info("SDK SEND SHM TRY :", util.ShmMaxTryCount, " RESTART IT")
                  tryCount = 0
                  continue
               }
               tryCount++
@@ -258,6 +262,8 @@
   for {
      select {
      case <-socket.Context.Done():
         socket.Sock.Close()
         logger.Info("socket is close")
         return
      case v, ok := <-in:
@@ -273,14 +279,15 @@
                  logger.Error("failed send:sdkid=", sdkid)
               }
               // tryCount++
               // socket = util.MaybeRestartSocket(socket, &tryCount)
               if socket.UseSHM {
                  if tryCount > util.SHMMaxTryCount {
                  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")
                     logger.Info("SDK SEND SHM TRY :", util.ShmMaxTryCount, " RESTART IT")
                     tryCount = 0
                     continue
                  }
                  tryCount++