liuxiaolong
2019-08-31 df219a383a31b435f317ba6472cc87929cc052c6
camera/camera.go
@@ -5,10 +5,13 @@
   //"sync"
   "taskpubsub/logger"
   "basic.com/pubsub/protomsg.git"
   "github.com/gogo/protobuf/proto"
   "taskpubsub/sdk"
   "taskpubsub/util"
   "basic.com/pubsub/protomsg.git"
   "github.com/gogo/protobuf/proto"
   // "golang.org/x/sys/unix"
   "fmt"
   //"time"
@@ -21,6 +24,8 @@
var SocketManage = make(map[string]util.SocketContext)
var shm bool = false
var innerRecvTopic = []string{
   "virtual-faceextract-sdk-pull_2", //to web 以图搜图
}
@@ -31,7 +36,9 @@
   }
}
func Init() {
func Init(useShm bool) {
   shm = useShm
   logger.Info("============ camera info ====================")
   for _, cd := range util.CameraIds {
      logger.Info(cd)
@@ -54,7 +61,15 @@
   if _, isExist := SocketManage[id]; !isExist { //不存在
      url := fmt.Sprintf("ipc:///tmp/%s.ipc", id)
      socket, err := util.NewSocketListen(int(deliver.PushPull), url)
      m := deliver.PushPull
      if shm {
         m = deliver.Shm
         url = id
      }
      fmt.Println("CAMERA URL : ", url)
      socket, err := util.NewSocketListen(int(m), url, shm)
      if err != nil {
         logger.Error("create socket error")
         return
@@ -109,6 +124,8 @@
func Recv(socket util.SocketContext) {
   tryCount := 0
   var recvmessage []byte
   var imagemsg protomsg.Image
   var err error
@@ -120,6 +137,19 @@
      default:
         if recvmessage, err = socket.Sock.Recv(); err != nil {
            //logger.Error("[camera] err is: ", err)
            // fmt.Println("CAMERA RECV ERROR: ", 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")
                  tryCount = 0
                  continue
               }
               tryCount++
            }
            continue
         }
@@ -151,6 +181,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