panlei
2019-07-02 8be4f2aac9f180061e30600b1c55661571b10249
加个过人脸规则就退出的测试
2个文件已修改
32 ■■■■■ 已修改文件
insertdata/insertDataToEs.go 30 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
ruleserver/ruleToformula.go 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
insertdata/insertDataToEs.go
@@ -90,7 +90,7 @@
type AlarmRule struct {
    GroupId    string `json:"groupId"`
    AlarmLevel int32  `json:"alarmLevel"`
    AlarmLevel   `json:"alarmLevel"`
    RuleText   string `json:"ruleText"`
}
@@ -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{})
@@ -254,7 +257,7 @@
        if err != nil {
            log.Println("往ES插入数据失败",err)
        } else {
            log.Println("__________________________________________往ES插入数据成功")
            log.Println("__________________________________________往ES插入数据yolo成功")
        }
    }
}
@@ -286,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
}
ruleserver/ruleToformula.go
@@ -4,6 +4,7 @@
    "encoding/json"
    "fmt"
    "log"
    "os"
    "ruleprocess/cache"
    "sort"
    "strconv"
@@ -338,6 +339,7 @@
                log.Println("-------------------yolo结果标签",args.RuleResult["yolo"].([]Result))
                if args.RuleResult["faces"] != nil {
                    log.Println("-------------------face结果标签",args.RuleResult["faces"].([]Arg))
                    os.Exit(1)
                }
                return true
            } else {