---
panlei
2019-08-06 de102062b29cff395b1abe1605198bf8bfdc0cea
---
1个文件已修改
165 ■■■■■ 已修改文件
labelFilter/ruleForLabel.go 165 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
labelFilter/ruleForLabel.go
@@ -11,27 +11,27 @@
    "time"
)
func Judge(result ruleserver.ResultMsg){
func Judge(result ruleserver.ResultMsg) {
    // 装配成自己可以识别的数据
    label := new(Label)
    label.DataFormatToLabel(result)
    logger.Info("label的信息:",label.CameraName,label.DefenceLevel,label.Time)
    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)
    b, allRules := api.FindAllDetails()
    logger.Info("查看所有规则组:", allRules)
    if !b {
        logger.Error("查询时间推送规则失败!")
    }
    //遍历所有的规则组,看看符合哪个规则组,就把数据插入到哪个规则组的目标机器上
    for _,ruleGroup := range allRules {
        if ruleGroup.Enable {  // 大规则开关开启状态
    for _, ruleGroup := range allRules {
        if ruleGroup.Enable { // 大规则开关开启状态
            // 先过时间规则
            timeJudge(label,ruleGroup)
            timeJudge(label, ruleGroup)
            result := ""
            var Connector string
            if ruleGroup.IsSatisfyAll {
@@ -39,43 +39,43 @@
            } else {
                Connector = "||"
            }
            for _,rule := range ruleGroup.Rules {
            for _, rule := range ruleGroup.Rules {
                // 过每一个小规则
                if rule.TopicType == "camera" {
                    result = Connector + cameraJudge(label,rule)
                    result = Connector + cameraJudge(label, rule)
                }
                if rule.TopicType == "dbtable" {
                    result = Connector + baseJudge(label,rule)
                    result = Connector + baseJudge(label, rule)
                }
                if rule.TopicType == "task" {
                    result = Connector + taskJudge(label,rule)
                    result = Connector + taskJudge(label, rule)
                }
                if rule.TopicType == "person" {
                    result = Connector + taskJudge(label,rule)
                    result = Connector + taskJudge(label, rule)
                }
                if rule.TopicType == "alarmLevel" {
                    result = Connector + defenceJudge(label,rule)
                    result = Connector + defenceJudge(label, rule)
                }
            }
            // 得出总的判断结果
            if result != "" {
                logger.Info("总的判断结果:",result)
                logger.Info("总的判断结果:", result)
                result = result[2:] // 把前两位字符去掉
                expression, err:= govaluate.NewEvaluableExpression(result);
                expression, err := govaluate.NewEvaluableExpression(result);
                if err != nil {
                    logger.Error("表达式有误,请检查!")
                }
                result, err1:= expression.Evaluate(nil)
                result, err1 := expression.Evaluate(nil)
                if err1 != nil {
                    logger.Error("带入参数有误")
                }
                if result.(bool) {
                    logger.Info("通过规则,表达式为:",result)
                    logger.Info("通过规则,表达式为:", result)
                    // 推送服务器
                    pushData(ruleGroup.Urls,result.(ruleserver.ResultMsg))
                    pushData(ruleGroup.Urls, result.(ruleserver.ResultMsg))
                    os.Exit(1)
                } else {
                    logger.Info("没通过规则,表达式为:",result)
                    logger.Info("没通过规则,表达式为:", result)
                    os.Exit(1)
                }
            }
@@ -84,11 +84,12 @@
    //符合本规则组的就把数据推送到该目的服务器上
}
// 是否符合时间规则
func timeJudge(label *Label,groupRule protomsg.EventPush) string{
func timeJudge(label *Label, groupRule protomsg.EventPush) string {
    timeNow := time.Now().Format("2006-01-02 15:04:05")
    flag1 := isBefore(groupRule.TimeStart,timeNow)
    flag2 := isBefore(timeNow,groupRule.TimeEnd)
    flag1 := isBefore(groupRule.TimeStart, timeNow)
    flag2 := isBefore(timeNow, groupRule.TimeEnd)
    if flag1 && flag2 {
        return "true"
    } else {
@@ -96,16 +97,17 @@
        return "false"
    }
}
// 传入两个时间的字符串,判断第一个时间是否小于第二个时间
func isBefore(time1 string,time2 string) bool{
    expression, err:= govaluate.NewEvaluableExpression("t1 < t2");
func isBefore(time1 string, time2 string) bool {
    expression, err := govaluate.NewEvaluableExpression("t1 < t2");
    if err != nil {
        logger.Error("表达式有误,请检查!")
    }
    parameters := make(map[string]interface{},8)
    parameters := make(map[string]interface{}, 8)
    parameters["t1"] = time1
    parameters["t2"] = time2
    result, err1:= expression.Evaluate(parameters)
    result, err1 := expression.Evaluate(parameters)
    if err1 != nil {
        logger.Error("带入参数有误")
    }
@@ -113,20 +115,20 @@
}
// 是否符合摄像机规则
func cameraJudge(label *Label,rule *protomsg.EventPushRule) string{
func cameraJudge(label *Label, rule *protomsg.EventPushRule) string {
    formula := ""
    if rule.TopicArg == "name" {
        formula = label.CameraName + rule.Operator + rule.RuleValue
        formula = "'" + label.CameraName + "'" + rule.Operator + "'" + rule.RuleValue + "'"
    }
    if rule.TopicArg == "addr" {
        formula = label.CameraAddr + rule.Operator + rule.RuleValue
        formula = "'" + label.CameraAddr + "'" + rule.Operator + "'" + rule.RuleValue + "'"
    }
    expression, err:= govaluate.NewEvaluableExpression(formula);
    expression, err := govaluate.NewEvaluableExpression(formula);
    if err != nil {
        logger.Error("表达式有误,请检查!",formula)
        logger.Error("表达式有误,请检查!", formula)
        return "false"
    }
    result, err1:= expression.Evaluate(nil)
    result, err1 := expression.Evaluate(nil)
    if err1 != nil {
        logger.Error("带入参数有误")
        return "false"
@@ -139,9 +141,9 @@
}
// 是否符合底库规则
func baseJudge(label *Label,rule *protomsg.EventPushRule) string{
func baseJudge(label *Label, rule *protomsg.EventPushRule) string {
    if rule.RuleValue == "all" {
        for _,personGroup := range label.Persons {
        for _, personGroup := range label.Persons {
            for _, arg := range personGroup.Args {
                if len(arg.Liker) > 0 {
                    logger.Info("满足全部底库规则")
@@ -151,16 +153,16 @@
        }
        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);
        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)
                        logger.Error("表达式有误,请检查!", formula)
                        //return "false"
                    }
                    result, err1:= expression.Evaluate(nil)
                    result, err1 := expression.Evaluate(nil)
                    if err1 != nil {
                        logger.Error("带入参数有误")
                        //return "false"
@@ -177,35 +179,35 @@
}
//是否符合人员规则
func personJudge(label *Label,rule *protomsg.EventPushRule) string{
    for _,personGroup := range label.Persons  {
        for _,person := range personGroup.Args {
func personJudge(label *Label, rule *protomsg.EventPushRule) string {
    for _, personGroup := range label.Persons {
        for _, person := range personGroup.Args {
            formula := ""
            if rule.TopicArg == "sex"{
            if rule.TopicArg == "sex" {
                formula = strconv.Itoa(int(person.ThftRes.Gender)) + rule.Operator + rule.RuleValue
            }
            if rule.TopicArg == "race"{
            if rule.TopicArg == "race" {
                formula = strconv.Itoa(int(person.ThftRes.Race)) + rule.Operator + rule.RuleValue
            }
            if rule.TopicArg == "age"{
            if rule.TopicArg == "age" {
                formula = strconv.Itoa(int(person.ThftRes.Age)) + rule.Operator + rule.RuleValue
            }
            if rule.TopicArg == "otherInfo" {
                // 其他这个值先这么处理
                return "true"
            }
            expression, err:= govaluate.NewEvaluableExpression(formula);
            expression, err := govaluate.NewEvaluableExpression(formula);
            if err != nil {
                logger.Error("表达式有误,请检查!",formula)
                logger.Error("表达式有误,请检查!", formula)
                //return "false"
            }
            result, err1:= expression.Evaluate(nil)
            result, err1 := expression.Evaluate(nil)
            if err1 != nil {
                logger.Error("带入参数有误")
                //return "false"
            }
            if result.(bool) {
                logger.Info("符合人员规则!",formula)
                logger.Info("符合人员规则!", formula)
                return "true"
            }
        }
@@ -216,13 +218,13 @@
                    return "true"
                }
            } else {
                formula := strconv.Itoa(int(personGroup.AlarmLevel)) + rule.Operator + rule.RuleValue
                expression, err:= govaluate.NewEvaluableExpression(formula);
                formula := strconv.Itoa(int(personGroup.AlarmLevel)) + rule.Operator + formate(rule.RuleValue)
                expression, err := govaluate.NewEvaluableExpression(formula);
                if err != nil {
                    logger.Error("表达式有误,请检查!",formula)
                    logger.Error("表达式有误,请检查!", formula)
                    //return "false"
                }
                result, err1:= expression.Evaluate(nil)
                result, err1 := expression.Evaluate(nil)
                if err1 != nil {
                    logger.Error("带入参数有误")
                    //return "false"
@@ -241,7 +243,7 @@
}
// 是否符合布防等级规则
func defenceJudge(label *Label,rule *protomsg.EventPushRule) string{
func defenceJudge(label *Label, rule *protomsg.EventPushRule) string {
    if rule.RuleValue == "all" { // 布防等级为全部时只有有布防等级即为通过
        if len(label.DefenceLevel) > 0 {
            logger.Info("符合布防等级规则,布防等级为全部")
@@ -251,14 +253,14 @@
            return "false"
        }
    } else {
        for _,def := range label.DefenceLevel {
            formula := strconv.Itoa(int(def)) + rule.Operator + rule.RuleValue
            expression, err:= govaluate.NewEvaluableExpression(formula);
        for _, def := range label.DefenceLevel {
            formula := strconv.Itoa(int(def)) + rule.Operator + formate(rule.RuleValue)
            expression, err := govaluate.NewEvaluableExpression(formula);
            if err != nil {
                logger.Error("表达式有误,请检查!",formula)
                logger.Error("表达式有误,请检查!", formula)
                //return "false"
            }
            result, err1:= expression.Evaluate(nil)
            result, err1 := expression.Evaluate(nil)
            if err1 != nil {
                logger.Error("带入参数有误")
                //return "false"
@@ -272,15 +274,16 @@
        return "false"
    }
}
// 是否符合任务规则
func taskJudge(label *Label,rule *protomsg.EventPushRule) string{
    formula := label.TaskId + rule.Operator + rule.RuleValue
    expression, err:= govaluate.NewEvaluableExpression(formula);
func taskJudge(label *Label, rule *protomsg.EventPushRule) string {
    formula := "'"+label.TaskId +"'"+ rule.Operator +"'"+ rule.RuleValue+"'"
    expression, err := govaluate.NewEvaluableExpression(formula);
    if err != nil {
        logger.Error("表达式有误,请检查!",formula)
        logger.Error("表达式有误,请检查!", formula)
        return "false"
    }
    result, err1:= expression.Evaluate(nil)
    result, err1 := expression.Evaluate(nil)
    if err1 != nil {
        logger.Error("带入参数有误")
        return "false"
@@ -292,9 +295,31 @@
    logger.Info("不符合任务规则")
    return "false"
}
// 调用目标服务器的插入接口
func pushData (urls []*protomsg.PushUrl, data ruleserver.ResultMsg){
    for _,url := range urls {
        Push(url.Url,data)
func pushData(urls []*protomsg.PushUrl, data ruleserver.ResultMsg) {
    for _, url := range urls {
        Push(url.Url, data)
    }
}
}
// 把报警等级转化成汉字
func formate(str string) string {
    var i string
    if str == "一级" {
        i = "1"
    }
    if str == "二级" {
        i = "2"
    }
    if str == "三级" {
        i = "3"
    }
    if str == "四级" {
        i = "4"
    }
    if str == "五级" {
        i = "5"
    }
    return i
}