| | |
| | | |
| | | "github.com/long/test/sdk" |
| | | "github.com/long/test/util" |
| | | "github.com/gogo/protobuf/proto" |
| | | "basic.com/pubsub/protomsg.git" |
| | | |
| | | "context" |
| | |
| | | // "time" |
| | | ) |
| | | |
| | | //var SocketManage = make(map[string]SocketContext) |
| | | var SocketManage sync.Map |
| | | |
| | | var Initchannel = make(chan protomsg.Camera ) |
| | | |
| | | type SocketContext struct { |
| | | Sock deliver.Deliver |
| | |
| | | } |
| | | |
| | | 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") |
| | | } |
| | | |
| | | 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) |
| | | // web端初始化 |
| | | CreateCamera("virtual-faceextract-sdk-pull_2" , "web") |
| | | |
| | | id, socketlisten, err := NewCamerSocketListen(deliver.PushPull, camid, url) |
| | | |
| | | |
| | | // go AutoDelCamera(util.Cameraflag) |
| | | } |
| | | |
| | | // camera 接受数据 |
| | | func CreateCamera(id string, remote string) { |
| | | if _, ok := SocketManage.Load(id); !ok { |
| | | url := fmt.Sprintf("ipc:///tmp/%s.ipc", id) |
| | | |
| | | 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) |
| | | } |
| | | } |
| | | |
| | |
| | | }) |
| | | |
| | | var newcamera []string |
| | | |
| | | for _, value := range util.CameraIds { |
| | | newcamera = append(newcamera, value.Id) |
| | | } |
| | | 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 { |
| | |
| | | } |
| | | |
| | | // 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 |
| | |
| | | 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) |
| | | } |
| | | } |
| | | } |
| | |
| | | |
| | | // 计算分发的主题 |
| | | 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("分发的主题不存在") |
| | | } |