go-common/app/admin/main/esports/conf/conf.go
2019-04-22 18:49:16 +08:00

111 lines
1.8 KiB
Go

package conf
import (
"errors"
"flag"
"go-common/library/conf"
"go-common/library/database/orm"
"go-common/library/log"
bm "go-common/library/net/http/blademaster"
"go-common/library/net/http/blademaster/middleware/permit"
"go-common/library/net/rpc/warden"
"go-common/library/net/trace"
"github.com/BurntSushi/toml"
)
var (
confPath string
client *conf.Client
// Conf of config
Conf = &Config{}
)
// Config def.
type Config struct {
// base
// http
BM *bm.ServerConfig
// auth
Permit *permit.Config
// db
ORM *orm.Config
// log
Log *log.Config
// tracer
Tracer *trace.Config
// rule
Rule *Rule
// client
HTTPReply *bm.ClientConfig
// Warden Client
ArcClient *warden.ClientConfig
AccClient *warden.ClientConfig
// GameTypes game types.
GameTypes []*types
}
// Rule .
type Rule struct {
MaxCSVRows int
MaxBatchArcLimit int
MaxTreeContests int
}
type types struct {
ID int64
Name string
}
func local() (err error) {
_, err = toml.DecodeFile(confPath, &Conf)
return
}
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.Value("esports-admin.toml"); !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
}
func init() {
flag.StringVar(&confPath, "conf", "", "default config path")
}
// Init int config
func Init() error {
if confPath != "" {
return local()
}
return remote()
}