update

zhangmeng

2019-12-16 a61cd237d45c7bec7a3e0cdf32ed33c49b41b73b

 main.go

@@ -5,26 +5,50 @@
    "flag"
   _ "net/http/pprof"
    "time"
   "fmt"

   "taskpubsub/camera"
   "taskpubsub/sdk"
   "taskpubsub/tasktag"
   "taskpubsub/util"
    "taskpubsub/logger"

   "basic.com/valib/logger.git"
   "github.com/spf13/viper"
)

var initchan = make(chan bool)

var useShm bool

const (
   configFilePath  = "/opt/vasystem/config/"
   configFileType = "yaml"
   LOGBASEPATH = "/data/disk1/valog/taskpubsub.log"
)

var envirment = flag.String("e", "pro", "")

func init(){
    var logFile = "./taskpubsub.log"
    var logSaveDays    =    15
    
   viper.SetConfigType(configFileType)
   viper.SetConfigName(*envirment)
   viper.AddConfigPath(configFilePath)
   viper.AddConfigPath("./")
   err := viper.ReadInConfig()
   if err != nil {
      logger.Error("viper.ReadInConfig() Error")
      //panic(err)
   }

    var logFile = LOGBASEPATH
   if viper.GetString("LogBasePath") != "" {
      logFile = viper.GetString("LogBasePath") + "/taskpubsub.log"
   }

    // 日志初始化
    logger.Config(logFile, logger.InfoLevel)
   if viper.IsSet("LogLevel") &&
      viper.GetInt("LogLevel") >= logger.PanicLevel &&
      viper.GetInt("LogLevel") <= logger.DebugLevel {
      logger.Config(logFile, viper.GetInt("LogLevel"))
   }else{
      logger.Config(logFile, logger.DebugLevel)
   }
   var logSaveDays    =    15
    logger.SetSaveDays(logSaveDays)
   logger.Info("loginit success !")
   
@@ -37,10 +61,8 @@

   if useShm{
      logger.Info("USE SHARE MEMORY")
      fmt.Println("USE SHARE MEMORY")
   }else{
      logger.Info("USE PIPE")
      fmt.Println("USE PIPE")
   }
   // pprof 用于分析性能
   go func() {