env_test.go

package goenv

import (
	"fmt"
	"io"
	"log"
	"os"
	"strings"
	"testing"

	"gopkg.in/yaml.v3"
)

type student struct {
	Name   string `env:""`
	Age    int
	gender bool  `env:""` // 小写将被忽略
	Addr   *addr // struct, 将跟踪到下层
	addr2  addr
}

func (s *student) SetDefaults() {
	s.Name = "zhangsan"
	s.Age = 100
}

type addr struct {
	Home   string `env:"home"`
	School string `env:"school"`
}

func (a *addr) Init() {
	a.Home = "changshaaaaaaaaa"
}

var (
	_ = os.Setenv("APP__Stud01_Name", "zhugeliang")
	_ = os.Setenv("APP__Stud01_Age", "500")
	_ = os.Setenv("APP__Stud01_Gender", "true")
	_ = os.Setenv("APP__Stud01__addr_home", "addr: sichuan")
	_ = os.Setenv("APP__Stud01__addr2_home", "addr2: sichuan")
	_ = os.Setenv("APP__Addr_home", "zhongguo,sichuan,chengdu")
	_ = os.Setenv("APP__Addr_Home", "APP__Addr_Home")

	APPNAME     = "APP"
	CONFIG_FILE = `config.yml`
)

func Test_marshal(t *testing.T) {
	stu := &student{
		Name:   "zhangsan2",
		Age:    20,
		gender: false,
		Addr: &addr{
			Home:   "sichuan",
			School: "chengdu",
		},
		addr2: addr{
			Home:   "sichuan",
			School: "chengdu",
		},
	}
	config := &struct {
		Stud01 *student
		Addr   addr
	}{
		Stud01: stu,
		Addr:   addr{},
	}

	/* marshal */
	b, _ := Marshal(config, "APP")
	_ = output(b, os.Stdout)
	_ = os.WriteFile(CONFIG_FILE, b, 0644)
}

func Test_UnmarshalEnv(t *testing.T) {
	stu := student{}
	config := &struct {
		Stud01 *student
		Addr   addr
	}{
		Stud01: &stu,
		Addr:   addr{},
	}

	/* unmarshal */
	err := UnmarshalEnv(config, APPNAME)
	if err != nil {
		log.Fatal(err)
	}

	b, err := yaml.Marshal(config)
	if err != nil {
		log.Fatal(err)
	}
	_ = output(b, os.Stdout)
}

func Test_UnmarshalFile(t *testing.T) {
	stu := student{}
	config := &struct {
		Stud01 *student
		Addr   addr
	}{
		Stud01: &stu,
		Addr:   addr{},
	}

	/* unmarshal */
	err := UnmarshalFile(config, APPNAME, CONFIG_FILE)
	if err != nil {
		log.Fatal(err)
	}

	b, err := yaml.Marshal(config)
	if err != nil {
		log.Fatal(err)
	}
	_ = output(b, os.Stdout)
}

func Test_CallMethod(t *testing.T) {
	stu := &student{
		Addr: &addr{},
	}

	config := &struct {
		Student *student
	}{
		Student: stu,
	}

	err := CallSetDefaults(config)
	if err != nil {
		panic(err)
	}

	// fmt.Println(config.Student)
	// fmt.Println(config.Student.Addr)

	b, err := Marshal(config, "APP")
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s\n", b)
}

func output(data []byte, w io.Writer) error {
	_, err := w.Write(data)
	return err
}

func Test_ReadEnv(t *testing.T) {

	err := os.Setenv("MY_VARS", "key1=value1, key2=value2")
	_ = os.Setenv("Emtpy_VARS", "")
	if err != nil {
		log.Fatal(err)
	}

	envs := os.Environ()

	for _, pair := range envs {
		kv := strings.Split(pair, "=")
		fmt.Printf("key =>%s , value ->%s\n", kv[0], strings.Join(kv[1:], "="))
	}

}

type Manager struct {
	ClassName  string `env:""`
	Filesystem *FileSystem
	// filesystem *FileSystem // 私有字段， 不会出现在配置中
	FFSys FileSystem
}

type FileSystem struct {
	DirPath string `env:""`
}

func (fs *FileSystem) SetDefaults() {
	if fs.DirPath == "" {
		fs.DirPath = "set-default"
	}
}

func Test_ConfP(t *testing.T) {

	config := &struct {
		Manager *Manager
	}{
		Manager: &Manager{},
	}

	data, err := Marshal(config, "AppName")
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s\n", data)
}

// Test 3
type MysqlServer struct {
	ListenAddr string `env:"listenAddr"`
	Auth       string `env:"auth"`
	DBName     string `env:"dbName"`
}

func (my *MysqlServer) SetDefaults() {
	if my.ListenAddr == "" {
		my.ListenAddr = "localhost:3306"
	}
}

type RedisServer struct {
	DSN string `env:"dsn"`
}

func (r *RedisServer) SetDefaults() {
	if r.DSN == "" {
		r.DSN = "redis://:Password@localhost:6379/0"
	}
}

func Test_ConfP_Server(t *testing.T) {

	config := &struct {
		MysqlServer *MysqlServer
		RedisServer *RedisServer
	}{
		MysqlServer: &MysqlServer{},
		RedisServer: &RedisServer{},
	}

	// 设置默认值
	err := CallSetDefaults(config)
	if err != nil {
		panic(err)
	}

	// 序列化配置
	data, err := Marshal(config, "AppName")
	if err != nil {
		panic(err)
	}
	fmt.Printf("%s\n", data)
	_ = os.WriteFile("default.yml", data, os.ModePerm)

	err = UnmarshalFile(config, "AppName", "config.yml")
	if err != nil {
		panic(err)
	}

	fmt.Println("my_auth =>", config.MysqlServer.Auth)
	fmt.Println("redis_dsn =>", config.RedisServer.DSN)

}