添加日志系统

panlei

2019-07-05 40e90ca2c934c8505f2a9e9b1c4f076657008369

 main.go

@@ -1,40 +1,56 @@
package main

import (
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "bufio"
   "bytes"
   "encoding/json"
   "errors"
   "flag"
   "fmt"
   "github.com/golang/protobuf/proto"
   "gocv.io/x/gocv"
   "image"
   "image/color"
   "io"
   "log"
   "mime/multipart"
   "net/http"
   "os"
   "ruleprocess/cache"
   "ruleprocess/insertdata"
   "ruleprocess/ruleserver"
   "ruleprocess/util"
   "time"

   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "github.com/golang/protobuf/proto"
   "ruleprocess/logger"
   "sync"
)
var dbIp = flag.String("dbIp","127.0.0.1","dbserver ip")
var dbPort = flag.Int("dbPort",8001,"default dbPort=8001")
var surveyPort = flag.Int("surveyPort",40007,"survey port") //心跳
var pubPort = flag.Int("pubPort",50007,"pubsub port") //数据更新
var initchan = make(chan bool)
func init () {
   var logFile = "/opt/ruleLog/error.log"
   var logSaveDays   =   15

   // 日志初始化
   logger.Config(logFile, logger.DebugLevel)
   logger.SetSaveDays(logSaveDays)
   logger.Info("loginit success !")
}
func main() {
   //wg := sync.WaitGroup{}
   //wg.Add(1)
   //go ruleserver.TimeTicker()
   //nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1)
   //wg.Wait()
   resp,err := DrawPolygonOnImage()
   if err != nil {
      log.Println("画框或者上传有问题")
   }
   log.Println(resp["fileUrl"].(string))
   flag.Parse()
   wg := sync.WaitGroup{}
   wg.Add(1)

   dbapi.Init(*dbIp,*dbPort)
   go cache.Init(initchan,*dbIp,*surveyPort,*pubPort)
   fmt.Println("cache init completed!!!",<- initchan)//dbserver初始化完毕

   go ruleserver.TimeTicker()
   nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1)
   wg.Wait()
   //resp,err := DrawPolygonOnImage()
   //if err != nil {
   //   logger.Println("画框或者上传有问题")
   //}
   //logger.Println(resp["fileUrl"].(string))
}
func nReciever(url string, m deliver.Mode, count int) {
   c := deliver.NewServer(m, url)
@@ -103,7 +119,7 @@
   }
   i := protomsg.Image{}
   err = proto.Unmarshal(bdata, &i)

   logger.Info("----------------看看有几个算法：",len(m.Tasklab.Sdkinfos))
   for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo算法
      if sdkinfo.Sdktype == "Yolo" {
         arg := ruleserver.SdkData{}
@@ -124,13 +140,13 @@
         arg.IsStatic = false
         if len(sdkinfo.Sdkdata) > 1 {
            // 大于1才有数据
            fmt.Println("----------------------------------------------------",m.Cid)
            yoloParam := protomsg.ParamYoloObj{}
            err = proto.Unmarshal(sdkinfo.Sdkdata, &yoloParam)
            if err != nil {
               fmt.Println("解析YOLO sdk数据时出现错误", err)
               continue
            }
            logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看yolo中有几个目标：",len(yoloParam.Infos))
            for _, info := range yoloParam.Infos {
               if info.Typ == 0 {
                  photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob)*100,IsYolo:true}
@@ -161,13 +177,13 @@
         arg.KeepRight = false
         arg.IsStatic = false
         if len(sdkinfo.Sdkdata) > 1 {
            fmt.Println("----------------------------------------------------",m.Caddr)
            faceParam := protomsg.ParamFacePos{}
            err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
            if err != nil {
               fmt.Println("解析FACE sdk数据时出现错误", err)
               continue
            }
            logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看人脸检测中有几个目标：",len(faceParam.Faces))
            for _, info := range faceParam.Faces {
               photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.Quality)*100, IsYolo:false,ThftRes:*(info.Result)}
               arg.Photo = append(arg.Photo, photoMap)
@@ -195,13 +211,22 @@
         arg.KeepRight = false
         arg.IsStatic = false
         if len(sdkinfo.Sdkdata) > 1 {
            fmt.Println("----------------------------------------------------",m.Caddr)
            faceParam := protomsg.ParamFaceFeature{}
            err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam)
            if err != nil {
               fmt.Println("解析FACE sdk数据时出现错误", err)
               continue
            }
            for _,faceExtCom := range faceParam.ExtComp  {
               binfos := protomsg.Binfos{}
               err := proto.Unmarshal(faceExtCom.Comp, &binfos)
               if err != nil {
                  fmt.Println("解析人脸比对数据时出现错误", err)
                  continue
               }
               logger.Info("_________________________________________________________________解析出的人脸的相似的底库数据",binfos)
            }
            logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看人脸提取中有几个目标：",len(faceParam.ExtComp))
            for _, extComp := range faceParam.ExtComp {
               baseinfos := protomsg.Binfos{}
               err1 := proto.Unmarshal(extComp.Comp, &baseinfos)
@@ -264,7 +289,7 @@
   //client := &http.Client{Timeout: timeout}
   //resp, err := client.Do(request)
   //if err != nil {
   //   log.Fatal(err)
   //   logger.Fatal(err)
   //   return nil, err
   //}
   //defer func() {
@@ -287,7 +312,7 @@
   //   body := &bytes.Buffer{}
   //   _, err := body.ReadFrom(resp.Body)
   //   if err != nil {
   //      log.Fatal(err)
   //      logger.Fatal(err)
   //   }
   //   fmt.Println(resp.StatusCode)
   //   //fmt.Println(resp.Header)
@@ -298,7 +323,7 @@
   //      return nil, err
   //   }
      return nil, nil
   }
   //}
}

// 把图片转成二进制流