panlei
2019-07-02 40dad23a110592e05e0845d0b6b76bf5588e0826
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"`
}
@@ -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
}