You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
159 lines
3.4 KiB
Go
159 lines
3.4 KiB
Go
package conf
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/bwmarrin/snowflake"
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
type appcfg struct {
|
|
Redis struct {
|
|
Ip string `json:"ip"`
|
|
Port int `json:"port"`
|
|
} `json:"redis"`
|
|
|
|
Queue struct {
|
|
Ip string `json:"ip"`
|
|
Port int `json:"port"`
|
|
} `json:"queue"`
|
|
Mysql struct {
|
|
Ip string `json:"ip"`
|
|
Port int `json:"port"`
|
|
User string `json:"user"`
|
|
Password string `json:"password"`
|
|
Db string `json:"db"`
|
|
} `json:"mysql"`
|
|
Mongo struct {
|
|
Ip string `json:"ip"`
|
|
Port int `json:"port"`
|
|
} `json:"mongo"`
|
|
Global struct {
|
|
Gamename string `json:"gamename"`
|
|
Version string `json:"version"`
|
|
} `json:"global"`
|
|
}
|
|
|
|
// user:password@tcp(host:port)/dbname?charset=utf8mb4
|
|
// 用户名:密码@tcp(主机:端口)/数据库名称?charset=utf8mb4
|
|
const _dsn = "%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=true&loc=Local"
|
|
|
|
const _addr = "%s:%d"
|
|
|
|
const _mongo_addr = "mongodb://%s:%d"
|
|
|
|
var (
|
|
defConfig = "./conf/dev.toml" //配置文件路径，方便测试
|
|
prodConfig = "./conf/prod.toml" //配置文件路径，方便测试
|
|
Config *appcfg //运行配置实体
|
|
APPNAME string
|
|
APPVERSION string
|
|
ISTEST bool
|
|
USEOTS bool
|
|
DSN string
|
|
RedisAddr string
|
|
QueueAddr string
|
|
MongoAddr string
|
|
|
|
ObtenationIterations int = 3
|
|
|
|
node, _ = snowflake.NewNode(1)
|
|
|
|
IP string = "192.168.3.235"
|
|
PORT int = 9080
|
|
)
|
|
|
|
func GetNode() *snowflake.Node {
|
|
return node
|
|
}
|
|
|
|
func GetDsn() string {
|
|
return DSN
|
|
}
|
|
|
|
func GetRedisAddr() string {
|
|
return RedisAddr
|
|
}
|
|
|
|
func GetQueueAddr() string {
|
|
return QueueAddr
|
|
}
|
|
|
|
func GetConfig() *appcfg {
|
|
return Config
|
|
}
|
|
|
|
func GetMongoAddr() string {
|
|
return MongoAddr
|
|
}
|
|
|
|
func init() {
|
|
fmt.Println("conf init")
|
|
_, err := os.Stat(defConfig)
|
|
color.Red("Stat ok = %v", err)
|
|
if err != nil {
|
|
_, err1 := os.Stat(prodConfig)
|
|
if err1 != nil {
|
|
ISTEST = false
|
|
}
|
|
} else {
|
|
ISTEST = true
|
|
}
|
|
|
|
// ISTEST = false
|
|
if ISTEST {
|
|
color.Red("当前为测试环境")
|
|
var err error
|
|
Config, err = initDevCfg()
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
APPNAME = Config.Global.Gamename
|
|
color.Green("APPNAME=%s", APPNAME)
|
|
APPVERSION = Config.Global.Version
|
|
color.Green("APPVERSION=%s", APPVERSION)
|
|
|
|
color.Green("Mysql=%s", Config.Mysql)
|
|
DSN = fmt.Sprintf(_dsn, Config.Mysql.User, Config.Mysql.Password, Config.Mysql.Ip, Config.Mysql.Port, Config.Mysql.Db)
|
|
RedisAddr = fmt.Sprintf(_addr, Config.Redis.Ip, Config.Redis.Port)
|
|
QueueAddr = fmt.Sprintf(_addr, Config.Queue.Ip, Config.Queue.Port)
|
|
MongoAddr = fmt.Sprintf(_mongo_addr, Config.Mongo.Ip, Config.Mongo.Port)
|
|
|
|
} else {
|
|
color.Red("当前为生产环境")
|
|
var err error
|
|
Config, err = initProdCfg()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
APPNAME = Config.Global.Gamename
|
|
APPVERSION = Config.Global.Version
|
|
DSN = fmt.Sprintf(_dsn, Config.Mysql.User, Config.Mysql.Password, Config.Mysql.Ip, Config.Mysql.Port, Config.Mysql.Db)
|
|
RedisAddr = fmt.Sprintf(_addr, Config.Redis.Ip, Config.Redis.Port)
|
|
QueueAddr = fmt.Sprintf(_addr, Config.Queue.Ip, Config.Queue.Port)
|
|
MongoAddr = fmt.Sprintf(_mongo_addr, Config.Mongo.Ip, Config.Mongo.Port)
|
|
|
|
}
|
|
}
|
|
|
|
func initDevCfg() (*appcfg, error) {
|
|
app := &appcfg{}
|
|
_, err := toml.DecodeFile(defConfig, &app)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return app, nil
|
|
}
|
|
|
|
func initProdCfg() (*appcfg, error) {
|
|
app := &appcfg{}
|
|
_, err := toml.DecodeFile(prodConfig, &app)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return app, nil
|
|
}
|