chenshijun
2019-09-11 9389544d4cc56ef00084e8dbb6afb73e1ec00eca
sdk/sdk.go
@@ -53,6 +53,8 @@
   // es
   SdkMap["es"] = make(chan protomsg.SdkMessage)
   logger.Info("create es channel: ")
   logger.Info("SdkMap:", SdkMap)
   go DealEsTopic()
   go autoUpdateSdk(util.Sdkflag)
@@ -175,11 +177,18 @@
func GetSdkSendTopic(sdkmsg protomsg.SdkMessage) (sendTopic string) {
   if int(sdkmsg.Tasklab.Index) < len(sdkmsg.Tasklab.Sdkinfos) {
      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 == "" {
         logger.Info("----------Sdktype:yitusoutu")
         logger.Info("分发的主题:", sendTopic, "!Sdktype:", sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype)
      }
   } else {
      sendTopic = "es"
   }
   logger.Debug("分发的主题: ", sendTopic, "位置:", int(sdkmsg.Tasklab.Index)+1, "/", len(sdkmsg.Tasklab.Sdkinfos))
   return
   return sendTopic
}
func Recv(socket util.SocketContext) {
@@ -201,6 +210,9 @@
            repsdkmsg.Tasklab.Index++
            //调用计算函数, 分发给下一个主题
            nexttopic := GetSdkSendTopic(repsdkmsg)
            if "facedetect-sdk-no-track" == nexttopic || "virtual-faceextract-sdk-pull" == nexttopic {
               logger.Info("nexttopic:", nexttopic)
            }
            SdkMap[nexttopic] <- repsdkmsg
         }
      }
@@ -223,10 +235,14 @@
            }
            if err := socket.Sock.Send(data); err != nil {
               logger.Error("failed send:sdkid=", sdkid)
               if "facedetect-sdk-no-track" == sdkid || "virtual-faceextract-sdk-pull" == sdkid {
                  logger.Error("failed send:sdkid=", sdkid)
               }
               continue
            }
            logger.Debug(sdkid, " send success: ", len(data))
            if "facedetect-sdk-no-track" == sdkid || "virtual-faceextract-sdk-pull" == sdkid {
               logger.Info(sdkid, " send success: ", len(data))
            }
         } else {
            logger.Debug(sdkid, " 主题关闭, 关闭send()")
            return