102 lines
1.9 KiB
Go
102 lines
1.9 KiB
Go
package conf
|
|
|
|
import (
|
|
"errors"
|
|
"flag"
|
|
"image/color"
|
|
|
|
"go-common/library/cache/memcache"
|
|
"go-common/library/conf"
|
|
ecode "go-common/library/ecode/tip"
|
|
"go-common/library/log"
|
|
bm "go-common/library/net/http/blademaster"
|
|
"go-common/library/net/http/blademaster/middleware/rate"
|
|
"go-common/library/net/http/blademaster/middleware/verify"
|
|
"go-common/library/net/trace"
|
|
"go-common/library/time"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
)
|
|
|
|
var (
|
|
confPath string
|
|
// Conf .
|
|
Conf = &Config{}
|
|
)
|
|
|
|
// Config captcha service config struct.
|
|
type Config struct {
|
|
XLog *log.Config
|
|
Tracer *trace.Config
|
|
Ecode *ecode.Config
|
|
BM *HTTPServers
|
|
Verify *verify.Config
|
|
Rate *rate.Config
|
|
Memcache *Memcache
|
|
Captcha *Captcha
|
|
Business []*Business
|
|
}
|
|
|
|
// Memcache represent mc conf
|
|
type Memcache struct {
|
|
*memcache.Config
|
|
Expire time.Duration
|
|
}
|
|
|
|
// HTTPServers Http Servers
|
|
type HTTPServers struct {
|
|
Outer *bm.ServerConfig
|
|
}
|
|
|
|
// Business third business confs.
|
|
type Business struct {
|
|
BusinessID string
|
|
LenStart int
|
|
LenEnd int
|
|
Width int
|
|
Length int
|
|
TTL time.Duration
|
|
}
|
|
|
|
// Captcha captcha service conf.
|
|
type Captcha struct {
|
|
OuterHost string
|
|
Capacity int
|
|
DisturbLevel int // 4 normal, 8 medium, 16 high
|
|
Ext string // jpeg
|
|
Fonts []string
|
|
BkgColors []color.RGBA
|
|
FrontColors []color.RGBA
|
|
Interval time.Duration
|
|
}
|
|
|
|
func init() {
|
|
flag.StringVar(&confPath, "conf", "", "config path")
|
|
}
|
|
|
|
// Init captcha service init.
|
|
func Init() (err error) {
|
|
if confPath == "" {
|
|
return configCenter()
|
|
}
|
|
_, err = toml.DecodeFile(confPath, &Conf)
|
|
return
|
|
}
|
|
|
|
// configCenter connect to config center, get configs.
|
|
func configCenter() (err error) {
|
|
var (
|
|
client *conf.Client
|
|
value string
|
|
ok bool
|
|
)
|
|
if client, err = conf.New(); err != nil {
|
|
return
|
|
}
|
|
if value, ok = client.Toml2(); !ok {
|
|
return errors.New("load config center error")
|
|
}
|
|
_, err = toml.Decode(value, &Conf)
|
|
return
|
|
}
|