support config center

common/env.go

@@ -0,0 +1,6 @@
+package common
+
+// EnvKeyGloryEnv if is set to dev,then:
+// 1. choose config center with namespace dev
+// 2. choose config path like "config/glory_dev.yaml
+const EnvKeyGloryEnv = "GLORY_ENV" //

config/config.go

@@ -15,6 +15,8 @@ import (
 // EnvKeyGloryConfigPath is absolute/relate path to glory.yaml
 const EnvKeyGloryConfigPath = "GLORY_CONFIG_PATH" // default val is "config/glory.yaml"
 
+const EnvKeyGloryConfigCenterPath = "GLORY_CONFIG_CENTER_PATH" // default val is "config/config_center.yaml"
+
 // EnvKeyGloryEnv if is set to dev,then:
 // 1. choose config center with namespace dev
 // 2. choose config path like "config/glory_dev.yaml
@@ -54,6 +56,28 @@ func GetConfigPath() string {
 	return configPath
 }
 
+func GetConfigCenterPath() string {
+	configPath := ""
+	env := os.Getenv(EnvKeyGloryEnv)
+
+	configFilePath := DefaultConfigPath
+	if os.Getenv(EnvKeyGloryConfigCenterPath) != "" {
+		configFilePath = os.Getenv(EnvKeyGloryConfigCenterPath)
+	}
+	prefix := strings.Split(configFilePath, ".yaml")
+	// prefix == ["config/glory", ""]
+	if len(prefix) != 2 {
+		panic("Invalid config file path = " + configFilePath)
+	}
+	// get target env yaml file
+	if env != "" {
+		configPath = prefix[0] + "_" + env + ".yaml"
+	} else {
+		configPath = configFilePath
+	}
+	return configPath
+}
+
 func loadFileConfig() error {
 	conf := NewServerConfig()
 	configPath := GetConfigPath()
@@ -79,11 +103,8 @@ func loadFileConfig() error {
 
 // oadConfigCenterConfig load config center's config from file, default is config/config_center_config.yaml
 func loadConfigCenterConfig() bool {
-	configCenterConfigFilePath := DefaultConfigCenterConfigPath
-	if path := os.Getenv(EnvKeyGloryConfigCenterConfigPath); path != "" {
-		configCenterConfigFilePath = path
-	}
-	yamlFile, err := ioutil.ReadFile(configCenterConfigFilePath)
+	configCenterPath := GetConfigCenterPath()
+	yamlFile, err := ioutil.ReadFile(configCenterPath)
 	if err != nil {
 		fmt.Println("config center info: can't load config center config at " + DefaultConfigCenterConfigPath)
 		return false
@@ -95,6 +116,7 @@ func loadConfigCenterConfig() bool {
 		return false
 	}
 	configCenterConfig.checkAndFix()
+	// TODO: 扩展支持更多的配置中心
 	if configCenterConfig.Name != "nacos" && configCenterConfig.Name != "" {
 		fmt.Println("error: config center name ", configCenterConfig.Name, "not support")
 		return false

config_manager/config_center.go

@@ -0,0 +1,33 @@
+package configmanager
+
+import (
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+var (
+	configCenter              ConfigCenter
+	configCenterViperInstance *viper.Viper
+)
+
+func RegisterConfigBuilder(builder ConfigCenterBuilder) error {
+	if configCenter != nil {
+		return fmt.Errorf("only allow one config center builder")
+	}
+	// 初始化配置中心
+	var err error
+	configCenter, err = builder(configCenterViperInstance)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func GetConfigCenter() (ConfigCenter, error) {
+	if configCenter == nil {
+		return configCenter, fmt.Errorf("config center not register")
+	}
+	return configCenter, nil
+}

config_manager/const.go

@@ -0,0 +1,5 @@
+package configmanager
+
+const DefaultConfigCenterConfigPath = "config/config_center.yaml"
+
+const EnvKeyGloryConfigCenterPath = "GLORY_CONFIG_CENTER_PATH" // default val is "config/config_center.yaml"

config_manager/init.go

@@ -0,0 +1,47 @@
+package configmanager
+
+import (
+	"log"
+	"os"
+	"strings"
+
+	"github.com/glory-go/glory/common"
+	"github.com/spf13/viper"
+)
+
+func init() {
+	loadConfigCenterConfig()
+}
+
+func GetConfigCenterPath() string {
+	configPath := ""
+	env := os.Getenv(common.EnvKeyGloryEnv)
+
+	configFilePath := DefaultConfigCenterConfigPath
+	if os.Getenv(EnvKeyGloryConfigCenterPath) != "" {
+		configFilePath = os.Getenv(EnvKeyGloryConfigCenterPath)
+	}
+	prefix := strings.Split(configFilePath, ".yaml")
+	// prefix == ["config/glory", ""]
+	if len(prefix) != 2 {
+		panic("Invalid config file path = " + configFilePath)
+	}
+	// get target env yaml file
+	if env != "" {
+		configPath = prefix[0] + "_" + env + ".yaml"
+	} else {
+		configPath = configFilePath
+	}
+	return configPath
+}
+
+func loadConfigCenterConfig() {
+	path := GetConfigCenterPath()
+
+	viperInstance := viper.GetViper()
+	viperInstance.SetConfigFile(path)
+	if err := viper.ReadInConfig(); err != nil {
+		log.Panic("load config center meets error", err)
+	}
+	configCenterViperInstance = viperInstance
+}

config_manager/interface.go

@@ -0,0 +1,12 @@
+package configmanager
+
+import "github.com/spf13/viper"
+
+type ConfigCenter interface {
+	// LoadConfig 从配置中心获取配置
+	LoadConfig(key string) (string, error)
+	// SyncConfig 持续从配置中心获取配置，当配置发生变更时，将会修改value指向的值
+	SyncConfig(key string, value *string) error
+}
+
+type ConfigCenterBuilder func(conf *viper.Viper) (ConfigCenter, error)