common/conf.go

package common
import (
	"encoding/json"
	"fmt"
	"gopkg.in/yaml.v3"
	"io/ioutil"
	"pro2d/common/logger"
	"pro2d/common/snow"
	"strings"
)
type RedisConf struct {
	Address string `json:"address"`
	Auth    string `json:"auth"`
	DB      int    `json:"db"`
}
type Etcd struct {
	Endpoints   []string `json:"endpoints"`
	DialTimeout int      `json:"dialtimeout"`
}
type MongoConf struct {
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Host     string `yaml:"host"`
	Port     int    `yaml:"port"`
	TimeOut  int    `yaml:"timeout"`
	MaxNum   int    `yaml:"maxnum"`
	DBName   string `yaml:"dbname"`
}
type SConf struct {
	ID             string     `yaml:"id"`
	Name           string     `yaml:"name"`
	IP             string     `yaml:"ip"`
	Port           int        `yaml:"port"`
	Encipher       bool       `yaml:"encipher"`
	DebugPort      int        `yaml:"debugport"`
	MongoConf      *MongoConf `yaml:"mongo"`
	RedisConf      *RedisConf `yaml:"redis"`
	WorkerPoolSize int        `yaml:"pool_size"`
	PluginPath     string     `yaml:"plugin_path"`
}
type LogConsole struct {
	Level string `yaml:"level" json:"level"`
	Color bool   `yaml:"color" json:"color"`
}
type LogFile struct {
	Level    string `yaml:"level" json:"level"`
	Daily    bool   `yaml:"daily" json:"daily"`
	Maxlines int    `yaml:"maxlines" json:"maxlines"`
	Maxsize  int    `yaml:"maxsize" json:"maxsize"`
	Maxdays  int    `yaml:"maxdays" json:"maxdays"`
	Append   bool   `yaml:"append" json:"append"`
	Permit   string `yaml:"permit" json:"permit"`
}
type LogConn struct {
	Net            string `yaml:"net" json:"net"`
	Addr           string `yaml:"addr" json:"addr"`
	Level          string `yaml:"level" json:"level"`
	Reconnect      bool   `yaml:"reconnect" json:"reconnect"`
	ReconnectOnMsg bool   `yaml:"reconnectOnMsg" json:"reconnectOnMsg"`
}
type LogConf struct {
	TimeFormat string      `yaml:"TimeFormat" json:"TimeFormat"`
	LogConsole *LogConsole `yaml:"Console" json:"Console"`
	LogFile    *LogFile    `yaml:"File" json:"File"`
	LogConn    *LogConn    `yaml:"Conn" json:"Conn"`
}
type TestClient struct {
	Ip    string `yaml:"ip"`
	Port  int    `yaml:"port"`
	Count int    `yaml:"count"`
}
type ServerConf struct {
	ID           string      `yaml:"id"`
	Name         string      `yaml:"name"`
	WorkerID     int64       `yaml:"workerid"`
	DatacenterID int64       `yaml:"datacenterid"`
	AccountConf  *SConf      `yaml:"server_account"`
	GameConf     *SConf      `yaml:"server_game"`
	LogConf      *LogConf    `yaml:"logconf" json:"logconf"`
	TestClient   *TestClient `yaml:"test_client"`
	Etcd         *Etcd       `yaml:"etcd"`
}
var (
	GlobalConf ServerConf
	SnowFlack  *snow.Snowflake
)
func init() {
	configFile, err := ioutil.ReadFile("conf/conf.yaml")
	if err != nil {
		fmt.Printf("conf faild: %v", err)
		return
	}
	//初始化配置
	if err = yaml.Unmarshal(configFile, &GlobalConf); err != nil {
		fmt.Printf("yaml unmarshal faild: %v", err)
		return
	}
	c, err := json.Marshal(&GlobalConf.LogConf)
	if err != nil {
		fmt.Errorf("log conf %v", err)
		return
	}
	//初始化日志
	err = logger.SetLogger(string(c), strings.ToLower(GlobalConf.GameConf.Name))
	if err != nil {
		fmt.Errorf("log conf %v", err)
		return
	}
	//初始化雪花算法
	SnowFlack = snow.NewSnowflake(GlobalConf.WorkerID, GlobalConf.DatacenterID)
}