DelayNoMore/battle_srv/common/conf.go

package common

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"go.uber.org/zap"
)

// 隐式导入
var Conf *config

const (
	APP_NAME        = "server"
	SERVER_ENV_PROD = "PROD"
	SERVER_ENV_TEST = "TEST"
)

type generalConf struct {
	AppRoot           string `json:"-"`
	ConfDir           string `json:"-"`
	TestEnvSQLitePath string `json:"-"`
	PreConfSQLitePath string `json:"-"`
	ServerEnv         string `json:"-"`
}

type mysqlConf struct {
	DSN      string `json:"-"`
	Host     string `json:"host"`
	Port     int    `json:"port"`
	Dbname   string `json:"dbname"`
	Username string `json:"username"`
	Password string `json:"password"`
}

type sioConf struct {
	HostAndPort string `json:"hostAndPort"`
}

type botServerConf struct {
	SecondsBeforeSummoning int    `json:"secondsBeforeSummoning"`
	Protocol               string `json:"protocol"`
	Host                   string `json:"host"`
	Port                   int    `json:"port"`
	SymmetricKey           string `json:"symmetricKey"`
}

type redisConf struct {
	Dbname   int    `json:"dbname"`
	Host     string `json:"host"`
	Password string `json:"password"`
	Port     int    `json:"port"`
}

type config struct {
	General   *generalConf
	MySQL     *mysqlConf
	Sio       *sioConf
	Redis     *redisConf
	BotServer *botServerConf
}

func MustParseConfig() {
	Conf = &config{
		General:   new(generalConf),
		MySQL:     new(mysqlConf),
		Sio:       new(sioConf),
		Redis:     new(redisConf),
		BotServer: new(botServerConf),
	}
	execPath, err := os.Executable()
	ErrFatal(err)

	pwd, err := os.Getwd()
	Logger.Debug("os.GetWd", zap.String("pwd", pwd))
	ErrFatal(err)

	appRoot := pwd
	confDir := filepath.Join(appRoot, "configs")
	Logger.Debug("conf", zap.String("dir", confDir))
	if isNotExist(confDir) {
		appRoot = filepath.Dir(execPath)
		confDir = filepath.Join(appRoot, "configs")
		Logger.Debug("conf", zap.String("dir", confDir))
		if isNotExist(confDir) {
			i := strings.LastIndex(pwd, "battle_srv")
			if i == -1 {
				Logger.Fatal("无法找到配置目录，cp -rn configs.template configs，并配置相关参数，再启动")
			}
			appRoot = pwd[:(i + 10)]
			confDir = filepath.Join(appRoot, "configs")
			Logger.Debug("conf", zap.String("dir", confDir))
			if isNotExist(confDir) {
				Logger.Fatal("无法找到配置目录，cp -rn configs.template configs，并配置相关参数，再启动")
			}
		}
	}
	Conf.General.AppRoot = appRoot
	testEnvSQLitePath := filepath.Join(confDir, "test_env.sqlite")
	if !isNotExist(testEnvSQLitePath) {
		Conf.General.TestEnvSQLitePath = testEnvSQLitePath
	}
	preConfSQLitePath := filepath.Join(confDir, "pre_conf_data.sqlite")
	if !isNotExist(preConfSQLitePath) {
		Conf.General.PreConfSQLitePath = preConfSQLitePath
	}
	Conf.General.ConfDir = confDir
	Conf.General.ServerEnv = os.Getenv("ServerEnv")

	loadJSON("mysql.json", Conf.MySQL)
	setMySQLDSNURL(Conf.MySQL)
	loadJSON("sio.json", Conf.Sio)
	loadJSON("redis.json", Conf.Redis)
	loadJSON("bot_server.json", Conf.BotServer)
}

func setMySQLDSNURL(c *mysqlConf) {
	var dsn = fmt.Sprintf("%s:%s@tcp(%s:%d)/%s",
		c.Username, c.Password, c.Host, c.Port, c.Dbname)
	c.DSN = dsn

}

func loadJSON(fp string, v interface{}) {
	if !filepath.IsAbs(fp) {
		fp = filepath.Join(Conf.General.ConfDir, fp)
	}
	_, err := os.Stat(fp)
	ErrFatal(err)

	fd, err := os.Open(fp)
	ErrFatal(err)
	defer fd.Close()
	Logger.Info("Opened json file successfully.", zap.String("fp", fp))
	err = json.NewDecoder(fd).Decode(v)
	ErrFatal(err)
	Logger.Info("Loaded json file successfully.", zap.String("fp", fp))
}

// Please only use this auxiliary function before server is fully started up, but not afterwards (启动过程可以使用，运行时不准使用).
func ErrFatal(err error) {
	if err != nil {
		Logger.Fatal("ErrFatal", zap.NamedError("err", err))
	}
}

func isNotExist(p string) bool {
	if _, err := os.Stat(p); err != nil {
		return true
	}
	return false
}