Go--解析yaml文件

yaml 文件是目前最常用的配置文件，使用go语言编写代码和工具时，也会用到yaml文件，将服务配置及中间件等信息定义到yaml文件中，后续可根据实际场景来选用。

//先下载外部包
go get -u gopkg.in/yaml.v2

 

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结构体
        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)

    //连接数据库
    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)     //最大连接超时时间，应小于数据库本身的链接超时时间
    conn.SetMaxIdleConns(10)                  //最大闲置连接数（并发时可以同时获取的连接,也是用完后放回池里面的互用的连接, 从而提升性能）
    conn.SetMaxOpenConns(100)                 //最大打开连接数，0为不限制

    //记得结束后关闭连接
    defer conn.Close()
}

 

3、输出结果

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、补充

4.1 已弃用 'ReadFile'

从 Go 语言 1.16 开始，ioutil.ReadAll、ioutil.ReadFile 和 ioutil.ReadDir 被弃用，因为 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

 4.2 yaml文件里的变量命名规则

yaml文件里的变量需全用小写字母。使用大写字母、中划线、下划线等，如 CK: 127.1.1.1 mySQL : xxx Myql : xxx my_sql : xxx 会在解读yaml配置文件时出问题，解读出来是空值，且不会报错