快捷導(dǎo)航

Golang解析yaml文件的方法小結(jié)

更新時(shí)間：2024年11月01日 09:19:11 作者：心恩惠動(dòng)

Go 語言沒有內(nèi)置解析 yaml 文件的功能,實(shí)現(xiàn) yaml 的解析可以使用第三方庫,下面我們就來看看如何使用opkg.in/yaml.v2 和 gopkg.in/yaml.v3實(shí)現(xiàn)解析yaml吧

1、定義yaml配置文件

conf/test.yaml

ck:
  host: 11.11.11.11
  port: 9001
  database: db2
  username: test2
  password: 123ck456
user: test
password: 123yy456
host: 10.10.10.10
port: 3306
dbname: db1

2、主程序

package main

import (
    "database/sql"
    "fmt"
    "gopkg.in/yaml.v3"
    "io/ioutil"
    "time"
)

type Conn struct {
    Ck struct {
        Host     string `yaml:"host"`
        Port     string `yaml:"port"`
        Database string `yaml:"database"`
        Username string `yaml:"username"`
        Password string `yaml:"password"`
    }
}
type Mysql struct {
    User     string `yaml:"user"`
    Password string `yaml:"password"`
    Host     string `yaml:"host"`
    Port     string `yaml:"port"`
    DBname   string `yaml:"dbname"`
}

const cfgFile = "conf/test.yaml"

func main() {
    data, err := ioutil.ReadFile(cfgFile)　　//讀取配置文件的路徑
    if err != nil {
        fmt.Printf("err: %v\n", err)
        return
    }

    var conf Conn
    if err := yaml.Unmarshal(data, &conf); err != nil {　　//將yaml文件中的信息反序列化給conf結(jié)構(gòu)體
        fmt.Printf("err: %v\n", err)
        return
    }
    yml, _ := yaml.Marshal(conf)　　//序列化為yaml文件
    fmt.Printf("conf: %#v\n", conf)
    fmt.Printf("%s\n", yml)

    var confs Mysql
    if err := yaml.Unmarshal(data, &confs); err != nil {
        fmt.Printf("err: %v\n", err)
        return
    }
    yaml, _ := yaml.Marshal(confs)
    fmt.Printf("conf: %#v\n", confs)
    fmt.Printf("%s\n", yaml)

    //連接數(shù)據(jù)庫
    user := confs.User
    password := confs.Password
    host := confs.Host
    port := confs.Port
    database := confs.DBname
    dataSourceName := user + ":" + password + "@tcp(" + host + ":" + port + ")/" + database + "?charset=utf8&parseTime=True"
    conn, err := sql.Open("mysql", dataSourceName)
    if err != nil {
        return
    }
    conn.SetConnMaxLifetime(time.Minute * 30)     //最大連接超時(shí)時(shí)間，應(yīng)小于數(shù)據(jù)庫本身的鏈接超時(shí)時(shí)間
    conn.SetMaxIdleConns(10)                  //最大閑置連接數(shù)（并發(fā)時(shí)可以同時(shí)獲取的連接,也是用完后放回池里面的互用的連接, 從而提升性能）
    conn.SetMaxOpenConns(100)                 //最大打開連接數(shù)，0為不限制

    //記得結(jié)束后關(guān)閉連接
    defer conn.Close()
}

3、輸出結(jié)果

conf: main.Conn{Ck:struct { Host string "yaml:\"host\""; Port string "yaml:\"por
t\""; Database string "yaml:\"database\""; Username string "yaml:\"username\"";
Password string "yaml:\"password\"" }{Host:"11.11.11.11", Port:"9001", Database:
"db2", Username:"test2", Password:"123ck456"}}
ck:
    host: 11.11.11.11
    port: "9001"
    database: db2
    username: test2
    password: 123ck456

conf: main.Mysql{User:"test", Password:"123yy456", Host:"10.10.10.10", Port:"330
6", DBname:"db1"}
user: test
password: 123yy456
host: 10.10.10.10
port: "3306"
dbname: db1

4、補(bǔ)充

4.1 已棄用 'ReadFile'

從 Go 語言 1.16 開始，ioutil.ReadAll、ioutil.ReadFile 和 ioutil.ReadDir 被棄用，因?yàn)?io/ioutil 包被棄用。

解決方法如下，使用 io 或 os 包中相同的方法替換，即修改自己按如下參照修改包名即可

ioutil.ReadAll -> io.ReadAll
    ioutil.ReadFile -> os.ReadFile
    ioutil.ReadDir -> os.ReadDir
    // others
    ioutil.NopCloser -> io.NopCloser
    ioutil.ReadDir -> os.ReadDir
    ioutil.TempDir -> os.MkdirTemp
    ioutil.TempFile -> os.CreateTemp
    ioutil.WriteFile -> os.WriteFile