toolbar/precog/config/config.go

   1 package config
   2
   3 import (
   4         "encoding/json"
   5         "os"
   6
   7         log "github.com/sirupsen/logrus"
   8         "github.com/vapor/crypto/ed25519/chainkd"
   9
  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 }
  35
  36 type Config struct {
  37         MySQLConfig      common.MySQLConfig `json:"mysql"`
  38         CheckFreqSeconds uint64             `json:"check_frequency_seconds"`
  39         Policy           Policy             `json:"policy"`
  40         Nodes            []Node             `json:"bootstrap_nodes"`
  41         API              API                `json:"api"`
  42 }
  43
  44 type Policy struct {
  45         Confirmations      uint64 `json:"confirmations"`
  46         RequiredLantencyMS uint64 `json:"required_lantency_ms"`
  47 }
  48
  49 type Node struct {
  50         Alias     string       `json:"alias"`
  51         PublicKey chainkd.XPub `json:"public_key"`
  52         Host      string       `json:"host"`
  53         Port      uint16       `json:"port"`
  54 }
  55
  56 type API struct {
  57         ListeningPort bool   `json:"listening_port"`
  58         AccessToken   string `json:"access_token"`
  59         IsReleaseMode bool   `json:"is_release_mode"`
  60 }