From 82e97bd7595ede5f59befe3dd977c8ae11ec848c Mon Sep 17 00:00:00 2001
From: sunty <suntianyu0923@163.com>
Date: 星期一, 05 八月 2019 14:39:58 +0800
Subject: [PATCH] add Cascade modification dbtablepersons enable status

---
 controllers/fileController.go |   40 +++++++++++++++++++++++++++++++++++-----
 1 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/controllers/fileController.go b/controllers/fileController.go
index d801abd..32e9cfc 100644
--- a/controllers/fileController.go
+++ b/controllers/fileController.go
@@ -17,6 +17,7 @@
 	"mime/multipart"
 	"net/http"
 	"path"
+	"sort"
 	"strconv"
 	"strings"
 	"time"
@@ -179,11 +180,25 @@
 	Sex string `json:"sex"`
 	TableId string `json:"tableId"`
 	TableName string `json:"tableName"`
+	Enable int32 `json:"enable"`
+}
+
+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)
@@ -221,6 +236,7 @@
 				PhoneNum: p.PhoneNum,
 				Sex: p.Sex,
 				TableId: p.TableId,
+				Enable: p.Enable,
 			}
 			dbTableInfos, _ := esApi.Dbtablefosbyid([]string{p.TableId}, config.EsInfo.EsIndex.DbTables.IndexName, esServerIp, esServerPort)
 			if dbTableInfos !=nil{
@@ -250,6 +266,8 @@
 					PhoneNum: p.PhoneNum,
 					Sex: p.Sex,
 					TableId: p.TableId,
+					BwType: strconv.Itoa(int(p.BwType)),
+					TableName: p.TableName,
 				})
 			}
 			vpE := CompareResult{
@@ -273,6 +291,7 @@
 			resultList = append(resultList,vpE)
 		}
 	}
+	sort.Sort(resultList)
 	return  resultList
 }
 
@@ -365,12 +384,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