Golagn 四種配置方式實現(xiàn)
說明
在實際的開發(fā)過程中,我們必然會用到MySQL、Redis等這樣的服務。為了實現(xiàn)系統(tǒng)的配置化,我們會把一些配置信息單獨放在一些文件中,使用到的地方直接讀取配置文件即可。常見的文件配置方式有很多種,例如json、tomal、yml或者文本格式。下面就針對幾種方式進行一一演示。
演示代碼
JSON配置
首先我們創(chuàng)建一個JSON的文件,里面配置我們需要的參數(shù)格式,示例:
- {
- "host": "127.0.0.1",
- "user": "root",
- "password": "123456",
- "port": "3306",
- "db": "demo"
- }
我們要讀取配文件,就需要用到Golang中自帶的json包。具體的讀取過程:讀取json文件內(nèi)容->使用json包進行反序列化->利用變量存反序列的數(shù)據(jù)。
- // 利用struct來定義json格式,與存儲。
- type DbJson struct {
- Host string `json:"host"`
- User string `json:"user"`
- Password string `json:"password"`
- Port string `json:"port"`
- Db string `json:"db"`
- }
- // 解析
- func GetJsonConfig() {
- // 1. 讀取json文件內(nèi)容
- file, err := ioutil.ReadFile("./config/json.json")
- if err != nil {
- fmt.Println("err1", err)
- return
- }
- db := new(DbJson)
- // 2. 將讀取到的json文件內(nèi)容,進行反序列化;將得到一個[]byte類型的切片
- err = json.Unmarshal(file, db)
- if err != nil {
- fmt.Println("err2", err)
- return
- }
- // 2.1 將讀取到的json文件內(nèi)容,進行反序列化,復制給map[string][]byte(和2中的效果是一樣的)
- allConfig := make(map[string]json.RawMessage, 0)
- err = json.Unmarshal(file, &allConfig)
- if err != nil {
- fmt.Println("err3", err)
- return
- }
- // 3. 循環(huán)map內(nèi)容
- for k, v := range allConfig {
- fmt.Println(k, string(v)) // 值為[]byte類型,將其轉(zhuǎn)為string
- }
- }
在2和2.1其實都是不同的實現(xiàn)方式。
yml配置
yml格式也是我們常見的文件配置格式,在Golang中,我們讀取該配置,主要用到了gopkg.in/yaml.v2包。同樣的,我們需要讀取配置文件->解析文件內(nèi)容。我們創(chuàng)建一個yml.yml文件,寫入下面的示例配置:
- host: 127.0.0.1
- user: root
- password: 123456
- port: 3306
- db: demo
需要注意的是,yml的配置項:與值中間是有一個空格的。
- // 定義一個struct來定義格式
- type DbYml struct {
- Host string `yaml:"host"`
- User string `yaml:"user"`
- Password string `yaml:"password"`
- Port string `yaml:"port"`
- Db string `yaml:"db"`
- }
- func GetYmlConfig() {
- // 1. 讀取配置文件內(nèi)容,將返回一個[]byte的內(nèi)容
- file, err := ioutil.ReadFile("./config/yml.yml")
- if err != nil {
- return
- }
- db := new(DbYml)
- // 2. 使用yaml包進行反序列化
- err = yaml.Unmarshal(file, db)
- if err != nil {
- return
- }
- fmt.Println(db.Host, db.User, db.Password, db.Port, db.Db)
- }
最終輸入結(jié)果:
- 127.0.0.1 root 123456 3306 demo
文本格式
讀取文件格式的內(nèi)容,就是按行讀取,然后針對每行的內(nèi)容進行解析。因為我們文本中的格式一般都是循序key=value的格式,因此我們只要讀取到改行的內(nèi)容,然后根據(jù)=進行分割即可。
首先我們創(chuàng)建一個文件.txt的文件內(nèi)容,大致內(nèi)容如下:
- host=127.0.0.1
- user=root
- password=123456
- port=3306
- db=demo
具體讀取配置代碼:
- func GetKeyValue() {
- allConfig := make(map[string]string)
- // 1. 讀取文件,得到文件句柄
- open, err := os.Open("./config/key.txt")
- if err != nil {
- fmt.Println("err1", err)
- return
- }
- // 2. 讀取文件內(nèi)容
- content := bufio.NewReader(open)
- for {
- // 3. 按行讀取文件內(nèi)容
- line, _, err := content.ReadLine()
- if err != nil {
- if err == io.EOF { // 去讀到結(jié)尾,就跳出循環(huán)讀取
- break
- }
- return
- }
- // 4. 處理每一行讀取到的文件內(nèi)容
- s := strings2.TrimSpace(string(line)) // 去掉左右空格
- index := strings2.Index(s, "=") // 因為配置是=,找到=的索引位置
- if index < 0 {
- continue
- }
- key := strings2.TrimSpace(s[:index]) // 截取=左側(cè)的值為key
- if len(key) == 0 {
- continue
- }
- value := strings2.TrimSpace(s[index+1:]) // 截取=右側(cè)的為value
- if len(value) == 0 {
- continue
- }
- allConfig[key] = value // 添加到map中,key為map的key,value為map的value
- }
- for k, v := range allConfig {
- fmt.Println(k, string(v))
- }
- defer open.Close() // 關(guān)閉關(guān)文件
- }
輸出的內(nèi)容大致如下:
- host 127.0.0.1
- user root
- password 123456
- port 3306
- db demo
tomal
使用toml格式的配置文件,主要用到了toml包進行解析出來。同樣的,首先我們是加載文件,將文件的路徑傳入到toml包中即可。首先我們創(chuàng)建一個toml文件的,定義如下內(nèi)容:
- [database]
- host="127.0.0.1"
- user="root"
- password="123456"
- port=[3306, 3307]
- db="demo"
下面是具體的解析代碼:
- import (
- "github.com/BurntSushi/toml"
- "path/filepath"
- )
- type DbToml struct {
- Db Database `toml:"database"`
- }
- type Database struct {
- Host string
- User string
- Password string
- Port []int32
- Db string
- }
- func GetToml() {
- // 1. 定義結(jié)構(gòu)體變量來接收解析的數(shù)據(jù)
- var config DbToml
- // 2. 獲取文件絕對路徑
- fileName, err := filepath.Abs("./config/toml.toml")
- if err != nil {
- fmt.Println("err1", err)
- return
- }
- // 3. 根據(jù)toml包的規(guī)則傳入文件路徑
- _, err1 := toml.DecodeFile(fileName, &config)
- if err1 != nil {
- fmt.Println("err2", err1)
- return
- }
- fmt.Println(config.Db.Host, config.Db.User, config.Db.Password, config.Db.Port[0], config.Db.Db)
- }
輸出結(jié)果如下:
- 127.0.0.1 root 123456 3306 demo