toolbar/precog/config/config.go

   1 package config
   2
   3 import (
   4         "encoding/json"
   5         "os"
   6
   7         log "github.com/sirupsen/logrus"
   8
   9         vaporJson "github.com/vapor/encoding/json"
  10         "github.com/vapor/toolbar/common"
  11 )
  12
  13 func NewConfig() *Config {
  14         if len(os.Args) <= 1 {
  15                 log.Fatal("Please setup the config file path")
  16         }
  17
  18         return NewConfigWithPath(os.Args[1])
  19 }
  20
  21 func NewConfigWithPath(path string) *Config {
  22         configFile, err := os.Open(path)
  23         if err != nil {
  24                 log.WithFields(log.Fields{"err": err, "file_path": os.Args[1]}).Fatal("fail to open config file")
  25         }
  26         defer configFile.Close()
  27
  28         cfg := &Config{}
  29         if err := json.NewDecoder(configFile).Decode(cfg); err != nil {
  30                 log.WithField("err", err).Fatal("fail to decode config file")
  31         }
  32
  33         return cfg
  34 }