---

panlei

2019-06-25 0b9b9fc5ec6c42cc22e69df7ae587a481143030b

 insertdata/insertDataToEs.go

@@ -2,18 +2,19 @@

import (
   "errors"
   "fmt"
   "encoding/json"
   "log"
   "net"
   "time"

   "basic.com/pubsub/protomsg.git"
   "basic.com/dbapi.git"
   "github.com/golang/protobuf/proto"
   "ruleprocess/ruleserver"
   "ruleprocess/util"
   "github.com/golang/protobuf/proto"
   "github.com/satori/go.uuid"
)

var weedfsUrl = "http://192.168.1.182:6333/submit"
// 人脸的数据结构
type PerVideoPicture struct {
   Id              string `json:"id"`
@@ -86,12 +87,12 @@

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

// 往ES插数据
func InsertToEs(msg ruleserver.ResultMsg) {
   fmt.Println("往ES插数据")
   log.Println("往ES插数据")

   for _, sdkinfo := range msg.Tasklab.Sdkinfos {
      if sdkinfo.Sdktype == "FaceDetect" {
@@ -100,7 +101,7 @@
            faceParam := protomsg.ParamFacePos{}
            err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
            if err1 != nil {
               fmt.Println("解析FACE sdk有误", err1)
               log.Println("解析FACE sdk有误", err1)
               continue
            }
            for _, face := range faceParam.Faces {
@@ -149,13 +150,13 @@
               requstbody, err := json.Marshal(pervideo)

               if err != nil {
                  fmt.Println("json parse error ", err)
                  log.Println("json parse error ", err)
                  return

               }
               err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
               if err != nil {
                  fmt.Println("es can not execute right.")
                  log.Println("es can not execute right.")
               }
            }
         } else {
@@ -168,7 +169,7 @@
            yoloObj := protomsg.ParamYoloObj{}
            err1 := proto.Unmarshal(sdkinfo.Sdkdata, &yoloObj)
            if err1 != nil {
               fmt.Println("解析YOLO sdk有误", err1)
               log.Println("解析YOLO sdk有误", err1)
               //continue
            }
            alarmRules := []AlarmRule{}
@@ -176,21 +177,33 @@
               alarmRules = append(alarmRules,AlarmRule{result.RuleGroupId,result.AlarmLevel,result.RuleText})
            }
            isAlarm := ""
            resp := make(map[string]interface{})
            resp["fileUrl"] = ""
            if len(alarmRules) > 0 {
               isAlarm = "1"
               // 解压缩并上传图片
               bdata, err := util.UnCompress(msg.Data)
               if err != nil {
                  panic("解压缩图片时出现错误")
               }
               i := protomsg.Image{}
               err = proto.Unmarshal(bdata, &i)
               resp,err = util.PostFormBufferData(weedfsUrl,i.Data,uuid.NewV4().String())
               log.Println("已报警并上传改帧图片到服务器")
            }else {
               isAlarm = "0"
               continue
               //continue
            }
            log.Println("图片上传返回值：",resp)
            // 查询本机信息
            flag,localConfig := dbapi.SysSetApi{}.GetServerInfo()
            if !flag {
               fmt.Println("查询本机信息失败！")
               log.Println("查询本机信息失败！")
            }
            // 查询cameraName
            camera,err := dbapi.CameraApi{}.GetCameraById(msg.Cid)
            if err == nil {
               fmt.Println("查询摄像机信息失败")
            if err != nil {
               log.Println("查询摄像机信息失败")
            }
            serverIp,err := GetLocalIP()
            peraction := Personaction{
@@ -207,7 +220,7 @@
               localConfig.ServerName,
               serverIp,
               "",
               "",
               resp["fileUrl"].(string),
               time.Now().Format("2006-01-02 15:04:05"),
               "",
               isAlarm,
@@ -218,13 +231,13 @@
            requstbody, err := json.Marshal(peraction)

            if err != nil {
               fmt.Println("json parse error ", err)
               log.Println("json parse error ", err)
               return

            }
            err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
            if err != nil {
               fmt.Println("es can not execute right.")
               log.Println("es can not execute right.")
            }
         } else {
            continue