From 24e2d92df7cad99e1e514fa581fe59765dbc4a12 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期五, 02 八月 2019 09:54:33 +0800 Subject: [PATCH] rm discovery server --- controllers/fileController.go | 38 +++++++++++++++++++++++++++++++++----- 1 files changed, 33 insertions(+), 5 deletions(-) diff --git a/controllers/fileController.go b/controllers/fileController.go index d801abd..b3b4406 100644 --- a/controllers/fileController.go +++ b/controllers/fileController.go @@ -17,6 +17,7 @@ "mime/multipart" "net/http" "path" + "sort" "strconv" "strings" "time" @@ -181,9 +182,22 @@ TableName string `json:"tableName"` } +type CompareSortResult []CompareResult + +func (s CompareSortResult) Len() int{ + return len(s) +} +func (s CompareSortResult) Swap(i,j int){ + s[i],s[j] = s[j],s[i] +} +func (s CompareSortResult) Less(i,j int) bool{ + return s[i].CompareScore > s[j].CompareScore +} + //濉厖鍚戝墠绔繑鍥炵殑鏁版嵁 -func FillDataToCompareResult(compResult *protomsg.SdkCompareResult) []CompareResult { - resultList :=make([]CompareResult,0) +func FillDataToCompareResult(compResult *protomsg.SdkCompareResult) CompareSortResult { + //resultList :=make([]CompareResult,0) + var resultList = CompareSortResult{} dbPersonM := make(map[string]*protomsg.SdkCompareEach, 0) captureM := make(map[string]*protomsg.SdkCompareEach, 0) @@ -250,6 +264,8 @@ PhoneNum: p.PhoneNum, Sex: p.Sex, TableId: p.TableId, + BwType: strconv.Itoa(int(p.BwType)), + TableName: p.TableName, }) } vpE := CompareResult{ @@ -273,6 +289,7 @@ resultList = append(resultList,vpE) } } + sort.Sort(resultList) return resultList } @@ -365,12 +382,23 @@ compareService := service.NewFaceCompareService(arg) compareOnce := compareService.Compare() totalData := compareOnce.CompareData + logger.Debug("comp 姣斿缁撴潫锛岀敤鏃讹細",time.Since(timeStart)) m := make(map[string]interface{},3) - if totalData!=nil && len(*totalData)>0{ + if totalData != nil && totalData.Len() > 0{ //鎸夊垎鍊兼帓搴� - service.SortByScore(totalData) - total := len(*totalData) + var scorePre []float32 + for _,f1 :=range *totalData { + scorePre = append(scorePre, f1.CompareScore) + } + //logger.Debug("comp 鎺掑簭鍓嶅垎鍊兼暟缁勶細",scorePre) + sort.Sort(totalData) + var scoreAf []float32 + for _,f2 :=range *totalData { + scoreAf = append(scoreAf, f2.CompareScore) + } + //logger.Debug("comp 鎺掑簭鍚庡垎鍊兼暟缁勶細",scoreAf) + total := totalData.Len() m["compareNum"] = compareOnce.CompareNum m["total"] = total -- Gitblit v1.8.0