update ynSwagger,sync to master

liuxiaolong

2019-11-13 f26a0cab5bca17b7eab57f6330e576271e17a17f

 controllers/dbtableperson.go

@@ -2,8 +2,14 @@

import (
   "basic.com/dbapi.git"
   "basic.com/fileServer/WeedFSClient.git"
   "basic.com/pubsub/protomsg.git"
   "encoding/base64"
   "encoding/json"
   "io/ioutil"
   "sort"
   "strconv"
   "time"
   "webserver/extend/logger"

   "github.com/gin-gonic/gin"
@@ -13,19 +19,24 @@
   "webserver/extend/esutil"
   "webserver/extend/util"
   "webserver/models"
   "webserver/cache"
   "webserver/service"

   esApi "basic.com/pubsub/esutil.git"
)

type DbPersonController struct {
}

// @Security ApiKeyAuth
// @Summary 添加底库人员
// @Description 添加底库人员
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param obj body models.Dbtablepersons true "底库人员数据"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"返回错误信息", success:false}"
// @Router /data/api-v/dbperson/addDbPerson [PUT]
func (dbc DbPersonController) AddDbPerson(c *gin.Context) {
   dbperson := new(models.Dbtablepersons)
@@ -71,6 +82,7 @@
   return result
}

// @Security ApiKeyAuth
// @Summary 修改底库人员
// @Description 修改底库人员
// @Accept  json
@@ -99,7 +111,12 @@
}

func UpdateDbPersonsOfDbTable(id string) (message string) {
   url := "http://" + config.EsInfo.Masterip + ":" + config.EsInfo.Httpport +
   localConf, err2 := cache.GetServerInfo()
   if err2 !=nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
      logger.Debug("localConfig is wrong!!!")
      return "修改失败"
   }
   url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) +
      "/" + config.EsInfo.EsIndex.Dbtablepersons.IndexName + "/_update_by_query?refresh"
   jsonDSL := `
         {
@@ -138,6 +155,222 @@
   return message
}

// @Security ApiKeyAuth
// @Summary 底库人员以图搜图
// @Description 底库人员以图搜图
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param condition body models.EsSearch true "底库以图搜图参数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/dbperson/queryDbPersonsByCompare [POST]
func (dbc DbPersonController) QueryDbPersonsByCompare(c *gin.Context) {
   var searchBody models.EsSearch
   err := c.BindJSON(&searchBody)
   if err !=nil || searchBody.PicUrl == "" || len(searchBody.DataBases) == 0{
      util.ResponseFormat(c, code.RequestParamError, "参数有误")
      return
   }
   var faceB []byte
   if face,ok := faceExtractedMap[searchBody.PicUrl];!ok{
      util.ResponseFormat(c, code.RequestParamError, "请重新上传图片")
      return
   } else {
      faceB = face.FaceBytes
   }

   analyServerId := ""
   conf, e := cache.GetServerInfo()
   if e ==nil && conf.ServerId != "" {
      analyServerId = conf.ServerId
   } else {
      util.ResponseFormat(c, code.ComError, "analyServerId为空，配置有误")
      return
   }

   arg := protomsg.CompareArgs{
      FaceFeature: faceB,
      CompareThreshold: searchBody.Threshold,
   }
   arg.TableIds = searchBody.DataBases
   arg.AnalyServerId = analyServerId
   compareService := service.NewFaceCompareService(arg)
   var totalData service.CompareList

   dbPersonTargets := compareService.CompareDbPersons()
   if dbPersonTargets !=nil {
      totalData = append(totalData,*dbPersonTargets...)
   }

   service.SetCompResultByNum(&service.CompareOnce{
      CompareNum: compareService.CompareNum,
      CompareData: &totalData,
   })

   m := make(map[string]interface{},3)
   if totalData != nil && totalData.Len() > 0{
      sort.Sort(totalData)
      total := totalData.Len()

      m["compareNum"] = compareService.CompareNum
      m["total"] = total
      var sCompResult protomsg.SdkCompareResult
      if total <= searchBody.Size {
         sCompResult.CompareResult = totalData
      } else {
         sCompResult.CompareResult = totalData[0:searchBody.Size]
      }
      resultList := FillDbPersonDataToCompareResult(&sCompResult)
      m["totalList"] = resultList

   } else {
      m["total"] = 0
      m["compareNum"] = compareService.CompareNum
      m["totalList"] = []CompareResult{}
   }
   util.ResponseFormat(c,code.Success,m)
}

//填充向前端返回的数据
func FillDbPersonDataToCompareResult(compResult *protomsg.SdkCompareResult) []models.DbPersonsCompVo {

   var resultList = make([]models.DbPersonsCompVo, len(compResult.CompareResult))
   dbPersonM := make(map[string]ScoreIndex, 0)
   personIds := make([]string,0)

   for idx,v :=range compResult.CompareResult{
      dbPersonM[v.Id] = ScoreIndex{
         Index: idx,
         CompareScore: v.CompareScore,
      }
      personIds = append(personIds,v.Id)
   }
   logger.Debug("comp len(personIds):", len(personIds))

   var dbpersons []protomsg.Dbperson
   if len(personIds) >0 {
      var dbpApi dbapi.DbPersonApi
      dbpersons, _ = dbpApi.Dbpersoninfosbyid(personIds)
   }

   if len(dbpersons) >0 {
      //var dtApi dbapi.DbTableApi
      for _,p :=range dbpersons {
         var dbP models.DbPersonsCompVo

         dbP.Id = p.Id
         dbP.TableId = p.TableId
         dbP.FaceFeature = p.FaceFeature
         dbP.PersonPicUrl = p.PersonPicUrl
         dbP.PersonName = p.PersonName
         dbP.Age = p.Age
         dbP.Sex = p.Sex
         dbP.IdCard = p.IdCard
         dbP.PhoneNum = p.PhoneNum
         dbP.MonitorLevel = p.MonitorLevel
         dbP.Reserved = p.Reserved
         dbP.IsDelete = int(p.IsDelete)
         dbP.Enable = int(p.Enable)
         dbP.CreateTime = p.CreateTime
         dbP.UpdateTime = p.UpdateTime
         dbP.CreateBy = p.CreateBy
         dbP.CompareScore = util.ParseScore(dbPersonM[p.Id].CompareScore)
         //dbTableInfos, _ := dtApi.DbtablesById([]string{ p.TableId })
         //if dbTableInfos !=nil{
         //   dbP.BwType = dbTableInfos[0].BwType
         //   dbP.TableName = dbTableInfos[0].TableName
         //}
         resultList[dbPersonM[p.Id].Index] = dbP
      }
   }

   return  resultList
}

// @Security ApiKeyAuth
// @Summary 更新底库人脸照片
// @Description 更新底库人脸照片
// @Accept multipart/form-data
// @Produce json
// @Tags dbperson 底库人员
// @Param id formData string true "人员id"
// @Param file formData file true "人脸图片"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/dbperson/updateFace [POST]
func (dbc DbPersonController) UpdateFace(c *gin.Context) {
   file, header, err := c.Request.FormFile("file")
   id := c.Request.FormValue("id")
   if err != nil || id == "" {
      util.ResponseFormat(c,code.RequestParamError,"参数有误")
      return
   }
   //文件的名称
   filename := header.Filename
   defer file.Close()
   // weedfs 上传
   fileBytes, err := ioutil.ReadAll(file)
   if err !=nil {
      util.ResponseFormat(c,code.ComError,"图片读取失败")
      return
   }

   //将上传的图片交人脸检测和人脸提取，获得特征
   var faceBase64=""
   faceArr, err, pI := service.GetFaceFeaFromSdk(fileBytes, time.Second*5)
   if faceArr ==nil {
      util.ResponseFormat(c,code.ComError,"未到提取人脸")
      return
   }
   var rcFace *protomsg.Rect
   if err ==nil && len(faceArr) >0 {
      if len(faceArr) >1 {
         util.ResponseFormat(c,code.ComError,"人脸大于一张，请换一张人脸图片")
         return
      }
      for _,r := range faceArr {
         //拿到人脸的坐标
         rcFace = r.Pos.RcFace

         faceBase64 = base64.StdEncoding.EncodeToString(r.Feats)//获取提取到的第一张人脸特征
         break
      }
   }
   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"
   //根据人脸坐标扣出人脸小图
   t1 := time.Now()
   cutFaceImgData := util.SubImg(*pI, int(rcFace.Left), int(rcFace.Top), int(rcFace.Right), int(rcFace.Bottom))
   logger.Debug("SubImg用时：", time.Since(t1))
   t1 = time.Now()
   weedFilePath, e := WeedFSClient.UploadFile(weedfsUri, filename, cutFaceImgData)
   logger.Debug("上传到weedfs用时:", time.Since(t1))
   t1 = time.Now()
   if e != nil {
      util.ResponseFormat(c,code.ComError,"人脸上传失败")
      return
   }
   m := map[string]interface{} {
      "faceFeature": faceBase64,
      "personPicUrl": weedFilePath,
   }
   util.ResponseFormat(c,code.Success, m)

   //var dbpApi dbapi.DbPersonApi
   //b,d := dbpApi.UpdateFace(id,faceBase64,weedFilePath)
   //if b {
   //   util.ResponseFormat(c,code.UpdateSuccess,d)
   //} else {
   //   util.ResponseFormat(c,code.UpdateFail,"更新人脸失败")
   //}
}

// @Security ApiKeyAuth
// @Summary 删除底库人员
// @Description 删除库人员
// @Accept  x-www-form-urlencoded
@@ -162,27 +395,31 @@
   }
}

type multiIds []string
type DelMultiPerson []string

// @Summary 批量删除底库人员
// @Description 批量删除库人员
// @Security ApiKeyAuth
// @Summary 删除底库人员
// @Description 删除库人员
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param uuids body controllers.multiIds true "底库人员ids"
// @Param uuids body controllers.DelMultiPerson true "底库人员ids "
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
// @Failure 500 {string} json "{"code":500,  msg:"返回错误信息", success:false}"
// @Router /data/api-v/dbperson/deleteMoreDbPerson [POST]
func (dbc DbPersonController) DeleteMoreDbPerson(c *gin.Context) {
   uuids := make([]string, 0, 5)
   var uuids DelMultiPerson
   err := c.BindJSON(&uuids)
   if err !=nil || len(uuids)==0{
      util.ResponseFormat(c,code.RequestParamError,"参数有误")
      return
   }
   logger.Debug("DeleteMoreDbPerson len(uuids):",len(uuids))
   var pApi dbapi.DbPersonApi
   paramBody := util.Struct2Map(uuids)
   b, _ := pApi.DeleteMoreDbPerson(paramBody)
   m := map[string]interface{}{
      "ids": uuids,
   }
   b, _ := pApi.DeleteMoreDbPerson(m)
   if b {
      util.ResponseFormat(c, code.Success, "删除底库人员成功")
   } else {
@@ -190,6 +427,7 @@
   }
}

// @Security ApiKeyAuth
// @Summary 查询底库人员列表
// @Description 查询库人员列表
// @Accept  json
@@ -253,6 +491,59 @@
   }
}

type JoinDbTVo struct {
   CaptureId string       `json:"captureId"`
   TableIds  []string       `json:"tableIds"`
}

// @Security ApiKeyAuth
// @Summary 抓拍人员加入底库
// @Description 抓拍人员加入底库
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param obj body controllers.JoinDbTVo true "底库数据"
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
// @Failure 500 {string} json "{"code":500,  msg:"返回错误信息", success:false}"
// @Router /data/api-v/dbperson/joinDbTable [POST]
func (dbc *DbPersonController) JoinDbTable(c *gin.Context) {
   var reqBody JoinDbTVo
   c.BindJSON(&reqBody)
   if reqBody.CaptureId == "" || len(reqBody.TableIds) ==0 {
      util.ResponseFormat(c,code.RequestParamError, "参数有误")
      return
   }
   localConf, err := cache.GetServerInfo()
   if err !=nil || localConf.AlarmIp == "" || localConf.AlarmPort <=0 {
      util.ResponseFormat(c,code.ComError,"报警设置有误")
      return
   }
   videopersons, e := esApi.Videopersonsinfosbyid([]string{reqBody.CaptureId}, config.EsInfo.EsIndex.VideoPersons.IndexName, localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)))
   if e ==nil && videopersons !=nil && len(videopersons) == 1{
      var personPicUrl = ""//人脸图片
      var feature = ""//特征
      if videopersons[0].PicSmUrl !=nil && len(videopersons[0].PicSmUrl) >0 {
         personPicUrl = videopersons[0].PicSmUrl[0]
      }
      fea, e2 := esApi.GetVideoPersonFaceFeatureById(reqBody.CaptureId, config.EsInfo.EsIndex.VideoPersons.IndexName, localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)))
      if e2 == nil && fea !="" {
         feature = fea
      }
      if personPicUrl != "" && feature != "" {
         var dbpApi dbapi.DbPersonApi
         b,d := dbpApi.JoinDbTable(reqBody.TableIds, feature, personPicUrl)
         if b {
            util.ResponseFormat(c,code.Success,d)
            return
         } else {
            util.ResponseFormat(c,code.ComError,"加入失败")
            return
         }
      }
   }
   util.ResponseFormat(c,code.ComError,"加入失败")
}

type DbtSearch struct {
   TableId string `json:"tableId"`
   OrderName string `json:"orderName"`
@@ -262,6 +553,63 @@
   Size int `json:"size"`
}

type DbPersonMove struct {
   PersonId string `json:"personId"`
   TableIds []string `json:"tableIds"`
}

// @Security ApiKeyAuth
// @Summary 人员移动
// @Description 人员移动
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param obj body controllers.DbPersonMove true "移动参数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/dbperson/move [POST]
func (dbc *DbPersonController) Move(c *gin.Context) {
   var reqBody DbPersonMove
   c.BindJSON(&reqBody)
   if reqBody.PersonId == "" || len(reqBody.TableIds) == 0 {
      util.ResponseFormat(c,code.RequestParamError, "参数有误")
      return
   }
   var dbpApi dbapi.DbPersonApi
   b,d := dbpApi.Move(reqBody.PersonId, reqBody.TableIds)
   if b {
      util.ResponseFormat(c,code.Success,d)
   } else {
      util.ResponseFormat(c,code.ComError,"")
   }
}

// @Security ApiKeyAuth
// @Summary 人员复制
// @Description 人员复制
// @Accept  json
// @Produce json
// @Tags dbperson 底库人员
// @Param obj body controllers.DbPersonMove true "复制参数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/dbperson/copy [POST]
func (dbc *DbPersonController) Copy(c *gin.Context) {
   var reqBody DbPersonMove
   c.BindJSON(&reqBody)
   if reqBody.PersonId == "" || len(reqBody.TableIds) == 0 {
      util.ResponseFormat(c,code.RequestParamError, "参数有误")
      return
   }
   var dbpApi dbapi.DbPersonApi
   b,d := dbpApi.Copy(reqBody.PersonId, reqBody.TableIds)
   if b {
      util.ResponseFormat(c,code.Success,d)
   } else {
      util.ResponseFormat(c,code.ComError,"")
   }
}

/*
// @Summary 查询底库人员列表
// @Description 查询库人员列表