micro-document/config/config.go

package config

import (
	"fmt"
	"os"

	"dubbo.apache.org/dubbo-go/v3/common/constant"
	"gopkg.in/yaml.v2"
)

var CF = new(Config)

type Config struct {
	DocDB struct {
		Host     string `yaml:"host"`
		User     string `yaml:"user"`
		Password string `yaml:"password"`
		DbName   string `yaml:"dbName"`
	} `yaml:"docDb"`
}

func ConfigInit() {
	var configFilePath string
	if os.Getenv("MODE_ENV") == "" {
		configFilePath = "../conf/config.yaml"
		os.Setenv(constant.ConfigFileEnvKey, "../conf/dubbogo.yaml")
	} else {
		configFilePath = "./conf/" + os.Getenv("MODE_ENV") + "/config.yaml"
		os.Setenv(constant.ConfigFileEnvKey, fmt.Sprintf("./conf/%s/%s", os.Getenv("MODE_ENV"), "dubbogo.yaml"))
	}
	fi, err := os.Open(configFilePath)
	if err != nil {
		panic(err)
	}
	defer fi.Close()
	if err := yaml.NewDecoder(fi).Decode(&CF); err != nil {
		panic(err)
	}
}