func AddLxMessage add Type,type=1 when face ,type=2 when action

liuxiaolong

2019-07-26 c9b888e47d0697ad6707b858c933a12037265806

 insertdata/insertDataToEs.go

@@ -55,7 +55,7 @@
   TaskName        string                 `json:"taskName"`
   SdkName         string                 `json:"sdkName"`
   Content         string                 `json:"content"`
   AlarmRules      []AlarmRule          `json:"alarmRules"`
   AlarmRules      []AlarmRule            `json:"alarmRules"`
   LikeDate        string                 `json:"likeDate"`
   Sex             string                 `json:"sex"`
   Age             int32                  `json:"age"`
@@ -144,10 +144,8 @@
               panic("解压缩图片时出现错误")
            }
            alarmRules := []AlarmRule{}
            for _, faceResult := range msg.RuleResult["face"].([]ruleserver.FaceResult) {
               alarm := ChangeToString(faceResult.AlarmLevel)
               alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText})
            }
            alarm := ChangeToString(faceResult.DefenceState,faceResult.AlarmLevel)
            alarmRules = append(alarmRules, AlarmRule{faceResult.RuleGroupId, alarm, faceResult.RuleText})
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            bigPhotoUrl := make(map[string]interface{})
@@ -223,7 +221,7 @@
            } else {
               logger.Info("插入es返回的信息：", resp1)
               // 发出录像信号
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}})
               ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1})
            }
         }
      }
@@ -299,7 +297,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}})
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:1 })
   }
   //if msg.RuleResult["cacheData"] != nil {
   //   InsertFace(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
@@ -314,7 +312,7 @@
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
         sdkNames = sdkNames + yoloResult.SdkName
         alarm := ChangeToString(yoloResult.AlarmLevel)
         alarm := ChangeToString(yoloResult.DefenceState,yoloResult.AlarmLevel)
         alarmRules = append(alarmRules, AlarmRule{yoloResult.RuleGroupId, alarm, yoloResult.RuleText})
      }
      isAlarm := 0
@@ -353,7 +351,7 @@
      }
      serverIp, err := GetLocalIP()
      url := []string{}
      url = append(url,strings.Split(resp["fileUrl"].(string), "/")[1])
      url = append(url, strings.Split(resp["fileUrl"].(string), "/")[1])
      if msg.RuleResult["cacheData"] != nil {
         InsertYolo(msg.RuleResult["cacheData"].(ruleserver.ResultMsg))
         // 把缓存的数据上传后得到地址存进去
@@ -371,7 +369,7 @@
         } else {
            logger.Info("上传的图片信息：", resp1)
         }
         url = append(url,strings.Split(resp1["fileUrl"].(string), "/")[1])
         url = append(url, strings.Split(resp1["fileUrl"].(string), "/")[1])
      }
      peraction := Personaction{
         uuid.NewV4().String(),
@@ -408,7 +406,7 @@
      } else {
         logger.Debug("插入es返回的数据信息是：", resp1)
         // 发出录像信号
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}})
         ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
         logger.Warn("__________________________________________往ES插入yolo数据成功")
         //os.Exit(1)
      }
@@ -478,7 +476,7 @@
   } else {
      logger.Info("插入es返回的信息：", resp1)
      // 发出录像信号
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}})
      ruleserver.AddLxMessage(&protomsg.VideotapeInfo{EsDataId: resp1["_id"].(string), CameraId: msg.Cid, TaskId: msg.Tasklab.Taskid, ImgId: i.Id, SdkIds: []string{}, Type:2 })
      logger.Warn("__________________________________________往ES插入yolo数据成功")
      //os.Exit(1)
   }
@@ -513,23 +511,28 @@
}

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

   return alarm
}