panlei
2019-07-05 e3aaa0b9b08e4e7d72b46762a216ee8c37969d9d
insertdata/insertDataToEs.go
@@ -3,10 +3,9 @@
import (
   "encoding/json"
   "errors"
   "fmt"
   "log"
   "net"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
   "time"
@@ -99,7 +98,7 @@
   // 直接从规则的标签数据里拿符合规则的人脸结果
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 {
      log.Println("往ES插人脸数据")
      logger.Info("往ES插人脸数据")
      for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) {
         // 上传大图
         // 解压缩并上传图片
@@ -111,12 +110,12 @@
         err = proto.Unmarshal(bdata, &i)
         bigPhotoUrl := make(map[string]interface{})
         bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         fmt.Println(bigPhotoUrl)
         logger.Info(bigPhotoUrl)
         if len(face.Liker) == 0 {
            // 人脸检测,没有相似的底库人员
            localConfig, err := cache.GetServerInfo()
            if err != nil {
               log.Println("查询本机信息失败!")
               logger.Info("查询本机信息失败!")
            }
            serverIp, err := GetLocalIP()
            // 解压缩并上传图片
@@ -127,15 +126,15 @@
            // 查询cameraName
            camera, err := cache.GetCameraById(msg.Cid)
            if err != nil {
               log.Println("查询摄像机信息失败")
               logger.Info("查询摄像机信息失败")
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            log.Println("-------------------------------------------看下宽和高", i.Width, i.Height)
            logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height)
            bytes := util.SubImg(i, 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("上传小图出错")
               logger.Info("上传小图出错")
            }
            sex := ""
            if face.ThftRes.Gender == 1 {
@@ -179,19 +178,19 @@
            requstbody, err := json.Marshal(pervideo)
            if err != nil {
               log.Println("json parse error ", err)
               logger.Info("json parse error ", err)
               return
            }
            err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
            log.Println(err)
            logger.Info(err.Error())
         } else {
            log.Println("跟底库的相似数据---------:", face.Liker)
            logger.Info("跟底库的相似数据---------:", face.Liker)
         }
      }
   }
   if msg.RuleResult["yolo"] != nil && len(msg.RuleResult["yolo"].([]ruleserver.Result)) > 0 {
      log.Println("往ES插yolo数据")
      logger.Info("往ES插yolo数据")
      var sdkNames string = ""
      alarmRules := []AlarmRule{}
      for _, yoloResult := range msg.RuleResult["yolo"].([]ruleserver.Result) {
@@ -211,27 +210,27 @@
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata, &i)
         //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         resp, err = util.DrawPolygonOnImage(msg.Cid, i)
         resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result))
         if err != nil {
            log.Println("画框或上传图片服务器出错", err)
            logger.Info("画框或上传图片服务器出错", err)
         } else {
            log.Println("已报警并上传改帧图片到服务器")
            logger.Info("已报警并上传改帧图片到服务器")
         }
      } else {
         isAlarm = 0
         // 不是报警数据不存
         return
      }
      // log.Println("图片上传返回值:", resp)
      // logger.Println("图片上传返回值:", resp)
      // 查询本机信息
      localConfig, err := cache.GetServerInfo()
      if err != nil {
         log.Println("查询本机信息失败!")
         logger.Info("查询本机信息失败!")
      }
      // 查询cameraName
      camera, err := cache.GetCameraById(msg.Cid)
      if err != nil {
         log.Println("查询摄像机信息失败")
         logger.Info("查询摄像机信息失败")
      }
      serverIp, err := GetLocalIP()
      peraction := Personaction{
@@ -259,15 +258,15 @@
      requstbody, err := json.Marshal(peraction)
      if err != nil {
         log.Println("json parse error ", err)
         logger.Info("json parse error ", err)
         return
      }
      err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
      if err != nil {
         log.Println("往ES插入数据失败", err)
         logger.Info("往ES插入数据失败", err)
      } else {
         log.Println("__________________________________________往ES插入yolo数据成功")
         logger.Info("__________________________________________往ES插入yolo数据成功")
      }
   }
}