171 lines
3.1 KiB
Go
171 lines
3.1 KiB
Go
package conf
|
|
|
|
import (
|
|
"flag"
|
|
|
|
"go-common/library/cache/memcache"
|
|
"go-common/library/cache/redis"
|
|
"go-common/library/conf"
|
|
"go-common/library/database/sql"
|
|
ecode "go-common/library/ecode/tip"
|
|
"go-common/library/log"
|
|
"go-common/library/log/infoc"
|
|
bm "go-common/library/net/http/blademaster"
|
|
"go-common/library/net/http/blademaster/middleware/antispam"
|
|
"go-common/library/net/http/blademaster/middleware/supervisor"
|
|
"go-common/library/net/http/blademaster/middleware/verify"
|
|
"go-common/library/net/rpc"
|
|
"go-common/library/net/rpc/warden"
|
|
"go-common/library/net/trace"
|
|
"go-common/library/queue/databus"
|
|
"go-common/library/time"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
var (
|
|
confPath string
|
|
// Conf Config
|
|
Conf *Config
|
|
)
|
|
|
|
// Config is favorte conf
|
|
type Config struct {
|
|
// base
|
|
// log
|
|
Log *log.Config
|
|
App *bm.App
|
|
// favorite config
|
|
Fav *Fav
|
|
Platform *Platform
|
|
// BM blademaster
|
|
BM *bm.ServerConfig
|
|
// rpc server2
|
|
RPCServer *rpc.ServerConfig
|
|
// db
|
|
MySQL *MySQL
|
|
// redis
|
|
Redis *Redis
|
|
// memcache
|
|
Memcache *Memcache
|
|
// databus
|
|
JobDatabus *databus.Config
|
|
// verify
|
|
Verify *verify.Config
|
|
// rpc client
|
|
RPCClient2 *RPC
|
|
// tracer
|
|
Tracer *trace.Config
|
|
// http client
|
|
HTTPClient *bm.ClientConfig
|
|
// ecode
|
|
Ecode *ecode.Config
|
|
// TopicClient
|
|
Topic *Topic
|
|
// Antispam
|
|
Antispam *antispam.Config
|
|
// Supervisior
|
|
Supervisor *supervisor.Config
|
|
// collector
|
|
Infoc2 *infoc.Config
|
|
//grpc warden
|
|
WardenServer *warden.ServerConfig
|
|
}
|
|
|
|
// RPC contain all rpc conf
|
|
type RPC struct {
|
|
Account *warden.ClientConfig
|
|
Archive *rpc.ClientConfig
|
|
Filter *rpc.ClientConfig
|
|
Rank *rpc.ClientConfig
|
|
}
|
|
|
|
// Topic Topic
|
|
type Topic struct {
|
|
TopicURL string
|
|
HTTPClient *bm.ClientConfig
|
|
}
|
|
|
|
// Fav config
|
|
type Fav struct {
|
|
// the max of the num of favorite folders
|
|
MaxFolders int
|
|
MaxPagesize int
|
|
MaxBatchSize int
|
|
MaxDataSize int
|
|
MaxParallelSize int
|
|
MaxRecentSize int
|
|
MaxNameLen int
|
|
MaxDescLen int
|
|
// the num of operation
|
|
MaxOperationNum int
|
|
// the num of default favorite
|
|
DefaultFolderLimit int
|
|
NormalFolderLimit int
|
|
// ApiHost api.bilibili.co .
|
|
APIHost string
|
|
// cache expire
|
|
Expire time.Duration
|
|
// cdtime cool down time
|
|
CleanCDTime time.Duration
|
|
// real-name switch
|
|
RealNameOn bool
|
|
}
|
|
|
|
// Platform config
|
|
type Platform struct {
|
|
MaxFolders int
|
|
MaxNameLen int
|
|
MaxDescLen int
|
|
}
|
|
|
|
// MySQL is mysql conf
|
|
type MySQL struct {
|
|
// favorite db
|
|
Fav *sql.Config
|
|
Read *sql.Config
|
|
Push *sql.Config
|
|
}
|
|
|
|
// Redis redis conf
|
|
type Redis struct {
|
|
*redis.Config
|
|
Expire time.Duration
|
|
CoverExpire time.Duration
|
|
}
|
|
|
|
// Memcache memcache conf
|
|
type Memcache struct {
|
|
*memcache.Config
|
|
Expire time.Duration
|
|
}
|
|
|
|
func init() {
|
|
flag.StringVar(&confPath, "conf", "", "config path")
|
|
}
|
|
|
|
// Init init conf
|
|
func Init() (err error) {
|
|
if confPath == "" {
|
|
return configCenter()
|
|
}
|
|
_, err = toml.DecodeFile(confPath, &Conf)
|
|
return
|
|
}
|
|
|
|
func configCenter() (err error) {
|
|
var (
|
|
ok bool
|
|
value string
|
|
client *conf.Client
|
|
)
|
|
if client, err = conf.New(); err != nil {
|
|
return
|
|
}
|
|
if value, ok = client.Toml2(); !ok {
|
|
panic(err)
|
|
}
|
|
_, err = toml.Decode(value, &Conf)
|
|
return
|
|
}
|