| | |
| | | //"sync" |
| | | "taskpubsub/logger" |
| | | |
| | | "basic.com/pubsub/protomsg.git" |
| | | "github.com/gogo/protobuf/proto" |
| | | "taskpubsub/sdk" |
| | | "taskpubsub/util" |
| | | |
| | | "basic.com/pubsub/protomsg.git" |
| | | "github.com/gogo/protobuf/proto" |
| | | |
| | | // "golang.org/x/sys/unix" |
| | | |
| | | "fmt" |
| | | //"time" |
| | |
| | | |
| | | var SocketManage = make(map[string]util.SocketContext) |
| | | |
| | | var shm bool = false |
| | | |
| | | var innerRecvTopic = []string{ |
| | | "virtual-faceextract-sdk-pull_2", //to web 以图搜图 |
| | | } |
| | |
| | | } |
| | | } |
| | | |
| | | func Init() { |
| | | func Init(useShm bool) { |
| | | shm = useShm |
| | | |
| | | logger.Info("============ camera info ====================") |
| | | for _, cd := range util.CameraIds { |
| | | logger.Info(cd) |
| | |
| | | if _, isExist := SocketManage[id]; !isExist { //不存在 |
| | | |
| | | url := fmt.Sprintf("ipc:///tmp/%s.ipc", id) |
| | | socket, err := util.NewSocketListen(int(deliver.PushPull), url) |
| | | m := deliver.PushPull |
| | | if shm { |
| | | m = deliver.Shm |
| | | url = id |
| | | |
| | | } |
| | | |
| | | fmt.Println("CAMERA URL : ", url) |
| | | socket, err := util.NewSocketListen(int(m), url, shm) |
| | | if err != nil { |
| | | logger.Error("create socket error") |
| | | return |
| | |
| | | |
| | | func Recv(socket util.SocketContext) { |
| | | |
| | | tryCount := 0 |
| | | |
| | | var recvmessage []byte |
| | | var imagemsg protomsg.Image |
| | | var err error |
| | |
| | | default: |
| | | if recvmessage, err = socket.Sock.Recv(); err != nil { |
| | | //logger.Error("[camera] err is: ", err) |
| | | // fmt.Println("CAMERA RECV ERROR: ", err) |
| | | |
| | | if socket.UseSHM { |
| | | if tryCount > util.SHMMaxTryCount { |
| | | socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM) |
| | | fmt.Println("CAMERA SHM TRY :", tryCount, " RESTART IT") |
| | | |
| | | tryCount = 0 |
| | | |
| | | continue |
| | | } |
| | | tryCount++ |
| | | } |
| | | continue |
| | | } |
| | | |
| | |
| | | for _, camsingle := range util.CameraTasks { |
| | | if cid == camsingle.Camera.Id { |
| | | for _, tasksingle := range camsingle.Tasks { |
| | | if !tasksingle.Enable { |
| | | continue |
| | | } |
| | | tasks = append(tasks, tasksingle.Taskid) |
| | | } |
| | | return |