Blame view

conf/conf.go 1.87 KB
ee23102d   zhangqijia   支持mongo, grpc接服务器
1
2
3
4
5
6
7
  package conf
  
  import (
  	"fmt"
  	lumberjack "gopkg.in/natefinch/lumberjack.v2"
  	"gopkg.in/yaml.v3"
  	"io/ioutil"
1584eb4b   zhangqijia   修复创建唯一索引的bug
8
  	"pro2d/components/db"
ee23102d   zhangqijia   支持mongo, grpc接服务器
9
10
11
12
13
14
15
16
17
  	"pro2d/utils"
  )
  
  type RedisConf struct {
  	Address string `json:"address"`
  	Auth 	string `json:"auth"`
  	DB 		int 		`json:"db"`
  }
  
ee23102d   zhangqijia   支持mongo, grpc接服务器
18
  type Etcd struct {
286f6dbe   zhangqijia   etcd 保存服务信息
19
20
  	Endpoints []string `json:"endpoints"`
  	DialTimeout int		`json:"dialtimeout"`
ee23102d   zhangqijia   支持mongo, grpc接服务器
21
22
23
24
25
26
27
  }
  
  type MongoConf struct {
  	User 		string		`yaml:"user"`
  	Password 	string		`yaml:"password"`
  	Host  		string		`yaml:"host"`
  	Port 		int			`yaml:"port"`
ee23102d   zhangqijia   支持mongo, grpc接服务器
28
29
30
31
32
  	TimeOut  	int			`yaml:"timeout"`
  	MaxNum  	int			`yaml:"maxnum"`
  }
  
  type SConf struct {
286f6dbe   zhangqijia   etcd 保存服务信息
33
  	ID string `yaml:"id"`
ee23102d   zhangqijia   支持mongo, grpc接服务器
34
35
36
  	Name string `yaml:"name"`
  	IP   string `yaml:"ip"`
  	Port int 	`yaml:"port"`
eb417b0b   zhangqijia   reactor mongo
37
  	DBName  	string		`yaml:"dbname"`
ee23102d   zhangqijia   支持mongo, grpc接服务器
38
39
40
41
42
43
44
  }
  
  type ServerConf struct {
  	ID 				string 		`yaml:"id"`
  	Name 			string 		`yaml:"name"`
  	WorkerID    	int64		`yaml:"workerid"`
  	DatacenterID 	int64		`yaml:"datacenterid"`
eb417b0b   zhangqijia   reactor mongo
45
  	MongoConf   	*MongoConf	`yaml:"mongo"`
ee23102d   zhangqijia   支持mongo, grpc接服务器
46
47
48
49
50
51
52
  	AccountConf 	*SConf 		`yaml:"server_account"`
  	GameConf		*SConf 		`yaml:"server_game"`
  	RedisConf   	*RedisConf  `yaml:"redis"`
  	LogConf  		*lumberjack.Logger `json:"logconf"`
  	Etcd 			*Etcd 		`yaml:"etcd"`
  }
  
eb417b0b   zhangqijia   reactor mongo
53
54
55
56
  var(
  	GlobalConf ServerConf
  	SnowFlack *utils.Snowflake
  )
ee23102d   zhangqijia   支持mongo, grpc接服务器
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
  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
  	}
  
  	//初始化日志
  	utils.InitLogger(GlobalConf.LogConf)
  
  	//初始化雪花算法
  	SnowFlack = utils.NewSnowflake(GlobalConf.WorkerID, GlobalConf.DatacenterID)
eb417b0b   zhangqijia   reactor mongo
74
  
1584eb4b   zhangqijia   修复创建唯一索引的bug
75
76
77
78
  	err = db.Connect(GlobalConf.MongoConf.User, GlobalConf.MongoConf.Password, GlobalConf.MongoConf.Host, GlobalConf.MongoConf.Port, GlobalConf.MongoConf.MaxNum, GlobalConf.MongoConf.TimeOut)
  	if err != nil {
  		utils.Sugar.Errorf("connect db err: %v", err)
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
79
  }