---

panlei

2019-08-06 50c70af0563026086caba166f15f073974f74b07

 labelFilter/ruleForLabel.go

@@ -4,16 +4,26 @@
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "github.com/knetic/govaluate"
   "os"
   "ruleprocess/logger"
   "ruleprocess/ruleserver"
   "strconv"
   "time"
)

func Judge(label *Label, result ruleserver.ResultMsg){
func Judge(result ruleserver.ResultMsg){
   // 装配成自己可以识别的数据
   label := new(Label)
   label.DataFormatToLabel(result)
   logger.Info("label的信息：",label.CameraName,label.DefenceLevel,label.Time)
   if label.CameraName == "" {
      logger.Error("无效数据，直接返回")
      return
   }
   //拿到所有规则组
   var api dbapi.EventPushApi
   b,allRules := api.FindAllDetails()
   logger.Info("查看所有规则组：",allRules)
   if !b {
      logger.Error("查询时间推送规则失败！")
   }
@@ -49,6 +59,7 @@
         }
         // 得出总的判断结果
         if result != "" {
            logger.Info("总的判断结果：",result)
            result = result[2:] // 把前两位字符去掉
            expression, err:= govaluate.NewEvaluableExpression(result);
            if err != nil {
@@ -58,13 +69,14 @@
            if err1 != nil {
               logger.Error("带入参数有误")
            }
            logger.Warn("规则判断结果是：",)
            if result.(bool) {
               logger.Info("通过规则，表达式为：",result)
               // 推送服务器
               pushData(ruleGroup.Urls,result.(ruleserver.ResultMsg))
               os.Exit(1)
            } else {
               logger.Info("没通过规则,表达式为：",result)
               os.Exit(1)
            }
         }
      }
@@ -102,65 +114,126 @@

// 是否符合摄像机规则
func cameraJudge(label *Label,rule *protomsg.EventPushRule) string{
   formula := ""
   if rule.TopicArg == "cameraName" {
      if label.CameraName == rule.RuleValue {
         logger.Info("符合摄像机名称规则")
         return "true"
      } else {
         logger.Info("不符合摄像机名称规则")
         return "false"
      }
      formula = label.CameraName + rule.Operator + rule.RuleValue
   }
   if rule.TopicArg == "cameraLocation" {
      if label.CameraAddr == rule.RuleValue {
         logger.Info("符合摄像机位置规则")
         return "true"
      } else {
         logger.Info("不符合摄像机位置规则")
         return "false"
      }
      formula = label.CameraAddr + rule.Operator + rule.RuleValue
   }
   return ""
   expression, err:= govaluate.NewEvaluableExpression(formula);
   if err != nil {
      logger.Error("表达式有误，请检查！",formula)
      return "false"
   }
   result, err1:= expression.Evaluate(nil)
   if err1 != nil {
      logger.Error("带入参数有误")
      return "false"
   }
   if result.(bool) {
      logger.Info("符合摄像机规则!")
      return "true"
   }
   return "false"
}

// 是否符合底库规则
func baseJudge(label *Label,rule *protomsg.EventPushRule) string{
   flag := "false"
   for _,personGroup := range label.Persons  {
      for _,arg := range personGroup.Args {
         for _,liker := range arg.Liker{
            if liker.TableId == rule.RuleValue {
               logger.Info("符合底库规则！")
               flag = "true"
   if rule.RuleValue == "all" {
      for _,personGroup := range label.Persons {
         for _, arg := range personGroup.Args {
            if len(arg.Liker) > 0 {
               logger.Info("满足全部底库规则")
               return "true"
            }
         }
      }
      return "false"
   } else {
      for _,personGroup := range label.Persons  {
         for _,arg := range personGroup.Args {
            for _,liker := range arg.Liker{
               formula := liker.TableId + rule.Operator + rule.RuleValue
               expression, err:= govaluate.NewEvaluableExpression(formula);
               if err != nil {
                  logger.Error("表达式有误，请检查！",formula)
                  //return "false"
               }
               result, err1:= expression.Evaluate(nil)
               if err1 != nil {
                  logger.Error("带入参数有误")
                  //return "false"
               }
               if result.(bool) {
                  logger.Info("符合底库规则!")
                  return "true"
               }
            }
         }
      }
      return "false"
   }
   return flag
}

//是否符合人员规则
func personJudge(label *Label,rule *protomsg.EventPushRule) string{
   for _,personGroup := range label.Persons  {
      for _,person := range personGroup.Args {
         if rule.TopicArg == "sex" && strconv.Itoa(int(person.ThftRes.Gender)) == rule.RuleValue{
            logger.Info("符合人员性别规则！")
            return  "true"
         formula := ""
         if rule.TopicArg == "sex"{
            formula = strconv.Itoa(int(person.ThftRes.Gender)) + rule.Operator + rule.RuleValue
         }
         if rule.TopicArg == "race" && strconv.Itoa(int(person.ThftRes.Race)) == rule.RuleValue {
            logger.Info("符合人员种族规则！")
         if rule.TopicArg == "race"{
            formula = strconv.Itoa(int(person.ThftRes.Race)) + rule.Operator + rule.RuleValue
            return "true"
         }
         if rule.TopicArg == "age" && strconv.Itoa(int(person.ThftRes.Age)) == rule.RuleValue {
            logger.Info("符合人员年龄规则！")
         if rule.TopicArg == "age"{
            formula = strconv.Itoa(int(person.ThftRes.Age)) + rule.Operator + rule.RuleValue
            return "true"
         }
         if rule.TopicArg == "others" {
            // 其他这个值先这么处理
            return "true"
         }
         expression, err:= govaluate.NewEvaluableExpression(formula);
         if err != nil {
            logger.Error("表达式有误，请检查！",formula)
            //return "false"
         }
         result, err1:= expression.Evaluate(nil)
         if err1 != nil {
            logger.Error("带入参数有误")
            //return "false"
         }
         if result.(bool) {
            logger.Info("符合人员规则！",formula)
            return "true"
         }
      }
      if strconv.Itoa(int(personGroup.AlarmLevel)) == rule.RuleValue {
      if rule.TopicArg == "monitorLevel" {
         if rule.RuleValue == "all" {
            if len(personGroup.Args) > 0 {
               logger.Info("符合人员等级为全部的规则")
               return "true"
            }
         } else {
            formula := strconv.Itoa(int(personGroup.AlarmLevel)) + rule.Operator + rule.RuleValue
            expression, err:= govaluate.NewEvaluableExpression(formula);
            if err != nil {
               logger.Error("表达式有误，请检查！",formula)
               //return "false"
            }
            result, err1:= expression.Evaluate(nil)
            if err1 != nil {
               logger.Error("带入参数有误")
               //return "false"
            }
            if result.(bool) {
               logger.Info("符合人员等级规则!")
               return "true"
            }
         }
         logger.Info("符合人员报警等级规则！")
         return "true"
      }
@@ -171,18 +244,50 @@

// 是否符合布防等级规则
func defenceJudge(label *Label,rule *protomsg.EventPushRule) string{
   for _,def := range label.DefenceLevel {
      if strconv.Itoa(int(def)) == rule.RuleValue {
         logger.Info("符合布防等级规则！")
   if rule.RuleValue == "all" { // 布防等级为全部时只有有布防等级即为通过
      if len(label.DefenceLevel) > 0 {
         logger.Info("符合布防等级规则，布防等级为全部")
         return "true"
      } else {
         logger.Info("不符合布防等级规则")
         return "false"
      }
   } else {
      for _,def := range label.DefenceLevel {
         formula := strconv.Itoa(int(def)) + rule.Operator + rule.RuleValue
         expression, err:= govaluate.NewEvaluableExpression(formula);
         if err != nil {
            logger.Error("表达式有误，请检查！",formula)
            //return "false"
         }
         result, err1:= expression.Evaluate(nil)
         if err1 != nil {
            logger.Error("带入参数有误")
            //return "false"
         }
         if result.(bool) {
            logger.Info("符合布防等级规则!")
            return "true"
         }
      }
      logger.Info("不符合布防等级规则")
      return "false"
   }
   logger.Info("不符合布防等级规则")
   return "false"
}
// 是否符合任务规则
func taskJudge(label *Label,rule *protomsg.EventPushRule) string{
   if label.TaskId == rule.RuleValue {
   formula := label.TaskId + rule.Operator + rule.RuleValue
   expression, err:= govaluate.NewEvaluableExpression(formula);
   if err != nil {
      logger.Error("表达式有误，请检查！",formula)
      return "false"
   }
   result, err1:= expression.Evaluate(nil)
   if err1 != nil {
      logger.Error("带入参数有误")
      return "false"
   }
   if result.(bool) {
      logger.Info("符合任务规则!")
      return "true"
   }
@@ -191,5 +296,7 @@
}
// 调用目标服务器的插入接口
func pushData (urls []*protomsg.PushUrl, data ruleserver.ResultMsg){

   for _,url := range urls {
      Push(url.Url,data)
   }
}