panlei
2019-09-12 08a17e00ecc337b888ab8807da3ab2738a269805
ruleserver/readyDataForRule.go
@@ -51,16 +51,11 @@
   }
   logger.Debug("本机信息和server信息:", localConfig, serverIp, serverPort)
   bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId)
   sock, err = req.NewSocket();
   if err != nil {
      logger.Error("创建请求socket失败: %s", err.Error())
   }
   serverIP, _ := GetLocalIP()
   if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
      logger.Error("请求socket拨号失败: %s", err.Error())
   }
   go Push1()
}
var sender chan *protomsg.CompareArgs = make(chan *protomsg.CompareArgs)
var receiver chan []byte = make(chan []byte)
type BaseInfo struct {
   TableId      string  `json:"tableId"`
   TableName    string  `json:"tableName"`
@@ -87,10 +82,14 @@
      }
   }
   logger.Debug("当前数据帧要匹配的规则组:-------------------------","摄像机id:",cameraId,"任务id",taskId)
   for _,ruleGroup := range taskGroup.GroupRules{
      logger.Info(ruleGroup.GroupText)
   if taskGroup == nil  {
      return nil
   } else {
      for _,ruleGroup := range taskGroup.GroupRules{
         logger.Info(ruleGroup.GroupText)
      }
      return taskGroup
   }
   return taskGroup
}
// 根据摄像机id拿到摄像机所有区域
@@ -180,7 +179,7 @@
      Source:false,
   }
   bytes := Push(comArg,sock)
   bytes := getCompareMsg(comArg)
   //bytes := []byte{}
   var scResult protomsg.SdkCompareResult
   err1 := proto.Unmarshal(bytes, &scResult)
@@ -298,7 +297,7 @@
func (a *AreaMap) CountAreaObjs(arg *SdkData) {
   a.targetNum = 0
   threshold := 0.0       // 相似度
   threshold := 0.8       // 相似度
   intersectionper := 0.2 // 占比
   size := 0.0            // 尺寸
@@ -351,6 +350,10 @@
      logger.Error("Failed set MaxRecvSize: %v", err)
      return nil
   }
   serverIP, _ := GetLocalIP()
   if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
      logger.Error("请求socket拨号失败: %s", err.Error())
   }
   //sock.AddTransport(ipc.NewTransport())
   sock.AddTransport(tcp.NewTransport())
@@ -374,6 +377,58 @@
   return msg
}
func Push1(){
   //var sock mangos.Socket
   var err error
   var msg []byte
   if sock, err = req.NewSocket(); err != nil {
      logger.Error("创建请求socket失败: %s", err.Error())
   }
   errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024)
   if errSize != nil {
      logger.Error("Failed set MaxRecvSize: %v", err)
   }
   //sock.AddTransport(ipc.NewTransport())
   sock.AddTransport(tcp.NewTransport())
   serverIP, _ := GetLocalIP()
   if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
      logger.Error("请求socket拨号失败: %s", err.Error())
   }
   logger.Info("序列化数据")
   for {
      select {
      // case <-ctx.Done():
      //    return
      case data := <- sender:
         bytes,err1 := proto.Marshal(data)
         logger.Info("数据长度为:",len(bytes))
         if err1 != nil {
            logger.Info("序列化失败:",err1)
         }
         logger.Debug("推送数据")
         //bytes := []byte("ndfasojdfaidsos")
         if err = sock.Send(bytes); err != nil {
            logger.Error("推送socket发送数据失败: %s", err.Error())
            //os.Exit(1)
         }
         if msg, err = sock.Recv(); err != nil {
            logger.Error("接收响应失败: %s", err.Error())
            //os.Exit(1)
         }
         logger.Debug("数据推送成功!收到响应,数据长度为:",len(msg))
         receiver <- msg
      default:
      }
   }
}
func getCompareMsg(data *protomsg.CompareArgs) []byte{
   sender <- data
   return <-receiver
}
// 获取本机ip
func GetLocalIP() (ipv4 string, err error) {
   var (