remove sdkname, sdkid, add ipc_id

554325746@qq.com

2019-07-10 7183fd16c435d2aaf85e24d3608c5c43645ee608

 camera/camera.go

@@ -3,22 +3,20 @@
import (
   "errors"

   "basic.com/dbapi.git"
   "basic.com/valib/deliver.git"

   "github.com/long/test/sdk"
   "github.com/long/test/util"
    "github.com/gogo/protobuf/proto"
    "basic.com/pubsub/protomsg.git"

   "context"
   "fmt"
   "sync"
   "time"
   // "time"
)

//var SocketManage = make(map[string]SocketContext)
var SocketManage sync.Map

var Initchannel = make(chan string)

type SocketContext struct {
   Sock    deliver.Deliver
@@ -26,43 +24,76 @@
   Cancel  context.CancelFunc
}

var camval dbapi.CameraApi

func Init() {

   fmt.Println("============ camera info ====================")
   for _, cd := range util.CameraIds {
      fmt.Println(cd)
      fmt.Println()
   }

   go CreateCamera(Initchannel)

    // 摄像机初始化
   for _, cam := range util.CameraIds {
      Initchannel <- cam.Id
        CreateCamera(cam.Id, "camera")
   }

    //  web端初始化
    CreateCamera("virtual-faceextract-sdk-pull_2" , "web")

//   go AutoDelCamera(util.Cameraflag)
}

func CreateCamera(camera chan string) {
   for camid := range camera {
      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 Recv(socketlisten, remote)
      }
}

         go func(cid string, sock SocketContext) {
            Recv(cid, sock)
         }(id, socketlisten)
//动态处理
func AutoDelCamera(cameraflag chan bool) {

   for _ = range cameraflag {
      fmt.Println("test autodelcameraflag")
      var oldcamera []string

      SocketManage.Range(func(k, v interface{}) bool {
         if str, ok := k.(string); ok {
            oldcamera = append(oldcamera, str)
         }
         return true
      })

        var newcamera []string
       newcamera = append(newcamera, util.Sdklist...)

      cameraChanDel := util.Difference(oldcamera, newcamera)
      fmt.Println(cameraChanDel)

      for key, op := range cameraChanDel {
         if op == "add" {
               CreateCamera(key, "camera") 
         } else {
            if sock, ok := SocketManage.Load(key); ok {
               if socket, sok := sock.(SocketContext); sok {
                  socket.Cancel()
                  SocketManage.Delete(key)
               }
            }
            fmt.Println("删除camera server : ", key)
         }
      }
   }
}

// 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
@@ -72,39 +103,57 @@
   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, socket SocketContext) {
   var msg []byte
   var err error
func Recv(socket SocketContext, remote string ) {

    var recvmessage []byte
    var imagemsg  protomsg.Image
    var err error
   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 recvmessage, err = socket.Sock.Recv(); err != nil {
            fmt.Println("err is: ", err)
            continue
         } else {
            fmt.Println()
            fmt.Println("============== one msg input ==========")
            fmt.Println("cameraid: ", cameraid, len(msg))
            for _, taskid := range GetAlltask(cameraid) {
               time.Sleep(5 * time.Second)
               fmt.Println("cameraid: ", cameraid, " taskid: ", taskid)
               Taskdolist(cameraid, taskid, msg)
            }
         }

            unmsg, err := util.UnCompress(recvmessage)
            if err != nil {
                fmt.Println(err)
                continue
            }

            if  err := proto.Unmarshal(unmsg,&imagemsg);  err != nil {
               fmt.Println("recv msg is not protomsgImage") 
               continue
            }

       //   fmt.Println("============== one msg input ==========")
         //   fmt.Println(imagemsg.Cid)
            switch remote {
               case "camera":
                    for _, taskid := range GetAlltask(imagemsg.Cid) {
                       //  time.Sleep(5 * time.Second)
                        fmt.Println("id: ", imagemsg.Cid, " taskid: ", taskid)
                        Taskdolist(imagemsg.Cid, "", taskid, recvmessage)
                  }
               case  "web":
                        fmt.Println("id: ", imagemsg.Cid , " taskid: ", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2")
                        Taskdolist(imagemsg.Cid,"", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2", recvmessage)
           }
      }
   }
}

//   根据cid 获取 所有的任务
//   据cid 获取 所有的任务
func GetAlltask(cid string) (tasks []string) {
   for _, camsingle := range util.CameraTasks {
      if cid == camsingle.Camera.Id {
@@ -117,21 +166,21 @@
   return
}

func Taskdolist(cid string, taskid string, data []byte) {
func Taskdolist(cid string, caddr string,  taskid string, data []byte) {

   //  数据加工(打标签)
   sdkmsg := sdk.SdkData(cid, taskid, data)
   sdkmsg := sdk.SdkData(cid, caddr, taskid, data)
   if sdkmsg.Tasklab == nil {
      fmt.Println("cid:%s 没有任务%s", cid, taskid)
      fmt.Printf("cid:%s 没有任务%s\n", cid, taskid)
      return
   }

   //  计算分发的主题
   SendTopic := sdk.SdkSendTopic(sdkmsg)
    fmt.Println(SendTopic)
   if _, ok := sdk.SdkMap[SendTopic]; ok {
      fmt.Println("分发的主题存在")
      sdk.SdkMap[SendTopic] <- sdkmsg
      //fmt.Println("重新开始循环: ", sdk.SdkMap)
         fmt.Println("dispute sendtopic success", SendTopic)
   } else {
      fmt.Println("分发的主题不存在")
   }