From 3b07d34fc238a5401896ada6e8d551a13c92e2fc Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期三, 31 七月 2019 15:19:20 +0800 Subject: [PATCH] fix compare --- service/FaceCompareService.go | 29 +++++++++++++++-------------- 1 files changed, 15 insertions(+), 14 deletions(-) diff --git a/service/FaceCompareService.go b/service/FaceCompareService.go index 71543ef..4c0de89 100644 --- a/service/FaceCompareService.go +++ b/service/FaceCompareService.go @@ -3,7 +3,6 @@ import ( esApi "basic.com/pubsub/esutil.git" "basic.com/pubsub/protomsg.git" - "encoding/json" "github.com/gogo/protobuf/proto" "github.com/satori/go.uuid" "nanomsg.org/go-mangos" @@ -122,15 +121,17 @@ if resultB == nil || len(*resultB) ==0 { continue } - rList :=make([]protomsg.Esinfo,0) - err = json.Unmarshal(*resultB, &rList) - if err !=nil{ - logger.Debug("comp recv result Unmarshal err:", err) + var sdkCompResult protomsg.SdkCompareResult + //rList :=make([]protomsg.Esinfo,0) + err = proto.Unmarshal(*resultB, &sdkCompResult) + if err !=nil { + logger.Debug("comp sdkCompareResult unmarshal err:", err) continue } - logger.Debug("comp len(rList):", len(rList)) - if len(rList) >0 { - resultList = append(resultList, FillDataToCompareResult(&rList)...) + + logger.Debug("comp len(rList):", len(sdkCompResult.CompareResult)) + if len(sdkCompResult.CompareResult) >0 { + resultList = append(resultList, FillDataToCompareResult(&sdkCompResult)...) } } logger.Debug("comp totalList.len:", len(resultList)) @@ -176,19 +177,19 @@ } //濉厖鍚戝墠绔繑鍥炵殑鏁版嵁 -func FillDataToCompareResult(rList *[]protomsg.Esinfo) []CompareResult { +func FillDataToCompareResult(compResult *protomsg.SdkCompareResult) []CompareResult { resultList :=make([]CompareResult,0) - dbPersonM := make(map[string]*protomsg.Esinfo,0) - captureM := make(map[string]*protomsg.Esinfo,0) + dbPersonM := make(map[string]*protomsg.SdkCompareEach, 0) + captureM := make(map[string]*protomsg.SdkCompareEach, 0) personIds :=make([]string,0) captureIds := make([]string,0) - for _,v :=range *rList{ + for _,v :=range compResult.CompareResult{ if v.Tableid == CaptureTable { - captureM[v.Id] = &v + captureM[v.Id] = v captureIds = append(captureIds,v.Id) } else { - dbPersonM[v.Id] = &v + dbPersonM[v.Id] = v personIds = append(personIds,v.Id) } } -- Gitblit v1.8.0