go-zero/core/conf/config.go

package conf

import (
	"fmt"
	"log"
	"os"
	"path"
	"reflect"
	"strings"

	"github.com/zeromicro/go-zero/core/jsonx"
	"github.com/zeromicro/go-zero/core/mapping"
	"github.com/zeromicro/go-zero/internal/encoding"
)

var loaders = map[string]func([]byte, any) error{
	".json": LoadFromJsonBytes,
	".toml": LoadFromTomlBytes,
	".yaml": LoadFromYamlBytes,
	".yml":  LoadFromYamlBytes,
}

type fieldInfo struct {
	name     string
	children map[string]fieldInfo
}

// Load loads config into v from file, .json, .yaml and .yml are acceptable.
func Load(file string, v any, opts ...Option) error {
	content, err := os.ReadFile(file)
	if err != nil {
		return err
	}

	loader, ok := loaders[strings.ToLower(path.Ext(file))]
	if !ok {
		return fmt.Errorf("unrecognized file type: %s", file)
	}

	var opt options
	for _, o := range opts {
		o(&opt)
	}

	if opt.env {
		return loader([]byte(os.ExpandEnv(string(content))), v)
	}

	return loader(content, v)
}

// LoadConfig loads config into v from file, .json, .yaml and .yml are acceptable.
// Deprecated: use Load instead.
func LoadConfig(file string, v any, opts ...Option) error {
	return Load(file, v, opts...)
}

// LoadFromJsonBytes loads config into v from content json bytes.
func LoadFromJsonBytes(content []byte, v any) error {
	var m map[string]any
	if err := jsonx.Unmarshal(content, &m); err != nil {
		return err
	}

	finfo := buildFieldsInfo(reflect.TypeOf(v))
	lowerCaseKeyMap := toLowerCaseKeyMap(m, finfo)

	return mapping.UnmarshalJsonMap(lowerCaseKeyMap, v, mapping.WithCanonicalKeyFunc(toLowerCase))
}

// LoadConfigFromJsonBytes loads config into v from content json bytes.
// Deprecated: use LoadFromJsonBytes instead.
func LoadConfigFromJsonBytes(content []byte, v any) error {
	return LoadFromJsonBytes(content, v)
}

// LoadFromTomlBytes loads config into v from content toml bytes.
func LoadFromTomlBytes(content []byte, v any) error {
	b, err := encoding.TomlToJson(content)
	if err != nil {
		return err
	}

	return LoadFromJsonBytes(b, v)
}

// LoadFromYamlBytes loads config into v from content yaml bytes.
func LoadFromYamlBytes(content []byte, v any) error {
	b, err := encoding.YamlToJson(content)
	if err != nil {
		return err
	}

	return LoadFromJsonBytes(b, v)
}

// LoadConfigFromYamlBytes loads config into v from content yaml bytes.
// Deprecated: use LoadFromYamlBytes instead.
func LoadConfigFromYamlBytes(content []byte, v any) error {
	return LoadFromYamlBytes(content, v)
}

// MustLoad loads config into v from path, exits on error.
func MustLoad(path string, v any, opts ...Option) {
	if err := Load(path, v, opts...); err != nil {
		log.Fatalf("error: config file %s, %s", path, err.Error())
	}
}

func addOrMergeFields(info map[string]fieldInfo, key, name string, fields map[string]fieldInfo) {
	if prev, ok := info[key]; ok {
		// merge fields
		for k, v := range fields {
			prev.children[k] = v
		}
	} else {
		info[key] = fieldInfo{
			name:     name,
			children: fields,
		}
	}
}

func buildFieldsInfo(tp reflect.Type) map[string]fieldInfo {
	tp = mapping.Deref(tp)

	switch tp.Kind() {
	case reflect.Struct:
		return buildStructFieldsInfo(tp)
	case reflect.Array, reflect.Slice:
		return buildFieldsInfo(mapping.Deref(tp.Elem()))
	default:
		return nil
	}
}

func buildStructFieldsInfo(tp reflect.Type) map[string]fieldInfo {
	info := make(map[string]fieldInfo)

	for i := 0; i < tp.NumField(); i++ {
		field := tp.Field(i)
		name := field.Name
		lowerCaseName := toLowerCase(name)
		ft := mapping.Deref(field.Type)

		// flatten anonymous fields
		if field.Anonymous {
			if ft.Kind() == reflect.Struct {
				fields := buildFieldsInfo(ft)
				for k, v := range fields {
					addOrMergeFields(info, k, v.name, v.children)
				}
			} else {
				info[lowerCaseName] = fieldInfo{
					name:     name,
					children: make(map[string]fieldInfo),
				}
			}
			continue
		}

		var fields map[string]fieldInfo
		switch ft.Kind() {
		case reflect.Struct:
			fields = buildFieldsInfo(ft)
		case reflect.Array, reflect.Slice:
			fields = buildFieldsInfo(ft.Elem())
		case reflect.Map:
			fields = buildFieldsInfo(ft.Elem())
		}

		addOrMergeFields(info, lowerCaseName, name, fields)
	}

	return info
}

func toLowerCase(s string) string {
	return strings.ToLower(s)
}

func toLowerCaseInterface(v any, info map[string]fieldInfo) any {
	switch vv := v.(type) {
	case map[string]any:
		return toLowerCaseKeyMap(vv, info)
	case []any:
		var arr []any
		for _, vvv := range vv {
			arr = append(arr, toLowerCaseInterface(vvv, info))
		}
		return arr
	default:
		return v
	}
}

func toLowerCaseKeyMap(m map[string]any, info map[string]fieldInfo) map[string]any {
	res := make(map[string]any)

	for k, v := range m {
		ti, ok := info[k]
		if ok {
			res[k] = toLowerCaseInterface(v, ti.children)
			continue
		}

		lk := toLowerCase(k)
		if ti, ok = info[lk]; ok {
			res[lk] = toLowerCaseInterface(v, ti.children)
		} else {
			res[k] = v
		}
	}

	return res
}