554325746@qq.com
2019-06-27 6fd3413cfa70086864216c59718f7f4fc58b1acc
camera/camera.go
@@ -7,6 +7,7 @@
   "github.com/long/test/sdk"
   "github.com/long/test/util"
    "github.com/gogo/protobuf/proto"
    "basic.com/pubsub/protomsg.git"
   "context"
@@ -15,10 +16,7 @@
   // "time"
)
//var SocketManage = make(map[string]SocketContext)
var SocketManage sync.Map
var Initchannel = make(chan protomsg.Camera )
type SocketContext struct {
   Sock    deliver.Deliver
@@ -27,39 +25,37 @@
}
func Init() {
   fmt.Println("============ camera info ====================")
   for _, cd := range util.CameraIds {
      fmt.Println(cd)
      fmt.Println()
   }
   go CreateCamera(Initchannel)
   go AutoDelCamera(util.Cameraflag)
    // 摄像机初始化
   for _, cam := range util.CameraIds {
      Initchannel <- cam
        CreateCamera(cam.Id, "camera")
   }
    //  web端初始化
    CreateCamera("virtual-faceextract-sdk-pull_2" , "web")
//   go AutoDelCamera(util.Cameraflag)
}
func CreateCamera(camera chan protomsg.Camera) {
   for cam := range camera {
        camid := cam.Id
        caddr := cam.Addr
      if _, ok := SocketManage.Load(camid); !ok {
         url := fmt.Sprintf("ipc:///tmp/%s.ipc", camid)
// camera 接受数据
func CreateCamera(id string, remote string) {
      if _, ok := SocketManage.Load(id); !ok {
         url := fmt.Sprintf("ipc:///tmp/%s.ipc", id)
         id, socketlisten, err := NewCamerSocketListen(deliver.PushPull, camid, url)
         socketlisten, err := NewCamerSocketListen(deliver.PushPull, id, url)
         if err != nil {
            fmt.Println("create socket error")
            continue
                return
         }
         go func(cid string, cameraaddr string,  sock SocketContext) {
            Recv(cid, cameraaddr, sock)
         }(id, caddr, socketlisten)
          go Recv(socketlisten, remote)
      }
   }
}
//动态处理
@@ -76,22 +72,15 @@
         return true
      })
      var newcamera []string
      for _, value := range util.CameraIds {
         newcamera = append(newcamera, value.Id)
      }
        var newcamera []string
       newcamera = append(newcamera, util.Sdklist...)
      cameraChanDel := util.Difference(oldcamera, newcamera)
      fmt.Println(cameraChanDel)
      for key, op := range cameraChanDel {
         if op == "add" {
                for _, value := range util.CameraIds {
                    if key == value.Id{
                    Initchannel <- value
                    }
                }
               CreateCamera(key, "camera")
         } else {
            if sock, ok := SocketManage.Load(key); ok {
               if socket, sok := sock.(SocketContext); sok {
@@ -106,7 +95,7 @@
}
// create server
func NewCamerSocketListen(mode int, cameraid string, url string) (cid string, socket SocketContext, err error) {
func NewCamerSocketListen(mode int, cameraid string, url string) (socket SocketContext, err error) {
   ctx, cancel := context.WithCancel(context.Background())
   socket.Context = ctx
@@ -116,33 +105,42 @@
   fmt.Println("new socket.Sock: ", socket.Sock)
   if socket.Sock == nil {
      return cameraid, socket, errors.New("create listen error")
      return socket, errors.New("create listen error")
   }
   SocketManage.Store(cameraid, socket)
   return cameraid, socket, nil
   return socket, nil
}
func Recv(cameraid string, caddr string , socket SocketContext) {
   var msg []byte
   var err error
func Recv(socket SocketContext, remote string ) {
    var recvmsg  protomsg.Recvmsg
   for {
      select {
      case <-socket.Context.Done():
         fmt.Println("listen recv quit")
         return
      default:
         if msg, err = socket.Sock.Recv(); err != nil {
            fmt.Println("err is: ", cameraid, err)
         if msg, err := socket.Sock.Recv(); err != nil {
            fmt.Println("err is: ", err)
            continue
         } else {
            fmt.Println()
            fmt.Println("============== one msg input ==========")
            for _, taskid := range GetAlltask(cameraid) {
               // time.Sleep(5 * time.Second)
               fmt.Println("cameraid: ", cameraid, " taskid: ", taskid)
                Taskdolist(cameraid, caddr, taskid, msg)
            }
         }
                 if  err := proto.Unmarshal(msg,&recvmsg);  err != nil {
                     fmt.Println("recv msg is not protomsg.Recvmg")
                     continue
                 }
            }
          fmt.Println("============== one msg input ==========")
            switch remote {
               case "camera":
                    for _, taskid := range GetAlltask(recvmsg.Id) {
                        // time.Sleep(5 * time.Second)
                        fmt.Println("id: ", recvmsg.Id , " taskid: ", taskid)
                        Taskdolist(recvmsg.Id, recvmsg.Addr, taskid, recvmsg.Picdata)
                  }
               case  "web":
                        fmt.Println("id: ", recvmsg.Id , " taskid: ", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2")
                        Taskdolist(recvmsg.Id, recvmsg.Addr, "92496BDF-2BFA-98F2-62E8-96DD9866ABD2", recvmsg.Picdata)
           }
      }
   }
}
@@ -171,8 +169,10 @@
   //  计算分发的主题
   SendTopic := sdk.SdkSendTopic(sdkmsg)
    fmt.Println(SendTopic)
   if _, ok := sdk.SdkMap[SendTopic]; ok {
      sdk.SdkMap[SendTopic] <- sdkmsg
         fmt.Println("dispute sendtopic success", SendTopic)
   } else {
      fmt.Println("分发的主题不存在")
   }