| | |
| | | "errors" |
| | | |
| | | "basic.com/valib/deliver.git" |
| | | "github.com/long/test/logger" |
| | | |
| | | "github.com/long/test/sdk" |
| | | "github.com/long/test/util" |
| | |
| | | "fmt" |
| | | "sync" |
| | | //"time" |
| | | "os" |
| | | |
| | | ) |
| | | |
| | |
| | | } |
| | | |
| | | func Init() { |
| | | fmt.Println("============ camera info ====================") |
| | | logger.Info("============ camera info ====================") |
| | | for _, cd := range util.CameraIds { |
| | | fmt.Println(cd) |
| | | fmt.Println() |
| | | logger.Info(cd) |
| | | logger.Info() |
| | | } |
| | | |
| | | // 摄像机初始化 |
| | |
| | | // web端初始化 |
| | | CreateCamera("virtual-faceextract-sdk-pull_2" , "web") |
| | | |
| | | // go AutoDelCamera(util.Cameraflag) |
| | | go AutoDelCamera(util.Cameraflag) |
| | | } |
| | | |
| | | // camera 接受数据 |
| | |
| | | |
| | | socketlisten, err := NewCamerSocketListen(deliver.PushPull, id, url) |
| | | if err != nil { |
| | | fmt.Println("create socket error") |
| | | logger.Error("create socket error") |
| | | return |
| | | } |
| | | |
| | |
| | | func AutoDelCamera(cameraflag chan bool) { |
| | | |
| | | for _ = range cameraflag { |
| | | fmt.Println("test autodelcameraflag") |
| | | logger.Info("test autodelcameraflag") |
| | | var oldcamera []string |
| | | |
| | | SocketManage.Range(func(k, v interface{}) bool { |
| | |
| | | }) |
| | | |
| | | var newcamera []string |
| | | newcamera = append(newcamera, util.Sdklist...) |
| | | |
| | | for _, camnew := range util.CameraIds { |
| | | newcamera = append(newcamera, camnew.Id) |
| | | } |
| | | newcamera = append(newcamera, "virtual-faceextract-sdk-pull_2") |
| | | |
| | | |
| | | cameraChanDel := util.Difference(oldcamera, newcamera) |
| | | fmt.Println(cameraChanDel) |
| | | logger.Info(cameraChanDel) |
| | | |
| | | for key, op := range cameraChanDel { |
| | | if op == "add" { |
| | | CreateCamera(key, "camera") |
| | | fmt.Println("add new camera id=========================") |
| | | os.Exit(1) |
| | | logger.Info("add new camera id=========================", key) |
| | | } else { |
| | | if sock, ok := SocketManage.Load(key); ok { |
| | | if socket, sok := sock.(SocketContext); sok { |
| | |
| | | SocketManage.Delete(key) |
| | | } |
| | | } |
| | | fmt.Println("删除camera server : ", key) |
| | | logger.Info("删除camera server : ", key) |
| | | } |
| | | } |
| | | } |
| | |
| | | socket.Cancel = cancel |
| | | |
| | | socket.Sock = deliver.NewServer(deliver.Mode(mode), url) |
| | | fmt.Println("new socket.Sock: ", socket.Sock) |
| | | logger.Info("new socket.Sock: ", socket.Sock) |
| | | |
| | | if socket.Sock == nil { |
| | | return socket, errors.New("create listen error") |
| | |
| | | for { |
| | | select { |
| | | case <-socket.Context.Done(): |
| | | fmt.Println("listen recv quit") |
| | | logger.Error("listen recv quit") |
| | | return |
| | | default: |
| | | if recvmessage, err = socket.Sock.Recv(); err != nil { |
| | | fmt.Println("err is: ", err) |
| | | //logger.Error("[camera] err is: ", err) |
| | | continue |
| | | } |
| | | |
| | | unmsg, err := util.UnCompress(recvmessage) |
| | | if err != nil { |
| | | fmt.Println(err) |
| | | logger.Error(err) |
| | | continue |
| | | } |
| | | |
| | | if err := proto.Unmarshal(unmsg,&imagemsg); err != nil { |
| | | fmt.Println("recv msg is not protomsgImage") |
| | | logger.Error("recv msg is not protomsgImage") |
| | | continue |
| | | } |
| | | |
| | | // fmt.Println("============== one msg input ==========") |
| | | // fmt.Println(imagemsg.Cid) |
| | | switch remote { |
| | | case "camera": |
| | | fmt.Printf("=== cid: has %d task\n", len(GetAlltask(imagemsg.Cid))) |
| | | for _, taskid := range GetAlltask(imagemsg.Cid) { |
| | | //time.Sleep(5 * time.Second) |
| | | fmt.Println("id: ", imagemsg.Cid, " taskid: ", taskid) |
| | | logger.Debug("id: ", imagemsg.Cid, " taskid: ", taskid) |
| | | Taskdolist(imagemsg.Cid, "", taskid, recvmessage) |
| | | } |
| | | case "web": |
| | | fmt.Println("id: ", imagemsg.Cid , " taskid: ", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2") |
| | | logger.Debug("id: ", imagemsg.Cid , " taskid: ", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2") |
| | | Taskdolist(imagemsg.Cid,"", "92496BDF-2BFA-98F2-62E8-96DD9866ABD2", recvmessage) |
| | | } |
| | | } |
| | |
| | | func Taskdolist(cid string, caddr string, taskid string, data []byte) { |
| | | |
| | | // 数据加工(打标签) |
| | | fmt.Printf("taskid %s: has %d data[]byte\n", taskid, len(data)) |
| | | logger.Debug("taskid: ",taskid, "has ", len(data), "data[]byte") |
| | | sdkmsg := sdk.SdkData(cid, caddr, taskid, data) |
| | | if sdkmsg.Tasklab == nil { |
| | | fmt.Printf("cid:%s 没有任务%s\n", cid, taskid) |
| | | logger.Error(cid, " not have taskid: ", taskid ) |
| | | return |
| | | } |
| | | |
| | | // 计算分发的主题 |
| | | SendTopic := sdk.SdkSendTopic(sdkmsg) |
| | | fmt.Println(SendTopic) |
| | | logger.Debug(SendTopic) |
| | | if _, ok := sdk.SdkMap[SendTopic]; ok { |
| | | sdk.SdkMap[SendTopic] <- sdkmsg |
| | | fmt.Println("dispute sendtopic success", SendTopic) |
| | | logger.Debug("dispute sendtopic success", SendTopic) |
| | | } else { |
| | | fmt.Println("分发的主题不存在") |
| | | logger.Debug("分发的主题不存在") |
| | | } |
| | | } |