| | |
| | | "context" |
| | | "fmt" |
| | | "sync" |
| | | // "time" |
| | | //"time" |
| | | "os" |
| | | |
| | | ) |
| | | |
| | | var SocketManage sync.Map |
| | |
| | | // web端初始化 |
| | | CreateCamera("virtual-faceextract-sdk-pull_2" , "web") |
| | | |
| | | |
| | | |
| | | // go AutoDelCamera(util.Cameraflag) |
| | | } |
| | | |
| | |
| | | fmt.Println("create socket error") |
| | | return |
| | | } |
| | | |
| | | go Recv(socketlisten, remote) |
| | | } |
| | | } |
| | |
| | | for key, op := range cameraChanDel { |
| | | if op == "add" { |
| | | CreateCamera(key, "camera") |
| | | fmt.Println("add new camera id=========================") |
| | | os.Exit(1) |
| | | } else { |
| | | if sock, ok := SocketManage.Load(key); ok { |
| | | if socket, sok := sock.(SocketContext); sok { |
| | |
| | | |
| | | func Recv(socket SocketContext, remote string ) { |
| | | |
| | | var recvmsg protomsg.Recvmsg |
| | | 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 { |
| | | if recvmessage, err = socket.Sock.Recv(); err != nil { |
| | | fmt.Println("err is: ", err) |
| | | continue |
| | | } else { |
| | | if err := proto.Unmarshal(msg,&recvmsg); err != nil { |
| | | fmt.Println("recv msg is not protomsg.Recvmg") |
| | | continue |
| | | } |
| | | } |
| | | |
| | | unmsg, err := util.UnCompress(recvmessage) |
| | | if err != nil { |
| | | fmt.Println(err) |
| | | continue |
| | | } |
| | | fmt.Println("============== one msg input ==========") |
| | | |
| | | 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(recvmsg.Id) { |
| | | // time.Sleep(5 * time.Second) |
| | | fmt.Println("id: ", recvmsg.Id , " taskid: ", taskid) |
| | | Taskdolist(recvmsg.Id, recvmsg.Addr, taskid, recvmsg.Picdata) |
| | | 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) |
| | | Taskdolist(imagemsg.Cid, "", taskid, recvmessage) |
| | | } |
| | | 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) |
| | | fmt.Println("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)) |
| | | sdkmsg := sdk.SdkData(cid, caddr, taskid, data) |
| | | if sdkmsg.Tasklab == nil { |
| | | fmt.Printf("cid:%s 没有任务%s\n", cid, taskid) |