liuxiaolong
2019-06-27 906c2c219c73b90eae9e5040d09d1b32b5d928d9
service/FaceSdkService.go
@@ -3,6 +3,7 @@
import (
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "encoding/base64"
   "fmt"
   "github.com/gogo/protobuf/proto"
   "github.com/pierrec/lz4"
@@ -14,9 +15,13 @@
)
const (
   Url_Service_PUSH = "ipc:///tmp///virtual-faceextract-sdk-pull_2.ipc"
   Url_Service_PULL = "ipc:///tmp///virtual-faceextract-sdk-pull_1.ipc"
   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
@@ -40,15 +45,15 @@
         fmt.Println("image is not compressible")
      }
      bc = bc[:n]
      //for {
      for {
         PushImgMsg(protomsg.Recvmsg{
            Id:uuid.NewV4().String(),
            Addr:"",
            Picdata:bc,
         })
         fmt.Println("pushed img")
         //time.Sleep(10*time.Second)
      //}
         time.Sleep(5*time.Second)
      }
   }
}
@@ -93,6 +98,7 @@
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() {
@@ -129,8 +135,32 @@
      }
      rMsg := protomsg.SdkMessage{}
      if err := proto.Unmarshal(resultBytes, &rMsg);err ==nil{
         fmt.Println("recv MSG:",rMsg)
         resultMap[rMsg.Cid] = rMsg
         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))
            //第一个sdk是人脸检测,第二个sdk是人脸提取
               for _,swt :=range sdkInfos{
                  if swt.Sdktype == "FaceExtract"{
                     fmt.Println("sdkName:",swt.SdkName)
                     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)
      }