package service
|
|
import (
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/deliver.git"
|
"encoding/json"
|
"fmt"
|
"github.com/gogo/protobuf/proto"
|
"github.com/satori/go.uuid"
|
"io/ioutil"
|
"time"
|
)
|
|
const (
|
Url_Service_PUSH = "tcp:///tmp///webserver-2.ipc"
|
Url_Service_PULL = "tcp:///tmp///webserver-1.ipc"
|
)
|
var imgPushChan chan protomsg.Recvmsg
|
|
func TestPushImgMsg() {
|
InitService()
|
imgData := readImgFile()
|
fmt.Println("imgData.len:",len(imgData))
|
for {
|
PushImgMsg(protomsg.Recvmsg{
|
Id:uuid.NewV4().String(),
|
Addr:"",
|
Picdata:imgData,
|
})
|
fmt.Println("pushed img")
|
time.Sleep(5*time.Second)
|
}
|
}
|
|
func readImgFile() []byte{
|
filePath := "/home/user/workspace/timg.jpg"
|
bytes, err := ioutil.ReadFile(filePath)
|
if err !=nil {
|
fmt.Println("Read img err:",err)
|
}
|
return bytes
|
}
|
|
func PushImgMsg(is protomsg.Recvmsg){
|
imgPushChan <- is
|
}
|
|
var resultMap map[string]protomsg.SdkMessage
|
|
|
func InitService(){
|
imgPushChan = make(chan protomsg.Recvmsg)
|
client_push := deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
|
client_pull := deliver.NewClient(deliver.PushPull, Url_Service_PULL)
|
for {
|
select {
|
case is := <- imgPushChan:
|
b, _ := json.Marshal(is)
|
client_push.Send(b)
|
}
|
}
|
|
//接收人脸提取结果
|
for {
|
resultBytes, err := client_pull.Recv()
|
if err !=nil{
|
fmt.Println("pull err:",err)
|
continue
|
}
|
rMsg := protomsg.SdkMessage{}
|
if err := proto.Unmarshal(resultBytes, &rMsg);err !=nil{
|
fmt.Println("recv MSG:",rMsg)
|
resultMap[rMsg.Cid] = rMsg
|
}
|
|
}
|
|
}
|
