用个100代替宽高试试

panlei

2019-07-02 a4e36f6078a80ceb908f5c3fcc77fdc0f7da056a

 insertdata/insertDataToEs.go

@@ -10,7 +10,6 @@
   "strings"
   "time"

   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/golang/protobuf/proto"
   "github.com/satori/go.uuid"
@@ -91,7 +90,7 @@

type AlarmRule struct {
   GroupId    string `json:"groupId"`
   AlarmLevel int32  `json:"alarmLevel"`
   AlarmLevel string `json:"alarmLevel"`
   RuleText   string `json:"ruleText"`
}

@@ -99,9 +98,9 @@
func InsertToEs(msg ruleserver.ResultMsg) {

   // 直接从规则的标签数据里拿符合规则的人脸结果
   if msg.RuleResult["faces"] != nil && len(msg.RuleResult["faces"].([]ruleserver.Arg)) > 0 {
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 {
      log.Println("往ES插人脸数据")
      for _, face := range msg.RuleResult["faces"].([]ruleserver.Arg) {
      for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) {
         // 上传大图
         // 解压缩并上传图片
         bdata, err := util.UnCompress(msg.Data)
@@ -115,8 +114,8 @@
         fmt.Println(bigPhotoUrl)
         if len(face.Liker) == 0 {
            // 人脸检测，没有相似的底库人员
            flag, localConfig := dbapi.SysSetApi{}.GetServerInfo()
            if !flag {
            localConfig, err := cache.GetServerInfo()
            if err != nil {
               log.Println("查询本机信息失败！")
            }
            serverIp, err := GetLocalIP()
@@ -127,9 +126,8 @@
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            i1 := protomsg.Image{}
            i1 = util.Subimg(i.Data, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height))
            resp, err := util.PostFormBufferData(weedfsUrl, i1, uuid.NewV4().String())
            bytes := util.Subimg(i.Data, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height))
            resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String())
            if err != nil {
               log.Println("上传小图出错")
            }
@@ -151,7 +149,7 @@
               face.ThftRes.Smile,
               face.ThftRes.Beauty,
               "不是每个人脸算法都有",
               resp["fileUrl"].(string),
               strings.Split(resp["fileUrl"].(string), "/")[1],
               "暂无集群",
               localConfig.ServerId,
               localConfig.ServerName,
@@ -173,6 +171,8 @@
            err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
            if err != nil {
               log.Println("es can not execute right.")
            } else {
               log.Println("___________________________________往ES插入人脸数据成功")
            }
         }else {
            log.Println("跟底库的相似数据---------：",face.Liker)
@@ -185,7 +185,8 @@
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         sdkNames = sdkNames + yoloResult.SdkName
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, yoloResult.AlarmLevel, yoloResult.RuleText})
         alarm := ChangeToString(yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
      }
      isAlarm := ""
      resp := make(map[string]interface{})
@@ -199,7 +200,7 @@
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata, &i)
         //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         resp, err =util.DrawPolygonOnImage(i)
         resp, err =util.DrawPolygonOnImage(msg.Cid,i)
         if err != nil {
            log.Println("画框或上传图片服务器出错",err)
         }else{
@@ -226,7 +227,7 @@
         uuid.NewV4().String(),
         msg.Cid,
         camera.Name,
         msg.Caddr,
         camera.Addr,
         msg.Tasklab.Taskid,
         msg.Tasklab.Taskname,
         sdkNames,
@@ -255,7 +256,7 @@
      if err != nil {
         log.Println("往ES插入数据失败",err)
      } else {
         log.Println("__________________________________________往ES插入数据成功")
         log.Println("__________________________________________往ES插入数据yolo成功")
      }
   }
}
@@ -287,3 +288,24 @@
   err = errors.New("ipv4 not found")
   return
}

// 把报警等级转化成汉字
func ChangeToString(i int32) string{
   alarm := ""
   if i == 1 {
      alarm = "一级"
   }
   if i == 2 {
      alarm = "二级"
   }
   if i == 3 {
      alarm = "三级"
   }
   if i == 4 {
      alarm = "四级"
   }
   if i == 5 {
      alarm = "五级"
   }
   return alarm
}