添加日志系统

panlei

2019-07-05 40e90ca2c934c8505f2a9e9b1c4f076657008369

 ruleserver/ruleToformula.go

@@ -3,9 +3,8 @@
import (
   "encoding/json"
   "fmt"
   "log"
   "os"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "sort"
   "strconv"
   "strings"
@@ -138,6 +137,7 @@
   RuleGroupId string // 规则组id
   AlarmLevel  int32  // 报警等级
   RuleText    string // 文字版规则组
   Location    []Rect
}
type LittleRuleResult struct {
   SdkName string // 记录下此结果是哪个sdk的结果
@@ -169,20 +169,20 @@
   //   if item.CameraId == cameraId && item.Type != "line" {
   //      // 需要根据比例把前台画的区域的坐标转化为相应摄像机拍摄的图像的大小   x坐标分别*image.width/页面区域宽  y坐标分别*image.height/页面区域高
   //      // 前台宽高固定
   //      //log.Println("------------符合条件的区域：",item)
   //      //logger.Println("------------符合条件的区域：",item)
   //      cameraPolygons = append(cameraPolygons, item)
   //   }
   //}
   cameraPolygons = cache.GetPolygonsByCameraId(cameraId)

   //log.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为：", len(cameraPolygons))
   //logger.Println("根据摄像机id查到的区域", cameraPolygons, "--区域数量为：", len(cameraPolygons))
   return cameraPolygons
}

// 把sdk从数据帧上提取的按照区域分类归置
func SdkDataFormat(cameraId string, arg *SdkData, cameraPolygons []protomsg.CameraPolygon) {
   for _, polygon := range cameraPolygons {
      //log.Println("++++++在这儿看一下区域啊:", polygon.Polygon)
      //logger.Println("++++++在这儿看一下区域啊:", polygon.Polygon)
      areaMap := AreaMap{cameraId: cameraId, areaId: polygon.Id, areaJson: polygon.Polygon, triggerLine: polygon.TriggerLine, directionLine: polygon.DirectionLine}
      // 为每个摄像机区域填充数据
      areaMap.CountAreaObjs(arg)
@@ -201,23 +201,23 @@
      }
      // 跑本摄像机的所有规则组 一组一组跑
      taskRuleList := GetRuleGroup(args.CameraId) // 本摄像机下所有任务组
      //log.Println("看下摄像机下的任务组：",taskRuleList)
      //logger.Println("看下摄像机下的任务组：",taskRuleList)
      // 得到属于该摄像机的若干组任务的完整规则（跟每一条完整规则比较之后得出本张图像对于某个规则是否报警的结果。放进map，比如本帧图像的id，所碰撞成功的规则id）
      args.RuleResult = make(map[string]interface{})
      args.RuleResult["yolo"] = []Result{}
      args.RuleResult["face"] = []Arg{}
      if len(taskRuleList) > 0 {
         for _, taskRule := range taskRuleList {
            //log.Println("------------------------------任务规则：",taskRule)
            //logger.Println("------------------------------任务规则：",taskRule)
            ruleList := taskRule.GroupRules // 获取的是task下面的任务组
            taskId := taskRule.TaskId
            //log.Println("------------本组任务下的规则组的数量：",len(ruleList))
            //logger.Println("------------本组任务下的规则组的数量：",len(ruleList))
            for i := 0; i < len(ruleList); i++ {
               temp := ruleList[i].Rules // temp为一组完整规则 在此需要判断规则是否是联动规则
               if len(temp) > 0 {
                  if ruleList[i].SetType == "linkTask" {
                     // groupId中含有link则为联动任务
                     //LinkTask(args, ruleList[i], taskId)
                     LinkTask(args, ruleList[i], taskId)
                  } else {
                     // 独立任务的处理
                     RunRule(args, ruleList[i], taskId)
@@ -234,7 +234,7 @@
         fmt.Println("比对规则有误", err.(string))
      }
   }()
   log.Println("+++++++++++规则开始运行+++++++++++++++++当前规则--：", (*groupRule).GroupText)
   logger.Info("+++++++++++规则开始运行+++++++++++++++++当前规则--：", (*groupRule).GroupText)
   resultSplice := []*LittleRuleResult{}
   // 先过完条件规则
   for j := 0; j < len(groupRule.Rules); j++ {
@@ -242,38 +242,20 @@
         for _, areaMap := range sdkData.AreaMapList {
            ruleResult := filterRule(groupRule.Rules[j], &areaMap)
            if ruleResult.Result != "" {
               log.Println("条件规则结果：", ruleResult.Result)
               logger.Info("条件规则结果：", ruleResult.Result)
               resultSplice = append(resultSplice, &ruleResult)
            }
         }
      }
   }
   // 过完条件数据后把符合条件的人脸数据塞进结果标签里
   faces := []Arg{}
   for _, sdkData := range args.Sdkdata {
      for _, areaMap := range sdkData.AreaMapList {
         for j := 0; j < len(groupRule.Rules); j++ {
            faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...)
         }
      }
   }
   // 配了人脸的算法才把人脸的数据甩出来打标签
   flag := false
   for j := 0; j < len(groupRule.Rules); j++ {
      if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
         flag = true
      }
   }
   if flag {
      args.RuleResult["face"] = faces
   }

   // 再过其他数据 这步直接得到结果（真或假） 过目标数量
   for j := 0; j < len(groupRule.Rules); j++ {
      for _, sdkData := range args.Sdkdata {
         for _, areaMap := range sdkData.AreaMapList {
            ruleResult := transferParameters(groupRule.Rules[j], &areaMap)
            if ruleResult.Result != "" {
               log.Println("数量规则结果：", ruleResult.Result)
               logger.Info("数量规则结果：", ruleResult.Result)
               resultSplice = append(resultSplice, &ruleResult)
            }
         }
@@ -285,7 +267,7 @@
         for _, areaMap := range sdkData.AreaMapList {
            ruleResult := timeRuleResult(groupRule.Rules[j], &areaMap)
            if ruleResult.Result != "" {
               log.Println("时间规则结果：", ruleResult.Result)
               logger.Info("时间规则结果：", ruleResult.Result)
               resultSplice = append(resultSplice, &ruleResult)
            }
         }
@@ -306,8 +288,21 @@
   for _, va := range resultSplice {
      completeFormula = completeFormula + va.Result
   }
   if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") || strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") {
      // 以这些开头的基本是联动任务
      if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") || strings.HasPrefix(completeFormula, ">=") || strings.HasPrefix(completeFormula, "<=") || strings.HasPrefix(completeFormula, "==") || strings.HasPrefix(completeFormula, "!=") {
         completeFormula = completeFormula[2 :]
      }
      if strings.HasPrefix(completeFormula, ">") || strings.HasPrefix(completeFormula, "<") {
         completeFormula = completeFormula[1 :]
      }
      logger.Info("-------------------看看拔毛后的表达式：",completeFormula)
      expression, _ := govaluate.NewEvaluableExpression(completeFormula)
      result, _ := expression.Evaluate(nil) // 得到数学公式的结果
      return result.(bool)
   }
   if completeFormula != "" {
      log.Println("结果公式-----------：", completeFormula)
      logger.Info("结果公式-----------：", completeFormula)
      expression, _ := govaluate.NewEvaluableExpression(completeFormula)
      if strings.HasPrefix(completeFormula, "&&") || strings.HasPrefix(completeFormula, "||") {
         panic("规则有误，得到的数学公式不可解析")
@@ -321,14 +316,34 @@
               timeEle.N = timeEle.InitN // 重置定时器
            }
         }
         logger.Info("--------------------走这里了吗---------------")
         return false
      } else {
         // 最后成功报警才把符合条件的人脸数据塞进结果标签里
         faces := []Arg{}
         for _, sdkData := range args.Sdkdata {
            for _, areaMap := range sdkData.AreaMapList {
               for j := 0; j < len(groupRule.Rules); j++ {
                  faces = append(faces, putFaceToResult(groupRule.Rules[j], &areaMap)...)
               }
            }
         }
         // 配了人脸的算法才把人脸的数据甩出来打标签
         flag11 := false
         for j := 0; j < len(groupRule.Rules); j++ {
            if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
               flag11 = true
            }
         }
         if flag11 {
            args.RuleResult["face"] = faces
         }
         // 去看池子里是否有与本帧数据有关的定时器，如果有，看此时是否走到0，没有此定时器或有定时器走到0的话返回成功报警
         var flag bool = true
         for k, timeEle := range TimeEleList {
            if strings.Contains(k, taskId) {
               if timeEle.N != 0 { // 跟这个任务有关的定时器要全部等于0
                  log.Println("———————————-------------不冤，你是被定时器打败的：")
                  logger.Info("———————————-------------不冤，你是被定时器打败的：")
                  flag = false
               }
            }
@@ -345,11 +360,30 @@
               }
            }
            if sdkName != "" {
               args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
               log.Println("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
               // 把他们的位置数据也传下去
               locations := []Rect{}
               for _, sdkData := range args.Sdkdata {
                  for _, areaMap := range sdkData.AreaMapList {
                     for j := 0; j < len(groupRule.Rules); j++ {
                        locations = append(locations, putYolosToResult(groupRule.Rules[j], &areaMap)...)
                     }
                  }
               }
               //// 配了人脸的算法才把人脸的数据甩出来打标签
               //flag11 := false
               //for j := 0; j < len(groupRule.Rules); j++ {
               //   if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" || groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a971" {
               //      flag11 = true
               //   }
               //}
               //if flag11 {
               //   args.RuleResult["face"] = faces
               //}
               args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText,locations})
               logger.Info("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
            }
            if args.RuleResult["face"] != nil {
               log.Println("-------------------face结果标签", args.RuleResult["face"].([]Arg))
               logger.Info("-------------------face结果标签", args.RuleResult["face"].([]Arg))
            }
            return true
         } else {
@@ -370,8 +404,21 @@
         }
      }
   }
   //log.Println("-----------------------------------------------听说你是空的？",faces)
   //logger.Println("-----------------------------------------------听说你是空的？",faces)
   return faces
}

func putYolosToResult(rule *protomsg.Rule, am *AreaMap) []Rect {
   locations := []Rect{}
   if rule.SdkId == am.sdkId && rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a97e" && rule.SdkId != "812b674b-2375-4589-919a-5c1c3278a971" {
      if len(am.filterData) > 0 {
         for _, data := range am.filterData {
            locations = append(locations, data.Location)
         }
      }
   }
   //logger.Println("-----------------------------------------------听说你是空的？",faces)
   return locations
}

// 计算区域内的目标数量以及将相似度、占比、尺寸等打包
@@ -392,12 +439,12 @@

   // }
   for _, obj := range arg.Photo {
      log.Println("------------------看看sdkData:", arg.SdkName, "的Photo数据----------------", obj, "----顺便看看占比-----：", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale))
      logger.Info("------------------看看sdkData:", arg.SdkName, "的Photo数据----------------", obj, "----顺便看看占比-----：", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale))
      if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) {
         // 这步要备齐表达式里所需要的所有参数
         a.targetNum++
         arg1 := Arg{obj.Score, PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale), float64(obj.Rects.Width * obj.Rects.Height), obj.IsYolo, obj.Rects, obj.SdkName, obj.ThftRes, obj.Liker}
         //log.Println("放进去的arg：-------", arg1)
         //logger.Println("放进去的arg：-------", arg1)
         a.args = append(a.args, arg1)
         a.filterData = append(a.filterData, arg1)
      }
@@ -407,7 +454,7 @@
   a.time = time.Unix(time.Now().Unix(), 0).String()[11:16]
   a.keepRight = arg.KeepRight
   a.isStatic = arg.IsStatic
   //log.Println("--------------------看看区域数据：",*a)
   //logger.Println("--------------------看看区域数据：",*a)
}

// 将字符串格式的坐标序列化为Point格式
@@ -448,7 +495,7 @@
// 联动任务的处理
func LinkTask(args *SdkDatas, groupRule *protomsg.GroupRule, taskId string) {
   // new一个定时器，如果以此groupId为标志的定时器不存在的话
   log.Println("------------------------------------------当前是联动任务，规则是：", groupRule.GroupText)
   logger.Info("------------------------------------------当前是联动任务，规则是：", groupRule.GroupText)
   var flag bool = true
   var timeEle = TimeElement{N: 3, InitN: 3, GroupId: groupRule.GroupId}
   for k, timeEle1 := range TimeEleList {
@@ -484,11 +531,13 @@
      for _, va := range timeEle.RuleResults {
         if va.CameraId == args.CameraId {
            va.Result = strconv.FormatBool(isOk)
            logger.Info("这个摄像机--", args.CameraId, "--被赋予了result", va.Result)
         }
      }
      // 判断结果数组是否完满(即被赋值完毕)可得出报警结果
      var isPerfect = true
      for _, va := range timeEle.RuleResults {
         logger.Info("---------------------瞅瞅当前数组________________：", *va)
         if va.Result == "" && va.RuleWithPre != "||" {
            isPerfect = false
         }
@@ -501,12 +550,12 @@
         for _, va := range timeEle.RuleResults {
            completeFormula = completeFormula + va.RuleWithPre + "" + va.Result
         }
         log.Println("---------------------------联动任务的公式", completeFormula)
         logger.Info("---------------------------联动任务的公式", completeFormula)
         if completeFormula != "" {
            expression, _ := govaluate.NewEvaluableExpression(completeFormula)
            result, _ := expression.Evaluate(nil) // 得到数学公式的结果
            if result.(bool) {
               log.Println("___________________________________________________________________联动任务报警")
               logger.Info("___________________________________________________________________联动任务报警")
               sdkName := ""
               for j := 0; j < len(groupRule.Rules); j++ {
                  for _, sdkData := range args.Sdkdata {
@@ -516,13 +565,13 @@
                  }
               }
               if sdkName != "" {
                  args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText})
                  log.Println("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
                  args.RuleResult["yolo"] = append(args.RuleResult["yolo"].([]Result), Result{taskId, sdkName, groupRule.GroupId, groupRule.AlarmLevel, groupRule.GroupText,[]Rect{}})
                  logger.Info("-------------------yolo结果标签", args.RuleResult["yolo"].([]Result))
               }
               if args.RuleResult["face"] != nil {
                  log.Println("-------------------face结果标签", args.RuleResult["face"].([]Arg))
                  logger.Info("-------------------face结果标签", args.RuleResult["face"].([]Arg))
               }
               os.Exit(0)
               //os.Exit(0)
            }
         }
      } else {
@@ -551,7 +600,7 @@
            // 只需要过滤阈值，过滤完后数组长度大于0即为报警，但要考虑如何对每一张都报警呢
            argValue, err := strconv.ParseFloat(rule.SdkArgValue, 64)
            if err != nil {
               log.Println("规则配置的阈值非法")
               logger.Info("规则配置的阈值非法")
               return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort}
            }
            flag := "false"
@@ -582,7 +631,7 @@
            return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + flag, rule.Sort}
         }
         if rule.SdkId == "812b674b-2375-4589-919a-5c1c3278a97e" { // 人脸检测
            log.Println("------------------------------------------------------------------------------------------------------------------------------------终于等到你")
            logger.Info("------------------------------------------------------------------------------------------------------------------------------------终于等到你")
            if rule.Operator == "==" || rule.Operator == ">=" || rule.Operator == "<=" || rule.Operator == "<" || rule.Operator == ">" || rule.Operator == "!=" {
               // 如果是不规矩的连接符统统返回false 规则也只能判断人脸的相似度，所以不存在别的连接符
               if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
@@ -598,13 +647,13 @@
                     var formula string
                     if rule.SdkArgAlias == "score" {
                        formula = strconv.FormatFloat(arg.Score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前相似度小公式：", formula)
                        logger.Info("当前相似度小公式：", formula)
                     } else if rule.SdkArgAlias == "proportion" {
                        formula = strconv.FormatFloat(arg.Proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前占比小公式：", formula)
                        logger.Info("当前占比小公式：", formula)
                     } else {
                        formula = strconv.FormatFloat(arg.Size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                        log.Println("当前尺寸小公式：", formula)
                        logger.Info("当前尺寸小公式：", formula)
                     }
                     expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
                     result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -615,7 +664,7 @@
                  am.targetNum = len(am.filterData) // 把符合条件的目标数量更新到targetNum字段
               }
               if am.targetNum > 0 {
                  log.Println("！！！！！！！！！人脸检测成功")
                  logger.Info("！！！！！！！！！人脸检测成功")
                  return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "true", rule.Sort}
               } else {
                  return LittleRuleResult{am.sdkName, rule.RuleWithPre + "" + "false", rule.Sort}
@@ -633,7 +682,7 @@
      // 处理的都是yolo数据
      if rule.SdkId == am.sdkId && rule.PolygonId == am.areaId { // 首先这条规则得是这个算法的规则，其次规则所对应的区域id要跟区域数据的id对的上
         if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" { // 判断的是相似值，占比，尺寸等过滤条件，如果再有，还可以再加
            log.Println("----------当前条件规则：---------", rule)
            logger.Info("----------当前条件规则：---------", rule)
            var args []Arg
            if rule.RuleWithPre == "&&" {
               args = am.filterData
@@ -646,13 +695,13 @@
               var formula string
               if rule.SdkArgAlias == "score" {
                  formula = strconv.FormatFloat(arg.Score, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前相似度小公式：", formula)
                  logger.Info("当前相似度小公式：", formula)
               } else if rule.SdkArgAlias == "proportion" {
                  formula = strconv.FormatFloat(arg.Proportion, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前占比小公式：", formula)
                  logger.Info("当前占比小公式：", formula)
               } else {
                  formula = strconv.FormatFloat(arg.Size, 'f', -1, 64) + " " + rule.Operator + " " + rule.SdkArgValue // 得到字符串公式
                  log.Println("当前尺寸小公式：", formula)
                  logger.Info("当前尺寸小公式：", formula)
               }
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -709,7 +758,7 @@
//         num++
//      }
//   }
//   log.Println("看看区域内符合条件的目标数量：----------", num)
//   logger.Println("看看区域内符合条件的目标数量：----------", num)
//   formula := strconv.Itoa(num) + " " + ">" + "0"
//   expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
//   result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果
@@ -755,13 +804,13 @@
func timeRuleResult(rule *protomsg.Rule, am *AreaMap) LittleRuleResult {
   if rule.PolygonId == am.areaId && rule.SdkId == am.sdkId { // 首先规则所对应的区域id要跟区域数据的id对的上
      if rule.SdkArgAlias == "time_rule" { // 判断是否符合时间规则
         log.Println("----------当前时间规则：---------", rule)
         logger.Info("----------当前时间规则：---------", rule)
         // 根据放值字段里存的时间规则的id去另一个表里查需要比对的时间段（比如当前时间是周三，应根据区域id查出其周三的几个布防时间段，数组）
         //fmt.Println("时间规则的测试")
         now := time.Now()
         index := getIndexOfWeek(now.Weekday().String())
         timeList := GetTimeById(rule.SdkArgValue, index)
         log.Println("当天的时间段集合：----------", timeList)
         logger.Info("当天的时间段集合：----------", timeList)
         //fmt.Println("从数据库中查出的时间规则：", timeList)
         // 判断图片数据的时间是否符合当前规则 在一个即为true，全不在为false

@@ -788,12 +837,12 @@
            flag := "true"
            for _, timeSlot := range timeList {
               formula := "'" + timeSlot.Start + "'" + " < " + "'" + am.time + "'"
               log.Println("-----------------时间规则不满足的公式start：", formula)
               logger.Info("-----------------时间规则不满足的公式start：", formula)
               expression, _ := govaluate.NewEvaluableExpression(formula) // 得到数学公式
               result, _ := expression.Evaluate(nil)                      // 得到数学公式的结果

               formula1 := "'" + timeSlot.End + "'" + " > " + "'" + am.time + "'"
               log.Println("-----------------时间规则不满足的公式end：", formula1)
               logger.Info("-----------------时间规则不满足的公式end：", formula1)
               expression1, _ := govaluate.NewEvaluableExpression(formula1) // 得到数学公式
               result1, _ := expression1.Evaluate(nil)                      // 得到数学公式的结果
               if result.(bool) && result1.(bool) {
@@ -844,7 +893,7 @@
   json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList)
   for _, timerange := range timeRangeList {
      if timerange.Day == index {
         //log.Println("取到的时间规则：", timerange.TimeRange)
         //logger.Println("取到的时间规则：", timerange.TimeRange)
         return timerange.TimeRange
      }
   }