加上人脸，过完规则的标签区分人脸和yolo

panlei

2019-06-26 a3a788801fcd2efb16d58e91f393bad8d7730b96

 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,89 +87,97 @@

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插数据")

   for _, sdkinfo := range msg.Tasklab.Sdkinfos {
      if sdkinfo.Sdktype == "FaceDetect" {
         if len(sdkinfo.Sdkdata) > 1 {

            faceParam := protomsg.ParamFacePos{}
            err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
            if err1 != nil {
               fmt.Println("解析FACE sdk有误", err1)
               continue
            }
            for _, face := range faceParam.Faces {
               pervideo := PerVideoPicture{
                  uuid.NewV4().String(),
                  msg.Cid,
                  msg.Caddr,
                  time.Now().Format("2006-01-02 15:04:05"),
                  "",
                  msg.Tasklab.Taskid,
                  msg.Tasklab.Taskname,
                  sdkinfo.SdkName,
                  "",
                  "",
                  face.Result.Gender,
                  face.Result.Age,
                  "",
                  face.Result.Race,
                  face.Result.Smile,
                  face.Result.Beauty,
                  "不是每个人脸算法都有",
                  "---",
                  "",
                  "",
                  "",
                  "",
                  "",
                  "",
                  "",
                  "",
                  0,
                  Base{
                     "是每个人脸算法都有吗",
                     "",
                     0,
                     "",
                     "",
                     "",
                     "",
                     "",
                     "",
                     "",
                     "",
                  },
               }
               requstbody, err := json.Marshal(pervideo)

               if err != nil {
                  fmt.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.")
               }
            }
         } else {
            continue
   log.Println("往ES插数据")
   // 直接从规则的标签数据里拿符合规则的人脸结果
   for _, result := range msg.RuleResult {
      if !result.IsYolo {
         for _,face1 := range result.Faces{
            println(face1)
         }
      }
   }
   for _, sdkinfo := range msg.Tasklab.Sdkinfos {
      //if sdkinfo.Sdktype == "FaceDetect" {
      //   if len(sdkinfo.Sdkdata) > 1 {
      //
      //      faceParam := protomsg.ParamFacePos{}
      //      err1 := proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
      //      if err1 != nil {
      //         log.Println("解析FACE sdk有误", err1)
      //         continue
      //      }
      //      for _, face := range faceParam.Faces {
      //         pervideo := PerVideoPicture{
      //            uuid.NewV4().String(),
      //            msg.Cid,
      //            msg.Caddr,
      //            time.Now().Format("2006-01-02 15:04:05"),
      //            "",
      //            msg.Tasklab.Taskid,
      //            msg.Tasklab.Taskname,
      //            sdkinfo.SdkName,
      //            "",
      //            "",
      //            face.Result.Gender,
      //            face.Result.Age,
      //            "",
      //            face.Result.Race,
      //            face.Result.Smile,
      //            face.Result.Beauty,
      //            "不是每个人脸算法都有",
      //            "---",
      //            "",
      //            "",
      //            "",
      //            "",
      //            "",
      //            "",
      //            "",
      //            "",
      //            0,
      //            Base{ // 只有人脸比对有这个信息，人脸检测并没有
      //               "是每个人脸算法都有吗",
      //               "",
      //               0,
      //               "",
      //               "",
      //               "",
      //               "",
      //               "",
      //               "",
      //               "",
      //               "",
      //            },
      //         }
      //         requstbody, err := json.Marshal(pervideo)
      //
      //         if err != nil {
      //            log.Println("json parse error ", err)
      //            return
      //
      //         }
      //         err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
      //         if err != nil {
      //            log.Println("es can not execute right.")
      //         }
      //      }
      //
      //   } else {
      //      continue
      //   }
      //}
      if sdkinfo.Sdktype == "Yolo" {
         if len(sdkinfo.Sdkdata) > 1 {

            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 +185,34 @@
               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,uuid.NewV4().String())
               log.Println("已报警并上传改帧图片到服务器")
            }else {
               isAlarm = "0"
               // 不是报警数据不存
               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 +229,7 @@
               localConfig.ServerName,
               serverIp,
               "",
               "",
               resp["fileUrl"].(string),
               time.Now().Format("2006-01-02 15:04:05"),
               "",
               isAlarm,
@@ -218,13 +240,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