zhangmeng
2019-12-05 e87bc90683436ee856c1d14ecab1f16940a20471
sdk/sdk.go
@@ -216,23 +216,26 @@
   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 {
            //       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++
            // }
            // 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++
            }
            continue
         } else {
            err = proto.Unmarshal(msg, &repsdkmsg)
@@ -259,6 +262,8 @@
   for {
      select {
      case <-socket.Context.Done():
         socket.Sock.Close()
         logger.Info("socket is close")
         return
      case v, ok := <-in:
@@ -274,19 +279,19 @@
                  logger.Error("failed send:sdkid=", sdkid)
               }
               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
            }