chenshijun
2019-09-11 a45d5a83c5a283444a407d648432e9dd08f89478
camera/camera.go
@@ -2,19 +2,11 @@
import (
   "basic.com/valib/deliver.git"
   //"sync"
   "taskpubsub/logger"
   "taskpubsub/sdk"
   "taskpubsub/util"
   "basic.com/valib/logger.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/gogo/protobuf/proto"
   // "golang.org/x/sys/unix"
   "fmt"
   //"time"
   "taskpubsub/sdk"
   "taskpubsub/util"
)
const (
@@ -60,7 +52,7 @@
func createCameraRecvServerAndListen(id string) {
   if _, isExist := SocketManage[id]; !isExist { //不存在
      url := fmt.Sprintf("ipc:///tmp/%s.ipc", id)
      url := "ipc:///tmp/" + id + ".ipc"
      m := deliver.PushPull
      if shm {
         m = deliver.Shm
@@ -68,7 +60,7 @@
      }
      fmt.Println("CAMERA URL : ", url)
      logger.Info("CAMERA URL : ", url)
      socket, err := util.NewSocketListen(int(m), url, shm)
      if err != nil {
         logger.Error("create socket error")
@@ -137,10 +129,11 @@
      default:
         if recvmessage, err = socket.Sock.Recv(); err != nil {
            //logger.Error("[camera] err is: ", err)
            if socket.UseSHM {
               if tryCount > util.SHMMaxTryCount {
                  socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
                  fmt.Println("CAMERA SHM TRY :", tryCount, " RESTART IT")
                  logger.Info("CAMERA SHM TRY :", tryCount, " RESTART IT")
                  tryCount = 0
@@ -162,6 +155,7 @@
            continue
         }
         if faceExtractWebCID == imagemsg.Cid { //以图搜图
            logger.Info("id: ", faceExtractWebTaskID)
            doTaskList(imagemsg.Cid, "", faceExtractWebTaskID, recvmessage)
         } else {
            taskIDs := GetAllTaskByID(imagemsg.Cid)
@@ -179,6 +173,9 @@
   for _, camsingle := range util.CameraTasks {
      if cid == camsingle.Camera.Id {
         for _, tasksingle := range camsingle.Tasks {
            if !tasksingle.Enable {
               continue
            }
            tasks = append(tasks, tasksingle.Taskid)
         }
         return
@@ -190,7 +187,7 @@
func doTaskList(cid string, caddr string, taskid string, data []byte) {
   //  数据加工(打标签)
   logger.Debug("taskid: ", taskid, "has ", len(data), "data[]byte")
   //logger.Info("taskid: ", taskid, "has ", len(data), "data[]byte")
   sdkmsg := sdk.ToSdkMsg(cid, caddr, taskid, data)
   if sdkmsg.Tasklab == nil {
      logger.Error(cid, " not have taskid: ", taskid)
@@ -199,11 +196,19 @@
   //  计算分发的主题
   SendTopic := sdk.GetSdkSendTopic(sdkmsg)
   logger.Debug(SendTopic)
   //logger.Info(SendTopic)
   if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
      logger.Info(SendTopic)
   }
   if _, ok := sdk.SdkMap[SendTopic]; ok {
      if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
         logger.Info(SendTopic)
      }
      sdk.SdkMap[SendTopic] <- sdkmsg
      logger.Debug("dispute sendtopic success", SendTopic)
      if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
         logger.Info("dispute sendtopic success", SendTopic)
      }
   } else {
      logger.Debug("分发的主题不存在")
      logger.Info("分发的主题不存在")
   }
}