go语言之程序中获取环境变量

“–————人与人之间还是直接点好——————”

go如何通过程序中获取环境变量到达控制效果~
如下是个测试程序,主要是用户通过启动程序前 设置环境变量

root$ export LOG_NUMBER=3

进而达到控制日志的等级的效果!其它功能可同理!

func init() {

 if fileNumber := os.Getenv("LOG_NUMBER"); fileNumber != "" {
 
  toInt, err := strconv.Atoi(fileNumber)
  
  if err != nil {
   log.Error("set LOG_NUMBER failed! example: export LOG_NUMBER=7 ", err)
   os.Exit(0)
  }
  
  log.SetLevel(log.DebugLevel)
  
 } else {
 
  log.SetLevel(log.DebugLevel)
  
 }
 
 // Log as JSON instead of the default ASCII formatter.
 log.SetFormatter(&log.JSONFormatter{})
 
 // Output to stdout instead of the default stderr
 // Can be any io.Writer, see below for File example
 log.SetOutput(logger)
 
 // Only log the warning severity or above.
 if t := os.Getenv("MQTT_DEBUG"); t != "" {
 
  log.SetLevel(log.DebugLevel)
  
 } else {
 
  log.SetLevel(log.WarnLevel)
  
 }
 
}

你可能感兴趣的:(Golang,go,linux)