---

panlei

2019-07-02 5d2c416efe7dc1d5a2f96785ae90061cf27453b5

 insertdata/insertDataToEs.go

@@ -90,7 +90,7 @@

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

@@ -98,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)
@@ -172,6 +172,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)
@@ -184,7 +186,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{})
@@ -254,7 +257,7 @@
      if err != nil {
         log.Println("往ES插入数据失败",err)
      } else {
         log.Println("__________________________________________往ES插入数据成功")
         log.Println("__________________________________________往ES插入数据yolo成功")
      }
   }
}
@@ -286,3 +289,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
}