panlei
2019-07-02 40dad23a110592e05e0845d0b6b76bf5588e0826
insertdata/insertDataToEs.go
@@ -6,10 +6,10 @@
   "fmt"
   "log"
   "net"
   "ruleprocess/cache"
   "strings"
   "time"
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/golang/protobuf/proto"
   "github.com/satori/go.uuid"
@@ -74,7 +74,7 @@
   TaskName        string      `json:"taskName"`
   SdkName         string      `json:"sdkName"`
   Content         string      `json:"content"`
   AlarmRules      []AlarmRule `json: "alarmRules"`
   AlarmRules      []AlarmRule `json:"alarmRules"`
   AnalyServerId   string      `json:"analyServerId"`
   AnalyServerName string      `json:"analyServerName"`
   AnalyServerIp   string      `json:"analyServerIp"`
@@ -90,7 +90,7 @@
type AlarmRule struct {
   GroupId    string `json:"groupId"`
   AlarmLevel int32  `json:"alarmLevel"`
   AlarmLevel string `json:"alarmLevel"`
   RuleText   string `json:"ruleText"`
}
@@ -114,8 +114,8 @@
         fmt.Println(bigPhotoUrl)
         if len(face.Liker) == 0 {
            // 人脸检测,没有相似的底库人员
            flag, localConfig := dbapi.SysSetApi{}.GetServerInfo()
            if !flag {
            localConfig, err := cache.GetServerInfo()
            if err != nil {
               log.Println("查询本机信息失败!")
            }
            serverIp, err := GetLocalIP()
@@ -150,7 +150,7 @@
               face.ThftRes.Smile,
               face.ThftRes.Beauty,
               "不是每个人脸算法都有",
               resp["fileUrl"].(string),
               strings.Split(resp["fileUrl"].(string), "/")[1],
               "暂无集群",
               localConfig.ServerId,
               localConfig.ServerName,
@@ -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{})
@@ -197,8 +200,13 @@
         }
         i := protomsg.Image{}
         err = proto.Unmarshal(bdata, &i)
         resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         log.Println("已报警并上传改帧图片到服务器")
         //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         resp, err =util.DrawPolygonOnImage(msg.Cid,i)
         if err != nil {
            log.Println("画框或上传图片服务器出错",err)
         }else{
            log.Println("已报警并上传改帧图片到服务器")
         }
      } else {
         isAlarm = "0"
         // 不是报警数据不存
@@ -206,12 +214,12 @@
      }
      // log.Println("图片上传返回值:", resp)
      // 查询本机信息
      flag, localConfig := dbapi.SysSetApi{}.GetServerInfo()
      if !flag {
      localConfig, err := cache.GetServerInfo()
      if err !=nil {
         log.Println("查询本机信息失败!")
      }
      // 查询cameraName
      camera, err := dbapi.CameraApi{}.GetCameraById(msg.Cid)
      camera, err := cache.GetCameraById(msg.Cid)
      if err != nil {
         log.Println("查询摄像机信息失败")
      }
@@ -220,7 +228,7 @@
         uuid.NewV4().String(),
         msg.Cid,
         camera.Name,
         msg.Caddr,
         camera.Addr,
         msg.Tasklab.Taskid,
         msg.Tasklab.Taskname,
         sdkNames,
@@ -249,7 +257,7 @@
      if err != nil {
         log.Println("往ES插入数据失败",err)
      } else {
         log.Println("__________________________________________往ES插入数据成功")
         log.Println("__________________________________________往ES插入数据yolo成功")
      }
   }
}
@@ -281,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
}