| | |
| | | "context" |
| | | "errors" |
| | | "fmt" |
| | | "os" |
| | | |
| | | // "github.com/long/test/httpclient" |
| | | "github.com/long/test/tasktag" |
| | | "github.com/long/test/util" |
| | | |
| | | "github.com/golang/protobuf/proto" |
| | | "github.com/gogo/protobuf/proto" |
| | | |
| | | "basic.com/pubsub/protomsg.git" |
| | | "basic.com/valib/deliver.git" |
| | |
| | | fmt.Println() |
| | | } |
| | | |
| | | // 手动输入的主题 |
| | | SdkMap["es"] = make(chan protomsg.SdkMessage) |
| | | fmt.Println("create es channel: ") |
| | | go es(SdkMap["es"]) |
| | | fmt.Println("create es channel: ") |
| | | |
| | | SdkMap["virtual-faceextract-sdk-pull"] = make(chan protomsg.SdkMessage) |
| | | fmt.Println("create virtual-faceextract-sdk-pull") |
| | | |
| | | Createwebserver("virtual-faceextract-sdk-pull") |
| | | |
| | | go Dealextern() |
| | | go AutoDelSdk(util.Sdkflag) |
| | | } |
| | | |
| | | func CreatesdkTopicandServer(sdkid string) { |
| | | |
| | | SdkMap[sdkid] = make(chan protomsg.SdkMessage) |
| | | fmt.Println("create sdk channel: ", sdkid) |
| | | |
| | | url := fmt.Sprintf("ipc:///tmp/%s%s", sdkid, postPull) |
| | | sdkidser, socketser, err := NewSdkSocketListen(deliver.PushPull, sdkid, url) |
| | | socketser, err := NewSdkSocketListen(deliver.PushPull, sdkid, url) |
| | | if err != nil { |
| | | delete(SdkMap, sdkid) |
| | | delete(SdkMap, sdkid) |
| | | fmt.Println(sdkid, "create server error!") |
| | | return |
| | | } |
| | | go Send(sdkidser, socketser, SdkMap[sdkid]) |
| | | |
| | | |
| | | go Send(sdkid, socketser, SdkMap[sdkid]) |
| | | |
| | | url = fmt.Sprintf("ipc:///tmp/%s%s", sdkid, postPush) |
| | | _, socketdial, err := NewSdkSocketListen(deliver.PushPull, sdkid, url) |
| | | socketdial, err := NewSdkSocketListen(deliver.PushPull, sdkid, url) |
| | | if err != nil { |
| | | delete(SdkMap, sdkid) |
| | | delete(SdkMap, sdkid) |
| | | fmt.Println(sdkid, "create dial error!") |
| | | return |
| | | } |
| | | go Recv(socketdial) |
| | | |
| | | } |
| | | |
| | | // web 接受端处理 |
| | | func Createwebserver(webid string ){ |
| | | url := fmt.Sprintf("ipc:///tmp/%s%s", webid, postPull) |
| | | socketser, err := NewSdkSocketListen(deliver.PushPull, webid, url) |
| | | if err != nil { |
| | | fmt.Println(webid, "create server error!") |
| | | return |
| | | } |
| | | go Send(webid, socketser, SdkMap[webid]) |
| | | } |
| | | |
| | | func DeletesdkTopicandServer(sdkid string) { |
| | | close(SdkMap[sdkid]) |
| | | delete(SdkMap, sdkid) |
| | |
| | | } |
| | | |
| | | //单独处理 es 主题的情况 |
| | | func es(sdkmsgchan chan protomsg.SdkMessage) { |
| | | for _ = range sdkmsgchan { |
| | | fmt.Println("this data is finish all sdk! ") |
| | | } |
| | | func Dealextern() { |
| | | for { |
| | | select { |
| | | case esmsg:= <-SdkMap["es"]: |
| | | fmt.Println(esmsg) |
| | | } |
| | | } |
| | | } |
| | | |
| | | //动态处理 |
| | |
| | | oldSdk = append(oldSdk, key) |
| | | } |
| | | util.Sdklist = append(util.Sdklist, "es") |
| | | util.Sdklist = append(util.Sdklist, "virtual-faceextract-sdk-pull") |
| | | sdkChanDel := util.Difference(oldSdk, util.Sdklist) |
| | | fmt.Println(sdkChanDel) |
| | | |
| | |
| | | } |
| | | |
| | | // create server |
| | | func NewSdkSocketListen(mode int, sdkid string, url string) (sid string, socket SocketContext, err error) { |
| | | func NewSdkSocketListen(mode int, sdkid string, url string) (socket SocketContext, err error) { |
| | | fmt.Println("url is: ", url) |
| | | ctx, cancel := context.WithCancel(context.Background()) |
| | | |
| | |
| | | socket.Sock = deliver.NewServer(deliver.Mode(mode), url) |
| | | |
| | | if socket.Sock == nil { |
| | | return sdkid, socket, errors.New("create listen error") |
| | | return socket, errors.New("create listen error") |
| | | } |
| | | |
| | | SocketManage[sdkid] = socket |
| | | return sdkid, socket, nil |
| | | return socket, nil |
| | | } |
| | | |
| | | func NewSdkSocketDial(mode int, sdkid string, url string) (sid string, socket SocketContext, err error) { |
| | |
| | | } |
| | | |
| | | func Recv(socket SocketContext) { |
| | | |
| | | var repsdkmsg = protomsg.SdkMessage{} |
| | | for { |
| | | select { |
| | |
| | | fmt.Println("receive len: ", len(msg)) |
| | | if err != nil { |
| | | fmt.Println("unmarshal error: ", err) |
| | | os.Exit(1) |
| | | continue |
| | | } |
| | | } |
| | | repsdkmsg.Tasklab.Index++ |
| | | //调用计算函数, 分发给下一个主题 |
| | | nexttopic := SdkSendTopic(repsdkmsg) |
| | | SdkMap[nexttopic] <- repsdkmsg |
| | | //调用计算函数, 分发给下一个主题 |
| | | nexttopic := SdkSendTopic(repsdkmsg) |
| | | SdkMap[nexttopic] <- repsdkmsg |
| | | } |
| | | } |
| | | } |
| | |
| | | if err != nil { |
| | | fmt.Println("proto marshal error ", err) |
| | | continue |
| | | } |
| | | |
| | | fmt.Printf("从管道sdkid=%s 接受数据 %d\n", sdkid, len(data)) |
| | | fmt.Println() |
| | | |
| | | fmt.Println("send len of data: ", len(data)) |
| | | if err := socket.Sock.Send(data); err != nil { |
| | | fmt.Println(socket.Sock) |
| | | fmt.Println("failed send") |
| | |
| | | fmt.Println(sdkid, " 主题关闭, 关闭send()") |
| | | return |
| | | } |
| | | |
| | | } |
| | | } |
| | | } |
| | | } |