# yaml測試樣例
# null 或 NULL 為關(guān)鍵字，不能寫

# 表示 bool 真假的幾個值
result_true: 
  - y
  - Y
  - yes
  - Yes
  - YES
  - true
  - True
  - TRUE
  - on
  - On
  - ON

# 數(shù)組的另一種形式
result_false: [n, N, no, No, NO , false, False, FALSE , off, Off, OFF]

# 名稱
# 字符串
name: conf file

# 版本
# 如按浮點(diǎn)，2.0會轉(zhuǎn)換成2
# 如按字符串，保留原樣
version: 2.0

# 布爾類，轉(zhuǎn)換為1或0
need: true

# 時間
time: 2020-10-03T09:21:13

empty: nul

# 對象
# 加雙引號會轉(zhuǎn)義\n，即會換行
my:
  name: late \n lee
  name1: "late \n lee"
  age: 99
  
# 塊
text: |
  hello
  world!

# 數(shù)組
fruit:
  - apple
  - apple1
  - apple2
  - apple3
  - apple4
  - apple5

# 多級數(shù)組
multi:
  sta:
    - 110 210 ddd 99
    - 133 135 1 2 1588 1509
    - 310-410
    - 333-444

# 多層級
loginfo:
  log:
    dir: log

# 多級對象
mymap:
  dir: "mymap"
  map_data:
    - name: "在線"
      attri: "在線電子"
      url: "http://abc.com"
    - name: "離線"
      attri: "離線電子"
      url: "http://ccc.com"
    # more

package test

import (
	"fmt"
	"os"
	"testing"

	"github.com/spf13/viper"
)

var (
	cfgFile string
)

type mapUrl_t struct {
	Name  string `json:"name"`
	Attri string `json:"attri"`
	Url   string `json:"url"`
}

func TestYaml(t *testing.T) {
	fmt.Println("test of yaml...")

	// 設(shè)置配置文件的2種方式
	if cfgFile != "" {
		// Use config file from the flag.
		viper.SetConfigFile(cfgFile)
	} else {
		viper.AddConfigPath("./")
		viper.SetConfigName("config")
		viper.SetConfigType("yaml")
	}

	viper.AutomaticEnv() // read in environment variables that match

	// 讀取
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println("'config.yaml' file read error:", err)
		os.Exit(0)
	}

	name := viper.GetString("name") // 讀取 字符串
	version := viper.GetString("version")

	need := viper.GetBool("need") // 讀取 布爾
	theTime := viper.GetString("time")
	empty := viper.GetString("empty")
	text := viper.GetString("text")

	fmt.Printf("need: %v name: %v\nversion: %v \ntime: %v \nempty: %s \ntext: %v\n", need, name, version, theTime, empty, text)

	// 多級讀取
	name = viper.GetString("my.name")
	name1 := viper.GetString("my.name1")
	age := viper.GetInt("my.age")
	fmt.Printf("name: %v, name1: %v age: %v \n", name, name1, age)

	// 字符串?dāng)?shù)組
	newSta := viper.GetStringSlice("multi.sta")
	for idx, value := range newSta {
		fmt.Printf("sta[%d]: %v\n", idx, value)
	}

	fruit := viper.GetStringSlice("fruit")
	fmt.Printf("fruit: %v\n", fruit)

	// 讀取不存在的字段，字符串為空，數(shù)值為0
	bad := viper.GetString("bad")
	bad1 := viper.GetInt("my.bad")
	fmt.Printf("bad: [%v] bad1: [%v]\n", bad, bad1)

	// 按數(shù)值、字符串讀取on、off等值
	result := viper.GetIntSlice("result_true")
	fmt.Printf("result true: [%v]\n", result)
	result1 := viper.GetStringSlice("result_true")
	fmt.Printf("result1 true: [%v]\n", result1)

	result = viper.GetIntSlice("result_false")
	fmt.Printf("result false: [%v]\n", result)
	result1 = viper.GetStringSlice("result_false")
	fmt.Printf("result1 false: [%v]\n", result1)

	logdir := viper.GetString("loginfo.log.dir")
	fmt.Printf("logdir: %v\n", logdir)

	// 多級對象
	// tmpMap := make([]mapUrl_t, 0, 20)
	var tmpMap []mapUrl_t

	viper.UnmarshalKey("mymap.map_data", &tmpMap)

	for _, item := range tmpMap {
		fmt.Printf("name: %v url: %v\n", item.Name, item.Url)
	}
}

go test -v -run TestYaml

test of yaml...
need: true name: conf file
version: 2
time: 2020-10-03T09:21:13
empty: nul
text: hello
world!

name: late \n lee, name1: late
 lee age: 99
sta[0]: 110 210 ddd 99
sta[1]: 133 135 1 2 1588 1509
sta[2]: 310-410
sta[3]: 333-444
fruit: [apple apple1 apple2 apple3 apple4 apple5]
bad: [] bad1: [0]
result true: [[1 1 1 1 1 1 1 1 1 1 1]]
result1 true: [[true true true true true true true true true true true]]
result false: [[0 0 0 0 0 0 0 0 0 0 0]]
result1 false: [[false false false false false false false false false false false]]
logdir: log
name: 在線 url: http://abc.com
name: 離線 url: http://ccc.com