80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
|
/*
|
|||
|
|
* @Author : huangzj
|
|||
|
|
* @Time : 2020/12/16 11:56
|
|||
|
|
* @Description：
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
package yaml
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type MySQLConfig struct {
|
|||
|
|
User string `yaml:"user"`
|
|||
|
|
Password string `yaml:"password"`
|
|||
|
|
Host string `yaml:"host"`
|
|||
|
|
Port int `yaml:"port"`
|
|||
|
|
DbName string `yaml:"dbname"`
|
|||
|
|
}
|
|||
|
|
type RedisConfig struct {
|
|||
|
|
Host string `yaml:"host"`
|
|||
|
|
Port int `yaml:"port"`
|
|||
|
|
Auth string `yaml:"-"` //标签表示不显示
|
|||
|
|
}
|
|||
|
|
type NginxProxyConfig struct {
|
|||
|
|
Counter int `yaml:"counter"`
|
|||
|
|
NginxList []string `yaml:"nginx_list"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ServiceYaml struct {
|
|||
|
|
MySQL MySQLConfig `yaml:"mysql"`
|
|||
|
|
Redis RedisConfig `yaml:"redis"`
|
|||
|
|
NginxProxy NginxProxyConfig `yaml:"nginx_proxy"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestYamlToStructByFile(t *testing.T) {
|
|||
|
|
config := new(ServiceYaml)
|
|||
|
|
_ = GetYamlToStructByFile("config.yaml", &config)
|
|||
|
|
|
|||
|
|
fmt.Printf("MySQL host : %s, port : %d, user : %s, password : %s, db_name : %s \n",
|
|||
|
|
config.MySQL.Host, config.MySQL.Port, config.MySQL.User, config.MySQL.Password, config.MySQL.DbName)
|
|||
|
|
|
|||
|
|
fmt.Printf("Redis host : %s, port %d, auth : %s\n",
|
|||
|
|
config.Redis.Host, config.Redis.Port, config.Redis.Auth)
|
|||
|
|
|
|||
|
|
fmt.Printf("Vip Counter: %d, Vip List : %v\n", config.NginxProxy.Counter, config.NginxProxy.NginxList)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestYamlToStructByString(t *testing.T) {
|
|||
|
|
config := new(ServiceYaml)
|
|||
|
|
fileByte, _ := ioutil.ReadFile("config.yaml")
|
|||
|
|
_ = GetYamlToStructByString(string(fileByte), &config)
|
|||
|
|
|
|||
|
|
fmt.Printf("MySQL host : %s, port : %d, user : %s, password : %s, db_name : %s \n",
|
|||
|
|
config.MySQL.Host, config.MySQL.Port, config.MySQL.User, config.MySQL.Password, config.MySQL.DbName)
|
|||
|
|
|
|||
|
|
fmt.Printf("Redis host : %s, port %d, auth : %s\n",
|
|||
|
|
config.Redis.Host, config.Redis.Port, config.Redis.Auth)
|
|||
|
|
|
|||
|
|
fmt.Printf("Vip Counter: %d, Vip List : %v\n", config.NginxProxy.Counter, config.NginxProxy.NginxList)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestYamlToMap(t *testing.T) {
|
|||
|
|
fileByte, _ := ioutil.ReadFile("config.yaml")
|
|||
|
|
_, m := GetYamlToMap(string(fileByte))
|
|||
|
|
for key, value := range m {
|
|||
|
|
fmt.Println(fmt.Sprintf("key:%v ,value: %v", key, value))
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestStructToYaml(t *testing.T) {
|
|||
|
|
config := new(ServiceYaml)
|
|||
|
|
fileByte, _ := ioutil.ReadFile("config.yaml")
|
|||
|
|
_ = GetYamlToStructByString(string(fileByte), &config)
|
|||
|
|
_, result := GetStructToYaml(config)
|
|||
|
|
fmt.Println(result)
|
|||
|
|
}
|