package service
|
|
import (
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/deliver.git"
|
"encoding/base64"
|
"fmt"
|
"github.com/gogo/protobuf/proto"
|
"github.com/pierrec/lz4"
|
"github.com/satori/go.uuid"
|
"gocv.io/x/gocv"
|
"time"
|
)
|
|
const (
|
Ipc_Push_Ext = "_2.ipc"
|
Ipc_Pull_Ext = "_1.ipc"
|
Ipc_Url_Pre = "ipc:///tmp///"
|
Virtual_FaceTaskId = "92496BDF-2BFA-98F2-62E8-96DD9866ABD2"
|
Virtual_FaceSdkId = "virtual-faceextract-sdk-pull"
|
Url_Service_PUSH = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Push_Ext
|
Url_Service_PULL = Ipc_Url_Pre + Virtual_FaceSdkId + Ipc_Pull_Ext
|
)
|
var imgPushChan chan protomsg.Recvmsg
|
var client_push deliver.Deliver
|
var client_pull deliver.Deliver
|
|
func TestPushImgMsg() {
|
InitService()
|
|
i := readImgFile()
|
if b, err := proto.Marshal(&i);err !=nil{
|
fmt.Println("protoImage marshal err")
|
return
|
} else {
|
bc := make([]byte,len(b))
|
ht := make([]int, 64<<10)
|
n,err := lz4.CompressBlock(b,bc,ht)
|
if err !=nil {
|
fmt.Println(err)
|
}
|
if n >= len(b){
|
fmt.Println("image is not compressible")
|
}
|
bc = bc[:n]
|
for {
|
PushImgMsg(protomsg.Recvmsg{
|
Id:uuid.NewV4().String(),
|
Addr:"",
|
Picdata:bc,
|
})
|
fmt.Println("pushed img")
|
time.Sleep(5*time.Second)
|
}
|
|
}
|
}
|
|
func readImgFile() protomsg.Image{
|
var i protomsg.Image
|
timeUnix := time.Now().Unix()
|
formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")
|
filePath := "/home/user/workspace/timg.jpg"
|
|
picMat := gocv.IMRead(filePath, gocv.IMReadColor)
|
|
defer picMat.Close()
|
|
if picMat.Empty() {
|
fmt.Println("file not exist")
|
return i
|
}
|
width := int32(picMat.Rows())
|
height := int32(picMat.Cols())
|
data := picMat.DataPtrUint8()
|
fmt.Printf("width:%d,height:%d,data.length:%d,timestamp:%s",width,height,len(data),formatTimeStr)
|
i = protomsg.Image{
|
Width: width,
|
Height: height,
|
Timestamp: formatTimeStr,
|
Data: data,
|
Id: timeUnix,
|
}
|
fmt.Println("gocv read img completed")
|
return i
|
}
|
|
func PushImgMsg(is protomsg.Recvmsg){
|
imgPushChan <- is
|
}
|
|
var resultMap map[string]protomsg.SdkMessage
|
|
|
func InitService(){
|
fmt.Println("service init!")
|
imgPushChan = make(chan protomsg.Recvmsg)
|
resultMap = make(map[string]protomsg.SdkMessage,0)
|
client_push = deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
|
client_pull = deliver.NewClient(deliver.PushPull, Url_Service_PULL)
|
defer func() {
|
client_push.Close()
|
client_pull.Close()
|
}()
|
go thSend()
|
|
go thRecv()
|
}
|
|
func thSend(){
|
for {
|
select {
|
case is := <- imgPushChan:
|
fmt.Println("imgPushChan in")
|
b, _ := proto.Marshal(&is)
|
err := client_push.Send(b)
|
if err !=nil {
|
fmt.Println("img Send err:",err)
|
}
|
default:
|
//fmt.Println("no img in")
|
}
|
}
|
}
|
|
func thRecv(){
|
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("received MSG:",rMsg.Cid)
|
perId := rMsg.Cid //数据id
|
if rMsg.Tasklab !=nil && rMsg.Tasklab.Taskid == Virtual_FaceTaskId {
|
sdkInfos := rMsg.Tasklab.Sdkinfos
|
fmt.Println("Len(sdkInfos)=",len(sdkInfos))
|
for _,swt :=range sdkInfos{
|
if swt.Sdktype == "FaceExtract"{
|
fmt.Println("sdkName:",swt.SdkName)
|
fmt.Println("sdkData.len:",len(swt.Sdkdata))
|
var pff protomsg.ParamFaceFeature
|
if err := proto.Unmarshal(swt.Sdkdata, &pff);err !=nil{
|
fmt.Println("ParamFaceFeature unmarshal err:",err)
|
} else {
|
fmt.Println("目标数：",len(pff.ExtComp))
|
for _,fea :=range pff.ExtComp{
|
base64Fea := base64.StdEncoding.EncodeToString(fea.Feats)
|
fmt.Println("perId:",perId)
|
fmt.Println("faceFeature:",base64Fea)
|
}
|
}
|
break
|
}
|
}
|
|
}
|
//resultMap[rMsg.Cid] = rMsg
|
} else {
|
fmt.Println("recv msg Err:",err)
|
}
|
|
}
|
}
|
