From 9303b69ea569bcb5e581147543a3fd58e90d0d25 Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期四, 20 八月 2020 20:05:23 +0800
Subject: [PATCH] add get buckets contrl

---
 controllers/fileController.go |  783 ++++++++++++++++++++++++++++++++++++-------------------
 1 files changed, 514 insertions(+), 269 deletions(-)

diff --git a/controllers/fileController.go b/controllers/fileController.go
index 8ebc80c..1f591af 100644
--- a/controllers/fileController.go
+++ b/controllers/fileController.go
@@ -3,33 +3,35 @@
 import (
 	"basic.com/dbapi.git"
 	"basic.com/fileServer/WeedFSClient.git"
+	esApi "basic.com/pubsub/esutil.git"
 	"basic.com/pubsub/protomsg.git"
+	"basic.com/valib/godraw.git"
+	"basic.com/valib/logger.git"
 	"bytes"
 	"encoding/base64"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/gin-gonic/gin"
-	"github.com/satori/go.uuid"
 	"image"
+	"image/color"
 	"image/jpeg"
 	"io/ioutil"
 	"log"
 	"mime/multipart"
-	"net/http"
 	"path"
 	"sort"
 	"strconv"
 	"strings"
 	"sync"
 	"time"
+	"webserver/cache"
 	"webserver/extend/code"
 	"webserver/extend/config"
 	"webserver/extend/esutil"
-	"webserver/extend/logger"
 	"webserver/extend/util"
 	"webserver/models"
 	"webserver/service"
-	esApi "basic.com/pubsub/esutil.git"
 )
 
 type FileController struct {
@@ -46,57 +48,65 @@
 
 //var weedfsUri = "http://192.168.1.182:9500/submit"
 
-// @Summary 渚濇嵁鍥剧墖娣诲姞搴曞簱浜哄憳
-// @Description  渚濇嵁鍥剧墖娣诲姞搴曞簱杩斿洖鏁版嵁浜哄憳
-// @Accept  mpfd
+// @Security ApiKeyAuth
+// @Summary 鍥剧墖涓婁紶
+// @Description  鍥剧墖涓婁紶
+// @Accept  multipart/form-data
 // @Produce json
-// @Tags dbperson 搴曞簱浜哄憳
-// @Param file formData file true "搴曞簱浜哄憳鍥剧墖"
-// @Param tableId formData string false "搴曞簱id,鏈塱d 鍒欏姞鍏ュ簳搴擄紝鏃犲垯鍙笂浼犲浘鐗�"
-// @Success 200 {string} json "{"code":200, msg:"鐩綍缁撴瀯鏁版嵁", success:true}"
-// @Failure 500 {string} json "{"code":500,  msg:"杩斿洖閿欒淇℃伅", success:false}"
+// @Tags 鍥剧墖涓婁紶
+// @Param file formData file true "鍥剧墖"
+// @Param picType formData string true "鍥剧墖绫诲瀷锛�0锛氳溅杈嗙収鐗囷紝1锛氫汉鑴哥収鐗�"
+// @Success 200 {string} json "{"code":200, msg:"", success:true}"
+// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
 // @Router /data/api-v/dbperson/fileupload [POST]
 func (fc FileController) Fileupload(c *gin.Context) {
 	//寰楀埌涓婁紶鐨勬枃浠�
 	file, header, err := c.Request.FormFile("file") //image杩欎釜鏄痷plaodify鍙傛暟瀹氫箟涓殑   'fileObjName':'image'
-	tableId := c.Request.FormValue("tableId")
-	if err != nil {
-		c.String(http.StatusBadRequest, "Bad request")
+	picType := c.Request.FormValue("picType") //鍥剧墖绫诲瀷
+	if err != nil || (picType != "0" && picType != "1") {
+		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟鏈夎")
 		return
 	}
 	//鏂囦欢鐨勫悕绉�
 	filename := header.Filename
-	fmt.Println(file, err, filename)
-	if err != nil {
-		log.Fatal(err)
-		filename = uuid.NewV4().String()
-	}
-	// fastdfs
-	{
-		//fileSuffix := path.Ext(filename)[1:] //鑾峰彇鏂囦欢鍚庣紑
-		//fmt.Println("鏂囦欢鍚庣紑 =", fileSuffix)
-		//data, _ := ioutil.ReadAll(file)
-		// fastdfs 涓婁紶  鏆傚簾寮�
-		//field := fdfsclient.UploadFileByBuffer(data, fileSuffix)
-	}
-	fileNameOnly, result, err := uploadFileReturnAddr(file, filename, tableId)
-	if err != nil {
-		if err.Error() == "NotFeatureFindError" {
-			util.ResponseFormat(c, code.NotFeatureFindError, filename)
-		} else {
-			util.ResponseFormat(c, code.ServiceInsideError, err.Error())
+	var result = make(map[string]string)
+	if picType == "1" {
+		_, _, err, faceFeature, faceUrlPath := uploadFileReturnAddr(file, filename, "", false)
+		if err != nil {
+			if err.Error() == "NotFeatureFindError" {
+				util.ResponseFormat(c, code.NotFeatureFindError, filename)
+			} else {
+				util.ResponseFormat(c, code.ServiceInsideError, err.Error())
+			}
+			return
 		}
-		return
-	}
-	if tableId != "" { //  涓婁紶
-		if result["success"].(bool) {
-			util.ResponseFormat(c, code.Success, result["data"])
-		} else {
-			util.ResponseFormat(c, code.ServiceInsideError, result["data"])
-		}
+		result["picUrl"] = faceUrlPath
+		result["faceFeature"] = faceFeature
 	} else {
-		util.ResponseFormat(c, code.Success, fileNameOnly)
+		localConf, err2 := cache.GetServerInfo()
+		if err2 !=nil || localConf.WebPicIp == "" {
+			logger.Debug("localConfig is wrong!!!")
+			util.ResponseFormat(c,code.ComError,"")
+			return
+		}
+		defer file.Close()
+		// weedfs 涓婁紶
+		fileBytes, err := ioutil.ReadAll(file)
+		if err !=nil {
+			util.ResponseFormat(c,code.ComError,"")
+			return
+		}
+		var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
+		weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, filename, fileBytes)
+		if e != nil {
+			logger.Debug("WeedFSClient.UploadFile err:", e)
+			util.ResponseFormat(c,code.ComError,"UploadFile err")
+			return
+		}
+		result["picUrl"] = weedFilePath
 	}
+
+	util.ResponseFormat(c, code.Success, result)
 }
 
 
@@ -106,8 +116,10 @@
 }
 var faceExtractedMap = make(map[string]FaceExtract,0)
 
+// @Security ApiKeyAuth
 // @Summary 浜鸿劯鎻愬彇
 // @Description  浜鸿劯鎻愬彇
+// @Accept multipart/form-data
 // @Produce json
 // @Tags 浠ュ浘鎼滃浘
 // @Param file formData file true "浜哄憳鍥剧墖"
@@ -120,7 +132,13 @@
 		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟鏈夎")
 		return
 	}
-	var weedfsUri = "http://"+config.WeedFs.Ip+":"+strconv.Itoa(config.WeedFs.UploadPort)+"/submit"
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.WebPicIp == "" {
+		logger.Debug("localConfig is wrong!!!")
+		return
+	}
+	var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
+	logger.Debug("weedfsUri:",weedfsUri)
 	resultMap :=make(map[string]interface{},0)
 	//灏嗕笂浼犵殑鍥剧墖浜や汉鑴告娴嬪拰浜鸿劯鎻愬彇锛岃幏寰楃壒寰�
 	fileBytes, _ := ioutil.ReadAll(file)
@@ -130,7 +148,11 @@
 		urlArr := make([]string,0)
 		for _,r := range faceArr {
 			rcFace := r.Pos.RcFace
-			cutFaceImgData := util.SubImg(*pI, int(rcFace.Left), int(rcFace.Top), int(rcFace.Right), int(rcFace.Bottom))
+			cutFaceImgData,cutErr := util.SubCutImg(pI, rcFace, 20)
+			if cutErr != nil {
+				logger.Debug("util.SubCutImg err:", cutErr)
+				continue
+			}
 			weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, "FaceUrl", cutFaceImgData)
 			if e == nil{
 				faceExtractedMap[weedFilePath] = FaceExtract{
@@ -141,7 +163,9 @@
 			}
 		}
 		//2.澶у浘鐢绘,鏍囪瘑浜鸿劯浣嶇疆
-		originFilePath, _ := WeedFSClient.UploadFile(weedfsUri, "FaceUrl", fileBytes)
+		drawedB, _ := drawPolygonOnImg(pI, &faceArr)
+
+		originFilePath, _ := WeedFSClient.UploadFile(weedfsUri, "FaceUrl", *drawedB)
 		resultMap["uploadImage"] = originFilePath
 		resultMap["smImage"] = urlArr
 		util.ResponseFormat(c,code.Success, resultMap)
@@ -150,158 +174,222 @@
 	}
 }
 
+func drawPolygonOnImg(i *protomsg.Image,faceArr *[]*protomsg.ResultFaceDetect) (*[]byte,error) {
+	img,err := godraw.ToImage(i.Data, int(i.Width), int(i.Height))
+	if err != nil {
+		return nil,err
+	}
+	red := color.RGBA{255, 0, 0, 0}
+	for _,faceResult := range *faceArr {
+		util.DrawRect(img.(*image.RGBA),int(faceResult.Pos.RcFace.Left),int(faceResult.Pos.RcFace.Top),int(faceResult.Pos.RcFace.Right),
+			int(faceResult.Pos.RcFace.Bottom), red)
+	}
+
+	quantity := 100
+	jpgData, err := godraw.ImageToJpeg(img, &quantity)
+	return &jpgData,err
+}
+
 type CompareResult struct {
 	Id string `json:"id"`
-	CompareScore float32 `json:"compareScore"`
+	CompareScore float64 `json:"compareScore"`
 	CameraId string `json:"cameraId"`
+	CameraName string `json:"cameraName"`
 	CameraAddr string `json:"cameraAddr"`
 	PicDate string `json:"picDate"`
 	Content string `json:"content"`
-	IsAlarm int `json:"isAlarm"`
-	PicMaxUrl string `json:"picMaxUrl"`
-	PicSmUrl []string `json:"picSmUrl"`
-	Sex string `json:"sex"`
-	AgeDescription string `json:"ageDescription"`
-	Race string `json:"race"`
+	IsAlarm bool `json:"isAlarm"`
+	PicMaxUrl []string `json:"picMaxUrl"`
+	TargetInfo []TargetInfo `json:"targetInfo"`
 	TaskId string `json:"taskId"`
 	TaskName string `json:"taskName"`
 	BaseInfo []DbPersonVo `json:"baseInfo"`
 	VideoUrl string `json:"videoUrl"`
 	SdkName string `json:"sdkName"`
+	AlarmRules []AlarmRuleVo `json:"alarmRules"`
 }
+
+type TargetInfo struct {
+	TargetId       string `json:"targetId"`
+	TargetType     string `json:"targetType"`
+	TargetScore    float64 `json:"targetScore"`
+	Feature        string `json:"feature"`
+	PicSmUrl       string `json:"picSmUrl"`
+	TargetLocation TargetLocation `json:"targetLocation"`
+}
+
+type TargetLocation struct {
+	TopLeft     protomsg.Location `json:"topLeft"`
+	BottomRight protomsg.Location `json:"bottomRight"`
+}
+
 type DbPersonVo struct {
-	BwType string `json:"bwType"`
-	CompareScore float32 `json:"compareScore"`
-	IdCard string `json:"idCard"`
-	MonitorLevel string `json:"monitorLevel"`
-	PersonId string `json:"personId"`
-	PersonName string `json:"personName"`
-	PersonPicUrl string `json:"personPicUrl"`
-	PhoneNum string `json:"phoneNum"`
-	Sex string `json:"sex"`
 	TableId string `json:"tableId"`
 	TableName string `json:"tableName"`
+	BwType string `json:"bwType"`
+	CompareScore float64 `json:"compareScore"`
+	TargetId string `json:"targetId"`
+	TargetName string `json:"targetName"`
+	TargetPicUrl string `json:"targetPicUrl"`
+	MonitorLevel string `json:"monitorLevel"`
+	Labels string `json:"labels"`
+	Content string `json:"content"`
 	Enable int32 `json:"enable"`
 }
 
-type CompareSortResult []CompareResult
+type ScoreIndex struct {
+	CompareScore float64
+	Index int
+}
 
-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
+type AlarmRuleVo struct {
+	GroupId      string `json:"groupId"`
+	AlarmLevel   string `json:"alarmLevel"`
+	RuleText     string `json:"ruleText"`
+	DefenceState bool `json:"defenceState"`
+	IsLink 		 bool `json:"isLink"`
+	LinkInfo 	 string `json:"linkInfo"`
 }
 
 //濉厖鍚戝墠绔繑鍥炵殑鏁版嵁
-func FillDataToCompareResult(compResult *protomsg.SdkCompareResult) CompareSortResult {
-	//resultList :=make([]CompareResult,0)
-	var resultList = CompareSortResult{}
+func FillDataToCompareResult(compResult *protomsg.SdkCompareResult) []CompareResult {
 
-	dbPersonM := make(map[string]*protomsg.SdkCompareEach, 0)
-	captureM := make(map[string]*protomsg.SdkCompareEach, 0)
+	var resultList = make([]CompareResult, len(compResult.CompareResult))
+	dbPersonM := make(map[string]ScoreIndex, 0)
+	captureM := make(map[string]ScoreIndex, 0)
 	personIds := make([]string,0)
 	captureIds := make([]string,0)
-	for _,v :=range compResult.CompareResult{
+	for idx,v :=range compResult.CompareResult{
 		if v.Tableid == service.CaptureTable {
-			captureM[v.Id] = v
+			captureM[v.Id] = ScoreIndex{
+				Index: idx,
+				CompareScore: util.ParseScore64(float64(v.CompareScore)),
+			}
 			captureIds = append(captureIds,v.Id)
 		} else {
-			dbPersonM[v.Id] = v
+			dbPersonM[v.Id] = ScoreIndex{
+				Index: idx,
+				CompareScore: util.ParseScore64(float64(v.CompareScore)),
+			}
 			personIds = append(personIds,v.Id)
 		}
 	}
 	logger.Debug("comp len(personIds):", len(personIds))
 	logger.Debug("comp len(captureIds):", len(captureIds))
-	esServerIp := config.EsInfo.Masterip
-	esServerPort := config.EsInfo.Httpport
-	index := config.EsInfo.EsIndex.Dbtablepersons.IndexName
+
 	var dbpersons []protomsg.Dbperson
 	if len(personIds) >0 {
-		dbpersons, _ = esApi.Dbpersoninfosbyid(personIds, index, esServerIp, esServerPort)
+		var dbpApi dbapi.DbPersonApi
+		dbpersons, _ = dbpApi.Dbpersoninfosbyid(personIds)
 	}
 
 	if len(dbpersons) >0 {
+		var dtApi dbapi.DbTableApi
 		for _,p :=range dbpersons {
 			var dbP = DbPersonVo {
-				PersonId: p.Id,
-				IdCard: p.IdCard,
-				CompareScore: util.ParseScore(dbPersonM[p.Id].CompareScore),
+				TargetId: p.Id,
+				CompareScore: dbPersonM[p.Id].CompareScore,
 				MonitorLevel: p.MonitorLevel,
-				PersonName: p.PersonName,
-				PersonPicUrl: p.PersonPicUrl,
-				PhoneNum: p.PhoneNum,
-				Sex: p.Sex,
+				TargetName: p.PersonName,
+				TargetPicUrl: p.PersonPicUrl,
+				Labels: p.PhoneNum+"/"+p.Sex+"/"+p.IdCard,
 				TableId: p.TableId,
+				Content: p.Reserved,
 				Enable: p.Enable,
 			}
-			dbTableInfos, _ := esApi.Dbtablefosbyid([]string{p.TableId}, config.EsInfo.EsIndex.DbTables.IndexName, esServerIp, esServerPort)
+			dbTableInfos, _ := dtApi.DbtablesById([]string{ p.TableId })
 			if dbTableInfos !=nil{
 				dbP.BwType = dbTableInfos[0].BwType
 				dbP.TableName = dbTableInfos[0].TableName
 			}
-			var cr = CompareResult{
+
+			resultList[dbPersonM[p.Id].Index] = CompareResult{
 				BaseInfo:[]DbPersonVo{ dbP },
 			}
-			resultList = append(resultList,cr)
 		}
 	}
 	if len(captureIds) >0 {
-		videopersons, _ := esApi.Videopersonsinfosbyid(captureIds, config.EsInfo.EsIndex.VideoPersons.IndexName, config.EsInfo.Masterip, config.EsInfo.Httpport)
-		logger.Debug("comp videoPersons.len:",len(videopersons))
-		for _,vp :=range videopersons {
-			isAlarmInt, _ := strconv.Atoi(vp.IsAlarm)
-			bi := make([]DbPersonVo,0)
+		localConf, err2 := cache.GetServerInfo()
+		if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
+			logger.Debug("localConfig is wrong!!!")
+			return nil
+		}
+		logger.Debug("captureIds:",strings.Join(captureIds,","))
+		aiTargets, _ := esApi.AIOceaninfosbyid(captureIds, config.EsInfo.EsIndex.AiOcean.IndexName, localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)))
+		logger.Debug("comp aiTargets.len:",len(aiTargets))
+		for _,vp :=range aiTargets {
+			var bi []DbPersonVo
 			for _,p :=range vp.BaseInfo {
 				bi = append(bi, DbPersonVo{
-					PersonId: p.PersonId,
-					IdCard: p.IdCard,
-					CompareScore: util.ParseScore(p.CompareScore),
-					MonitorLevel: parseMonitorLevel(p.MonitorLevel),
-					PersonName: p.PersonName,
-					PersonPicUrl: p.PersonPicUrl,
-					PhoneNum: p.PhoneNum,
-					Sex: p.Sex,
+					TargetId: p.TargetId,
+					CompareScore: util.ParseScore64(p.CompareScore),
+					MonitorLevel: p.MonitorLevel,
+					TargetName: p.TargetName,
+					TargetPicUrl: p.TargetPicUrl,
+					Labels: p.Labels,
 					TableId: p.TableId,
-					BwType: strconv.Itoa(int(p.BwType)),
+					BwType: p.BwType,
 					TableName: p.TableName,
 				})
 			}
+			var alarmRules []AlarmRuleVo
+			if vp.AlarmRules !=nil && len(vp.AlarmRules) >0 {
+				for _,ar :=range vp.AlarmRules {
+					alarmRules = append(alarmRules, AlarmRuleVo{
+						GroupId: ar.GroupId,
+						AlarmLevel: ar.AlarmLevel,
+						RuleText: ar.RuleText,
+						DefenceState: ar.DefenceState,
+						IsLink: ar.IsLink,
+						LinkInfo: ar.LinkInfo,
+					})
+				}
+			}
+			var ti = make([]TargetInfo,0)
+			if vp.TargetInfo !=nil {
+				for _,vti :=range vp.TargetInfo {
+					tl := protomsg.Location{
+						X: vti.TargetLocation.TopLeft.X,
+						Y: vti.TargetLocation.TopLeft.Y,
+					}
+					br := protomsg.Location{
+						X: vti.TargetLocation.BottomRight.X,
+						Y: vti.TargetLocation.BottomRight.Y,
+					}
+					tInfo := TargetInfo{
+						TargetId: vti.TargetId,
+						TargetType: vti.TargetType,
+						TargetScore: vti.TargetScore,
+						Feature: vti.Feature,
+						PicSmUrl: vti.PicSmUrl,
+						TargetLocation: TargetLocation{
+							TopLeft:tl,
+							BottomRight:br,
+						},
+					}
+					ti = append(ti, tInfo)
+				}
+			}
 			vpE := CompareResult{
 				Id: vp.Id,
-				CompareScore: util.ParseScore(captureM[vp.Id].CompareScore),
+				CompareScore: captureM[vp.Id].CompareScore,
 				CameraId: vp.CameraId,
+				CameraName: vp.CameraName,
 				CameraAddr: vp.CameraAddr,
 				PicDate: vp.PicDate,
 				PicMaxUrl: vp.PicMaxUrl,
-				PicSmUrl: vp.PicSmUrl,
-				IsAlarm: isAlarmInt,
-				Sex: vp.Sex,
-				AgeDescription: vp.AgeDescription,
-				Race: vp.Race,
+				TargetInfo: ti,
+				IsAlarm: vp.IsAlarm,
 				TaskName: vp.TaskName,
 				TaskId: vp.TaskId,
 				VideoUrl: vp.VideoUrl,
 				BaseInfo: bi,
 				SdkName: "浜鸿劯",
+				AlarmRules: alarmRules,
 			}
-			resultList = append(resultList,vpE)
+			resultList[captureM[vp.Id].Index] = vpE
 		}
 	}
-	var scorePre []float32
-	for _,f1 :=range resultList {
-		scorePre = append(scorePre, f1.CompareScore)
-	}
-	logger.Debug("comp 鎺掑簭鍓嶅垎鍊兼暟缁勶細",scorePre)
-	sort.Sort(resultList)
-	var scoreAf []float32
-	for _,f2 :=range resultList {
-		scoreAf = append(scoreAf, f2.CompareScore)
-	}
-	logger.Debug("comp 鎺掑簭鍚庡垎鍊兼暟缁勶細",scoreAf)
 	return  resultList
 }
 
@@ -318,6 +406,37 @@
 	return level
 }
 
+func GetFeaFromOneFaceImg(fileBytes []byte) (faceB []byte,oriImg string,smImgs []string, err error){
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.WebPicIp == "" {
+		logger.Debug("localConfig is wrong!!!")
+		return nil,"",nil,errors.New("localConf err")
+	}
+	var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
+	logger.Debug("weedfsUri:",weedfsUri)
+	faceArr, err, pI := service.GetFaceFeaFromSdk(fileBytes, time.Second*60)
+	if err ==nil && len(faceArr) ==1 {
+		rcFace := faceArr[0].Pos.RcFace
+		cutFaceImgData,_ := util.SubCutImg(pI, rcFace, 20)
+		weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, "FaceUrl", cutFaceImgData)
+		if e == nil{
+			faceExtractedMap[weedFilePath] = FaceExtract{
+				Url:weedFilePath,
+				FaceBytes:faceArr[0].Feats,
+			}
+			smImgs = append(smImgs, weedFilePath)
+		}
+		//2.澶у浘鐢绘,鏍囪瘑浜鸿劯浣嶇疆
+		drawedB, _ := drawPolygonOnImg(pI, &faceArr)
+
+		originFilePath, _ := WeedFSClient.UploadFile(weedfsUri, "FaceUrl", *drawedB)
+		return faceArr[0].Feats,originFilePath,smImgs,nil
+	} else {
+		return nil,"",nil,errors.New("no face")
+	}
+}
+
+// @Security ApiKeyAuth
 // @Summary 浠ュ浘鎼滃浘
 // @Description  浠ュ浘鎼滃浘
 // @Accept json
@@ -327,49 +446,73 @@
 // @Success 200 {string} json "{"code":200, msg:"", data:"", success:true}"
 // @Failure 500 {string} json "{"code":500, msg:"", data:"", success:false}"
 // @Router /data/api-v/dbperson/searchByPhoto [POST]
-func (controller FileController) SearchByPhoto(c *gin.Context) {
+func (fc FileController) SearchByPhoto(c *gin.Context) {
 	var searchBody models.EsSearch
 	err := c.BindJSON(&searchBody)
 	if err !=nil{
 		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟鏈夎")
 		return
 	}
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
+		logger.Debug("localConfig is wrong!!!")
+		util.ResponseFormat(c,code.ComError,"localConf wrong")
+		return
+	}
 	var faceB []byte
-	if searchBody.CaptureId != "" {//鍋氭煡鎵炬浜�,鎼滄墍鏈夋姄鎷嶅拰搴曞簱
-		searchBase64Fea, err := esApi.GetVideoPersonFaceFeatureById(searchBody.CaptureId, config.EsInfo.EsIndex.VideoPersons.IndexName, config.EsInfo.Masterip, config.EsInfo.Httpport)
-		if err !=nil {
-			util.ResponseFormat(c, code.ComError, "鎶撴媿鏁版嵁涓嶅瓨鍦紝璇锋鏌�")
-			return
-		}
-		if searchBase64Fea !=""{
-			decodeF, err := base64.StdEncoding.DecodeString(searchBase64Fea)
-			if err !=nil {
-				util.ResponseFormat(c, code.ComError, "鏈潯鎶撴�曠壒寰佷笉鏄痓ase64锛岃妫�鏌�")
+	logger.Debug("compTargetId:",searchBody.CompTargetId,",compTargetType:",searchBody.CompTargetType)
+	if searchBody.CompTargetId != "" {//鍋氭煡鎵炬浜�,鎸夋姄鎷嶇殑浜鸿劯鎴栬�呭簳搴撶殑浜鸿劯浠ュ浘鎼滃浘
+		if searchBody.CompTargetType == 0 {//鏈紶浜鸿劯鏄簳搴撲汉鑴�
+			var dbpApi dbapi.DbPersonApi
+			dbPersons, e := dbpApi.Dbpersoninfosbyid([]string{searchBody.CompTargetId})
+			if e ==nil && dbPersons !=nil && len(dbPersons) ==1 {
+				searchBase64Fea := dbPersons[0].FaceFeature
+				if searchBase64Fea != "" {
+					decodeF, err := base64.StdEncoding.DecodeString(dbPersons[0].FaceFeature)
+					if err !=nil {
+						util.ResponseFormat(c, code.ComError, "鏈潯搴曞簱浜哄憳鐗瑰緛涓嶆槸base64锛岃妫�鏌�")
+						return
+					}
+					faceB = decodeF
+				} else {
+					util.ResponseFormat(c, code.ComError, "鏈潯搴曞簱浜哄憳鐗瑰緛涓虹┖锛岃妫�鏌�")
+					return
+				}
+			} else {
+				util.ResponseFormat(c, code.ComError, "搴曞簱浜哄憳鏌ヨ澶辫触锛岃妫�鏌�")
 				return
 			}
-			faceB = decodeF
+		} else {
+			searchBase64Fea, err := esApi.GetVideoPersonFaceFeatureById(searchBody.CompTargetId, config.EsInfo.EsIndex.AiOcean.IndexName, localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)))
+			if err !=nil {
+				util.ResponseFormat(c, code.ComError, "鎶撴媿鏁版嵁涓嶅瓨鍦紝璇锋鏌�")
+				return
+			}
+
+			if searchBase64Fea !=""{
+				decodeF, err := base64.StdEncoding.DecodeString(searchBase64Fea)
+				if err !=nil {
+					util.ResponseFormat(c, code.ComError, "鏈潯鎶撴�曠壒寰佷笉鏄痓ase64锛岃妫�鏌�")
+					return
+				}
+				faceB = decodeF
+			} else {
+				util.ResponseFormat(c, code.ComError, "鏈潯浜鸿劯鎶撴媿鐗瑰緛涓虹┖锛岃妫�鏌�")
+				return
+			}
 		}
+
 	} else {//鍋氫互鍥炬悳鍥�
 		if searchBody.PicUrl == "" || len(searchBody.DataBases) == 0 {
-			util.ResponseFormat(c, code.RequestParamError, "鍙傛暟鏈夎")
+			util.ResponseFormat(c, code.RequestParamError, "浠ュ浘鎼滃浘PicUrl涓嶈兘涓虹┖")
 			return
 		}
 		if face,ok := faceExtractedMap[searchBody.PicUrl];!ok{
-			util.ResponseFormat(c, code.RequestParamError, "璇烽噸鏂颁笂浼犲浘鐗�")
+			util.ResponseFormat(c, code.RequestParamError, "浜鸿劯鐗瑰緛鏈娴嬶紝璇烽噸鏂颁笂浼犲浘鐗�")
 			return
 		} else {
 			faceB = face.FaceBytes
 		}
-	}
-
-	var sysSetApi dbapi.SysSetApi
-	analyServerId := ""
-	flag, sysconf := sysSetApi.GetServerInfo()
-	if flag {
-		analyServerId = sysconf.ServerId
-	} else {
-		util.ResponseFormat(c, code.ComError, "analyServerId涓虹┖锛岄厤缃湁璇�")
-		return
 	}
 
 	if faceB == nil {
@@ -381,11 +524,12 @@
 		CompareThreshold: searchBody.Threshold,
 	}
 
+	var hasCompEsPerson = false
 	if searchBody.DataBases !=nil {
 		for idx,tableId :=range searchBody.DataBases {
 			if tableId == "esData" {
 				searchBody.DataBases = append(searchBody.DataBases[:idx], searchBody.DataBases[idx+1:]...)
-				searchBody.DataBases = append(searchBody.DataBases,service.CaptureTable)
+				hasCompEsPerson = true
 				break
 			}
 		}
@@ -393,48 +537,54 @@
 	} else {
 		arg.TableIds = []string{}
 	}
+	alarmLevelTypes := strings.Replace(strings.Trim(fmt.Sprint(getAlarmLevel(searchBody.AlarmLevel)), "[]"), " ", "\",\"", -1)
 	arg.Source = true // 鏍囪瘑鏉ユ簮鏄痺eb
-	arg.AlarmLevel = searchBody.AlarmLevel
+	arg.AlarmLevel = alarmLevelTypes
 	arg.Tasks = searchBody.Tasks
 	arg.TreeNodes = searchBody.TreeNodes
 	arg.Tabs = searchBody.Tabs
 	arg.SearchTime = searchBody.SearchTime
 	arg.InputValue = searchBody.InputValue
 	arg.Collection = searchBody.Collection
-	arg.AnalyServerId = analyServerId
+	arg.AnalyServerId = localConf.ServerId
 
 	logger.Debug("arg.TableIds:", arg.TableIds, ",alarmLevel:",arg.AlarmLevel,",treeNodes:",arg.TreeNodes,",searchTime:",arg.SearchTime,
 		",inputValue:",arg.InputValue,",tasks:",arg.Tasks,",compThreshold:",arg.CompareThreshold)
 	timeStart := time.Now()
 
 	compareService := service.NewFaceCompareService(arg)
-	compareOnce := compareService.Compare()
-	totalData := compareOnce.CompareData
+	var totalData service.CompareList
+	if len(arg.TableIds) >0 {//鏈夋瘮瀵瑰簳搴�
+		dbPersonTargets := compareService.CompareDbPersons()
+		if dbPersonTargets !=nil {
+			totalData = append(totalData,*dbPersonTargets...)
+		}
+	}
+	if hasCompEsPerson {//鏈夋瘮瀵笶s鎶撴媿
+		esPersons := compareService.CompareVideoPersons()
+		if esPersons !=nil {
+			totalData = append(totalData, *esPersons...)
+		}
+	}
 
 	logger.Debug("comp 姣斿缁撴潫锛岀敤鏃讹細",time.Since(timeStart))
+	service.SetCompResultByNum(&service.CompareOnce{
+		CompareNum: compareService.CompareNum,
+		CompareData: &totalData,
+	})
+
 	m := make(map[string]interface{},3)
 	if totalData != nil && totalData.Len() > 0{
-		//鎸夊垎鍊兼帓搴�
-		//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["compareNum"] = compareService.CompareNum
 		m["total"] = total
 		var sCompResult protomsg.SdkCompareResult
 		if total <= searchBody.Size {
-			sCompResult.CompareResult = *totalData
+			sCompResult.CompareResult = totalData
 		} else {
-			sCompResult.CompareResult = (*totalData)[0:searchBody.Size]
+			sCompResult.CompareResult = totalData[0:searchBody.Size]
 		}
 		resultList := FillDataToCompareResult(&sCompResult)
 		m["totalList"] = resultList
@@ -442,82 +592,143 @@
 
 	} else {
 		m["total"] = 0
-		m["compareNum"] = searchBody.CompareNum
+		m["compareNum"] = compareService.CompareNum
 		m["totalList"] = []CompareResult{}
 	}
 	util.ResponseFormat(c,code.Success,m)
 }
 
-func GetCompareDataTwice(co *service.CompareOnce,arg *models.EsSearch) map[string]interface{} {
-	//camIds := arg.TreeNodes
-	//tabIds := arg.Tabs
-	//taskIds := arg.Tasks
-	//alarmLevels := arg.AlarmLevel
-	//timeArr := arg.SearchTime
-	//input := arg.InputValue
-	//resultList := make([]CompareResult,0)
-	//for _,each :=range *co.CompareData {
-	//	if each.Id !=""{
-	//		n :=0
-	//		if len(camIds) == 0 || isInArr(each.CameraId,camIds) {
-	//			n++
-	//		}
-	//		if len(tabIds) == 0 {
-	//			n++
-	//		}
-	//		if len(taskIds) == 0 || isInArr(each.TaskId, taskIds) {
-	//			n++
-	//		}
-	//		//鍒ゆ柇鎶ヨ绛夌骇
-	//		if len(alarmLevels) == 0 {
-	//
-	//		}
-	//		n++
-	//		//鍒ゆ柇鏃堕棿
-	//		if len(timeArr) == 0 {
-	//
-	//		}
-	//		n++
-	//		if input == ""{
-	//
-	//		}
-	//		n++
-	//		if n == 6 {
-	//			resultList = append(resultList,cr)
-	//		}
-	//	} else {
-	//		resultList = append(resultList,cr)
-	//	}
-	//}
+type PersonId struct {
+	Id string `json:"id"`
+}
 
-	//logger.Debug("GetCompareDataTwice.data:",resultList)
-	//service.SortByScore(resultList)
-	from := (arg.Page-1)*arg.Size
-	to := from + arg.Size
-	var sCompResult protomsg.SdkCompareResult
-	total := len(*co.CompareData)
-	if total <= to {
-		sCompResult.CompareResult = (*co.CompareData)[from:total]
-	} else {
-		sCompResult.CompareResult = (*co.CompareData)[from:to]
-	}
-	resultList := FillDataToCompareResult(&sCompResult)
-
+func GetCompareDataTwice(co *service.CompareOnce,searchBody *models.EsSearch) map[string]interface{} {
 	m := make(map[string]interface{},0)
-	m["total"] = total
-	m["compareNum"] = arg.CompareNum
-	m["totalList"] = resultList
-	//if from < len(resultList) {
-	//	if to <= len(resultList) {
-	//		m["totalList"] = resultList[from:to]
-	//	} else {
-	//		lte := len(resultList)
-	//		m["totalList"] = resultList[from:lte]
-	//	}
+
+	from := (searchBody.Page-1)*searchBody.Size
+	to := from + searchBody.Size
+
+	var hasCompEsPerson = false
+	if searchBody.DataBases !=nil {
+		for idx,tableId :=range searchBody.DataBases {
+			if tableId == "esData" {
+				hasCompEsPerson = true
+				searchBody.DataBases = append(searchBody.DataBases[:idx], searchBody.DataBases[idx+1:]...)
+				break
+			}
+		}
+	}
+	if hasCompEsPerson {//浜屾妫�绱S姣斿缁撴灉
+		arg := protomsg.CompareArgs{
+			CompareThreshold: searchBody.Threshold,
+		}
+		localConf, err := cache.GetServerInfo()
+		if err ==nil && localConf.AlarmIp != "" && localConf.ServerId != "" {
+			alarmLevelTypes := strings.Replace(strings.Trim(fmt.Sprint(getAlarmLevel(searchBody.AlarmLevel)), "[]"), " ", "\",\"", -1)
+			arg.Source = true // 鏍囪瘑鏉ユ簮鏄痺eb
+			arg.AlarmLevel = alarmLevelTypes
+			arg.Tasks = searchBody.Tasks
+			arg.TreeNodes = searchBody.TreeNodes
+			arg.Tabs = searchBody.Tabs
+			arg.SearchTime = searchBody.SearchTime
+			arg.InputValue = searchBody.InputValue
+			arg.Collection = searchBody.Collection
+			arg.AnalyServerId = localConf.ServerId
+
+			captureIds := esApi.GetAllLocalVideopersonsId(arg, config.EsInfo.EsIndex.AiOcean.IndexName, localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)), alarmLevelTypes)
+			logger.Debug("searchPhoto first Result.len:",len(*co.CompareData),"twice len(captureIds):",len(captureIds))
+			if captureIds !=nil {
+				var aResult protomsg.SdkCompareResult
+				aList := getTwiceSearchResult(co, &captureIds, searchBody)
+				aTotal := aList.Len()
+				if aTotal <= to {
+					aResult.CompareResult = (*aList)[from:aTotal]
+				} else {
+					aResult.CompareResult = (*aList)[from:to]
+				}
+				out := FillDataToCompareResult(&aResult)
+				m["total"] = aTotal
+				m["compareNum"] = searchBody.CompareNum
+				m["totalList"] = out
+				return m
+			} else {
+				m["total"] = 0
+				m["compareNum"] = searchBody.CompareNum
+				m["totalList"] = []interface{}{}
+				return m
+			}
+		}
+	} else {//搴曞簱鏁版嵁鐨勪簩娆℃绱�
+		var dbpApi dbapi.DbPersonApi
+		personIds, _ := dbpApi.FindLikePersonIds(searchBody.DataBases, searchBody.InputValue)
+		logger.Debug("searchPhoto first Result.len:",len(*co.CompareData),"personIds:",personIds)
+		if personIds !=nil {
+			var pIds []PersonId
+			b, _ := json.Marshal(personIds)
+			json.Unmarshal(b, &pIds)
+			if len(pIds) >0 {
+				var personIdArr []string
+				for _,pid :=range pIds{
+					personIdArr = append(personIdArr, pid.Id)
+				}
+				var aResult protomsg.SdkCompareResult
+				aList := getTwiceSearchResult(co, &personIdArr, searchBody)
+				aTotal := aList.Len()
+				if aTotal <= to {
+					aResult.CompareResult = (*aList)[from:aTotal]
+				} else {
+					aResult.CompareResult = (*aList)[from:to]
+				}
+				out := FillDataToCompareResult(&aResult)
+				m["total"] = aTotal
+				m["compareNum"] = searchBody.CompareNum
+				m["totalList"] = out
+				return m
+			}
+
+		} else {
+			m["total"] = 0
+			m["compareNum"] = searchBody.CompareNum
+			m["totalList"] = []interface{}{}
+			return m
+		}
+	}
+
+
+	//var sCompResult protomsg.SdkCompareResult
+	//total := len(*co.CompareData)
+	//if total <= to {
+	//	sCompResult.CompareResult = (*co.CompareData)[from:total]
 	//} else {
-	//	m["totalList"] = []CompareResult{}
+	//	sCompResult.CompareResult = (*co.CompareData)[from:to]
 	//}
+	//resultList := FillDataToCompareResult(&sCompResult)
+
+
+	m["total"] = 0
+	m["compareNum"] = searchBody.CompareNum
+	m["totalList"] = []interface{}{}
+
 	return m
+}
+
+func getTwiceSearchResult(co *service.CompareOnce, scopeIds *[]string, searchBody *models.EsSearch) *service.CompareList{
+	m := make(map[string]string)
+	for _,capId :=range *scopeIds {
+		m[capId] = capId
+	}
+	var totalData service.CompareList
+	for _,each :=range *co.CompareData {
+		if _,ok :=m[each.Id];ok && each.CompareScore >= searchBody.Threshold {
+			totalData = append(totalData, each)
+		}
+	}
+
+	if totalData != nil && totalData.Len() > 0{
+		sort.Sort(totalData)
+	}
+
+	return &totalData
 }
 
 func isInArr(id string,arr []string) bool {
@@ -538,7 +749,12 @@
 		util.ResponseFormat(c, code.RequestParamError, "鍙傛暟鏈夎")
 		return
 	}
-	var weedfsUri = "http://"+config.WeedFs.Ip+":"+strconv.Itoa(config.WeedFs.UploadPort)+"/submit"
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.WebPicIp == "" {
+		logger.Debug("localConfig is wrong!!!")
+		return
+	}
+	var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
 
 	//灏嗕笂浼犵殑鍥剧墖浜や汉鑴告娴嬪拰浜鸿劯鎻愬彇锛岃幏寰楃壒寰�
 	fileBytes, _ := ioutil.ReadAll(file)
@@ -552,7 +768,7 @@
 			break
 		}
 		//鏍规嵁浜鸿劯鍧愭爣鎵e嚭浜鸿劯灏忓浘
-		cutFaceImgData := util.SubImg(*pI, int(rcFace.Left), int(rcFace.Top), int(rcFace.Right), int(rcFace.Bottom))
+		cutFaceImgData,_ := util.SubCutImg(pI, rcFace, 20)
 		weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, "testCutFace", cutFaceImgData)
 		if e !=nil{
 			util.ResponseFormat(c,code.ComError,"鏂囦欢涓婁紶澶辫触")
@@ -570,25 +786,24 @@
 log.Fatalln(err)
 }*/
 
-func uploadFileReturnAddr(file multipart.File, filename string, tableId string) (string, map[string]interface{}, error) {
+func uploadFileReturnAddr(file multipart.File, filename string, tableId string, add2Db bool) (string, map[string]interface{}, error, string, string) {
 	defer file.Close()
-	field := ""
 	// weedfs 涓婁紶
 	fileBytes, err := ioutil.ReadAll(file)
 	if err !=nil {
-		return "",nil,err
+		return "",nil,err,"",""
 	}
 
 	//灏嗕笂浼犵殑鍥剧墖浜や汉鑴告娴嬪拰浜鸿劯鎻愬彇锛岃幏寰楃壒寰�
 	var faceBase64=""
 	faceArr, err, pI := service.GetFaceFeaFromSdk(fileBytes, time.Second*5)
 	if faceArr ==nil {
-		return "",nil,errors.New("NotFeatureFindError")
+		return "",nil,errors.New("NotFeatureFindError"),"",""
 	}
 	var rcFace *protomsg.Rect
 	if err ==nil && len(faceArr) >0 {
 		if len(faceArr) >1 {
-			return "",nil, errors.New("TooManyFaces")
+			return "",nil, errors.New("TooManyFaces"),"",""
 		}
 		for _,r := range faceArr {
 			//鎷垮埌浜鸿劯鐨勫潗鏍�
@@ -598,48 +813,55 @@
 			break
 		}
 	}
-	var weedfsUri = "http://"+config.WeedFs.Ip+":"+strconv.Itoa(config.WeedFs.UploadPort)+"/submit"
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.WebPicIp == "" {
+		logger.Debug("localConfig is wrong!!!")
+		return "",nil,err2,"",""
+	}
+	var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
 	//鏍规嵁浜鸿劯鍧愭爣鎵e嚭浜鸿劯灏忓浘
 	t1 := time.Now()
-	cutFaceImgData := util.SubImg(*pI, int(rcFace.Left), int(rcFace.Top), int(rcFace.Right), int(rcFace.Bottom))
+	cutFaceImgData,cutErr := util.SubCutImg(pI, rcFace, 20)
+	if cutErr != nil {
+		logger.Debug("util.SubCutImg err:", cutErr)
+		return "", nil, cutErr,"",""
+	}
 	logger.Debug("SubImg鐢ㄦ椂锛�", time.Since(t1))
 	t1 = time.Now()
-	fileInfo, e := esutil.PostFormData(weedfsUri, filename, "file", cutFaceImgData)
+	weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, filename, cutFaceImgData)
 	logger.Debug("涓婁紶鍒皐eedfs鐢ㄦ椂:", time.Since(t1))
 	t1 = time.Now()
 	if e != nil {
-		fmt.Println(e.Error())
-		return "", nil, e
-	} else {
-		field = fileInfo[picUrlField].(string) // 鏂囦欢璺緞
+		logger.Debug("WeedFSClient.UploadFile err:", e)
+		return "", nil, e,"",""
 	}
-	if strings.Contains(field,"/"){
-		idx := strings.Index(field, "/")
-		field = field[idx+1:]
-	}
+
 	ext := path.Ext(filename)
 	fileNameOnly := strings.TrimSuffix(filename, ext)
 
-	dbperson := new(models.Dbtablepersons)
-	dbperson.PersonPicUrl = field //  鍥剧墖璺粡
-	dbperson.TableId = tableId           //
-	dbperson.PersonName = fileNameOnly       // 鍥剧墖鍚�
-	dbperson.Enable = 1 //榛樿鏈夋晥
-	// 婕旂ずbase64缂栫爜
+	var result map[string]interface{}
+	if add2Db {
+		dbperson := new(models.Dbtablepersons)
+		dbperson.PersonPicUrl = weedFilePath //  鍥剧墖璺粡
+		dbperson.TableId = tableId           //
+		dbperson.PersonName = fileNameOnly       // 鍥剧墖鍚�
+		dbperson.Enable = 1 //榛樿鏈夋晥
+		dbperson.FaceFeature = faceBase64 // 鐗瑰緛鍊糱ase64 鐮�
 
-	dbperson.FaceFeature = faceBase64 // 鐗瑰緛鍊糱ase64 鐮�
+		result = addDbPerson(dbperson)
+	}
 
-	result := addDbPerson(dbperson)
-	logger.Debug("addDbPerson鐢ㄦ椂锛�", time.Since(t1))
-	return fileNameOnly, result, nil
+	logger.Debug("addDbPerson鐢ㄦ椂锛�", time.Since(t1), "result:", result)
+	return fileNameOnly, result, nil,faceBase64, weedFilePath
 
 }
 
 /**涓婁紶鏂规硶**/
 
+// @Security ApiKeyAuth
 // @Summary 鎵归噺娣诲姞搴曞簱浜哄憳
 // @Description  渚濇嵁鍥剧墖鎵归噺娣诲姞搴曞簱浜哄憳
-// @Accept  mpfd
+// @Accept  multipart/form-data
 // @Produce json
 // @Tags dbperson 搴曞簱浜哄憳
 // @Param files formData file[] true "澶氫釜搴曞簱浜哄憳鍥剧墖"
@@ -659,8 +881,10 @@
 	}
 	extNames := make([]string, 0)
 	addResult := make(map[string]interface{}, 0)
-	successList := make([]string, 0)
+	successList := make([]string,0)
 	failList := make([]string,0)
+	noFaceList := make([]string,0)
+	multiFaceList := make([]string,0)
 	tAllStart := time.Now()
 	var wg sync.WaitGroup
 	var lock sync.Mutex
@@ -670,18 +894,31 @@
 			defer wg.Done()
 			tIStart := time.Now()
 			filename := head.Filename
+			fileExt := path.Ext(filename)
+			fileExt = strings.ToLower(fileExt)
+			if fileExt !=".jpg" && fileExt != ".jpeg" && fileExt != ".png" {
+				lock.Lock()
+				failList = append(failList, filename)
+				lock.Unlock()
+				return
+			}
 			file, err := head.Open()
-			fmt.Println(file, err, filename)
 			if err != nil {
 				lock.Lock()
 				failList = append(failList, filename)
 				lock.Unlock()
 				return
 			}
-			field, _, err1 := uploadFileReturnAddr(file, filename, tableId)
+			field, _, err1,_,_ := uploadFileReturnAddr(file, filename, tableId, true)
 			lock.Lock()
 			if err1 != nil || field == "" {
-				failList = append(failList, filename)
+				if err1 !=nil && err1.Error() == "NotFeatureFindError" {
+					noFaceList = append(noFaceList, filename)
+				} else if err1 !=nil && err1.Error() == "TooManyFaces" {
+					multiFaceList = append(multiFaceList, filename)
+				} else {
+					failList = append(failList, filename)
+				}
 			} else {
 				successList = append(successList, filename)
 			}
@@ -693,6 +930,8 @@
 	logger.Debug("鍒�",len(fileHeaders),"寮犱汉鑴哥敤鏃讹細", time.Since(tAllStart))
 	addResult["successList"] = successList
 	addResult["failList"] = failList
+	addResult["noFaceList"] = noFaceList
+	addResult["multiFaceList"] = multiFaceList
 	addResult["fields"] = extNames
 
 	if len(successList)>0 {
@@ -710,9 +949,10 @@
 	IdCard string `json:"idCard"`
 }
 
+// @Security ApiKeyAuth
 // @Summary 涓婁紶鍥剧墖 骞跺垏鍥�
 // @Description  涓婁紶鍥剧墖 骞跺垏鍥�
-// @Accept  mpfd
+// @Accept  multipart/form-data
 // @Produce json
 // @Tags dbperson 搴曞簱浜哄憳
 // @Param file formData file true "搴曞簱浜哄憳鍥剧墖"
@@ -741,7 +981,12 @@
 		util.ResponseFormat(c, code.UploadFileError, err2.Error())
 		return
 	}
-	var weedfsUri = "http://"+config.WeedFs.Ip+":"+strconv.Itoa(config.WeedFs.UploadPort)+"/submit"
+	localConf, err2 := cache.GetServerInfo()
+	if err2 !=nil || localConf.WebPicIp == "" {
+		logger.Debug("localConfig is wrong!!!")
+		return
+	}
+	var weedfsUri = "http://"+localConf.WebPicIp+":"+strconv.Itoa(int(localConf.WebPicPort))+"/submit?collection=persistent"
 	{
 		uri := weedfsUri
 		fileInfo, e := esutil.PostFormBufferData(uri, filename, "file", uploadData)

--
Gitblit v1.8.0