panlei
2019-09-29 0d08415bcbc5d1b260861365f26b0532771da2c5
main.go
@@ -4,9 +4,10 @@
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "net/http"
   _ "net/http/pprof"
   "ruleprocess/insertdata"
   "ruleprocess/util"
   "runtime"
   "time"
   //"bufio"
@@ -14,7 +15,7 @@
   "flag"
   "fmt"
   "github.com/golang/protobuf/proto"
   "ruleprocess/logger"
   "basic.com/valib/logger.git"
   //"gocv.io/x/gocv"
   //"image"
   //"image/color"
@@ -22,6 +23,7 @@
   "ruleprocess/cache"
   "ruleprocess/ruleserver"
   "sync"
   "github.com/spf13/viper"
)
var dbIp = flag.String("dbIp", "127.0.0.1", "dbserver ip")
@@ -29,22 +31,35 @@
var surveyPort = flag.Int("surveyPort", 40007, "survey port") //心跳
var pubPort = flag.Int("pubPort", 50007, "pubsub port")       //数据更新
var initchan = make(chan bool)
var env =  flag.String("env","pro","env set")
func init() {
   flag.Parse()
   // 日志初始化
   logger.Config("./info.log", logger.DebugLevel)
   logger.Info("日志初始化成功!")
   insertdata.Init(*env)
   var logFile = "./logger/"
   if viper.GetString("LogBasePath") != "" {
      logFile = viper.GetString("LogBasePath")
   }
   logFile = logFile + "ruleprocess.log"
   fmt.Println("日志地址:",logFile)
   logger.Config(logFile, logger.DebugLevel)
   logger.SetSaveDays(7)
   logger.Info("日志初始化成功!")
}
func main() {
   //fmt.Println("缓存初始化完成",<- initchan)//dbserver初始化完毕
   go func() {
      http.ListenAndServe("0.0.0.0:8899",nil)
   }()
   flag.Parse()
   wg := sync.WaitGroup{}
   wg.Add(2)
   wg.Add(3)
   dbapi.Init(*dbIp, *dbPort)
   go cache.Init(initchan, *dbIp, *surveyPort, *pubPort)
   logger.Info("cache init completed!!!", <-initchan) //dbserver初始化完毕
   //fmt.Println("缓存初始化完成",<- initchan)//dbserver初始化完毕
   ruleserver.Init()
   go ruleserver.TimeTicker()
   go ruleserver.StartServer()
@@ -73,9 +88,9 @@
            fmt.Println("recv error : ", err)
            continue
         } else {
            runtime.GOMAXPROCS(runtime.NumCPU())
            logger.Debug("使用的cpu个数:",runtime.NumCPU())
            go func(msg []byte) {
            //runtime.GOMAXPROCS(runtime.NumCPU())
            //logger.Debug("使用的cpu个数:",runtime.NumCPU())
            //go func(msg []byte) {
               logger.Debug("当前时间戳:", time.Now().Unix())
               arg := ruleserver.SdkDatas{}
               //paramFormat(msg, &arg)
@@ -91,7 +106,7 @@
               logger.Debug("插入完Es所用时间:", time.Since(start))
               //事件推送
               //go labelFilter.Judge(resultMsg)
            }(msg)
            //}(msg)
         }
      }
   }
@@ -113,7 +128,7 @@
func paramFormat(msg []byte, args *ruleserver.SdkDatas) protomsg.SdkMessage {
   defer func() {
      if err := recover(); err != nil {
         logger.Info("解包过程的错误", err.(string))
         logger.Info("解包过程的异常捕获", err.(string))
      }
   }()
@@ -125,7 +140,7 @@
   }
   // 先进行一下追踪
   //ruleserver.FaceIsSame(&m)
   ruleserver.FaceIsSame(&m)
   args.CameraId = m.Cid
   args.TaskId = m.Tasklab.Taskid
   // 把图片的二进制解压缩进行画框在压缩回去