package conf import ( "context" "fmt" "sync" "go-common/app/interface/openplatform/monitor-end/model/kafka" "go-common/app/interface/openplatform/monitor-end/model/monitor" "go-common/library/cache/redis" "go-common/library/conf/paladin" "go-common/library/database/sql" "go-common/library/log" "go-common/library/log/infoc" bm "go-common/library/net/http/blademaster" "github.com/BurntSushi/toml" ) var ( confPath string // Conf is global config Conf = &Config{Prom: &Prom{}} ) // Config service config type Config struct { // http BM *bm.ServerConfig // XLog Log *log.Config // monitor Monitor *monitor.MonitorConfig // kafka Kafka *kafka.Config NeedConsume bool // Prometheus Prom *Prom mutex sync.Mutex // mysql MySQL *sql.Config // redis Redis *redis.Config // infoc CollectInfoc *infoc.Config // collect fronend log CollectFE bool // products Products string } // Prom . type Prom struct { Promed bool IgnoreNA bool Factor int Limit uint64 } // Set . func (e *Config) Set(text string) error { var ec Config if err := toml.Unmarshal([]byte(text), &ec); err != nil { return err } if ec.Prom == nil { e.Prom = &Prom{} return nil } e.mutex.Lock() e.Prom = &Prom{ Factor: ec.Prom.Factor, Promed: ec.Prom.Promed, Limit: ec.Prom.Limit, IgnoreNA: ec.Prom.IgnoreNA, } e.CollectFE = ec.CollectFE e.Products = ec.Products e.mutex.Unlock() // *e = ec return nil } // Init . func Init() { if err := paladin.Init(); err != nil { panic(err) } // var ec exampleConf // var setter if err := paladin.Watch("monitor-end.toml", Conf); err != nil { panic(err) } if err := paladin.Get("monitor-end.toml").UnmarshalTOML(Conf); err != nil { panic(err) } // use exampleConf // watch key go func() { for event := range paladin.WatchEvent(context.TODO(), "monitor-end.toml") { fmt.Println(event, Conf) } }() } // func init() { // flag.StringVar(&confPath, "conf", "", "config path") // } // func local() (err error) { // _, err = toml.DecodeFile(confPath, &Conf) // return // } // Init int config // func Init() error { // if confPath != "" { // return local() // } // return remote() // } // func remote() (err error) { // if client, err = conf.New(); err != nil { // return // } // if err = load(); err != nil { // return // } // go func() { // for range client.Event() { // log.Info("config reload") // if load() != nil { // log.Error("config reload error (%v)", err) // } // } // }() // return // } // func load() (err error) { // var ( // s string // ok bool // tmpConf *Config // ) // if s, ok = client.Toml2(); !ok { // return errors.New("load config center error") // } // if _, err = toml.Decode(s, &tmpConf); err != nil { // return errors.New("could not decode config") // } // *Conf = *tmpConf // return // }