| | |
| | | // es |
| | | SdkMap["es"] = make(chan protomsg.SdkMessage) |
| | | logger.Info("create es channel: ") |
| | | logger.Info("SdkMap:", SdkMap) |
| | | |
| | | go DealEsTopic() |
| | | |
| | | go autoUpdateSdk(util.Sdkflag) |
| | |
| | | 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) { |
| | |
| | | repsdkmsg.Tasklab.Index++ |
| | | //调用计算函数, 分发给下一个主题 |
| | | nexttopic := GetSdkSendTopic(repsdkmsg) |
| | | if "facedetect-sdk-no-track" == nexttopic || "virtual-faceextract-sdk-pull" == nexttopic { |
| | | logger.Info("nexttopic:", nexttopic) |
| | | } |
| | | SdkMap[nexttopic] <- repsdkmsg |
| | | } |
| | | } |
| | |
| | | } |
| | | |
| | | if err := socket.Sock.Send(data); err != nil { |
| | | 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 |