[agent-go] 调整代码结构

agent-go/register/NacosInitalization.go

@@ -0,0 +1,171 @@
+package register
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/nacos-group/nacos-sdk-go/v2/clients"
+	"github.com/nacos-group/nacos-sdk-go/v2/clients/config_client"
+	"github.com/nacos-group/nacos-sdk-go/v2/common/constant"
+	"github.com/nacos-group/nacos-sdk-go/v2/vo"
+	"github.com/spf13/viper"
+	"go.uber.org/zap"
+	"strconv"
+	"strings"
+)
+
+var group = ""
+
+func InitNacos(configFileName string) *viper.Viper {
+
+	v := parseAgentConfigFile(configFileName, nil)
+	group = v.GetString("spring.cloud.nacos.config.group")
+
+	// build the nacos connection
+	configClient := startNacosConnection(v)
+
+	// get all needed nacos config and merge
+	allNacosConfig := getAllNacosConfig(v, group, configClient)
+
+	for _, nacosConfigContent := range allNacosConfig {
+		log.Debug(fmt.Sprintf("nacos config conetent is %s", nacosConfigContent))
+
+		parseNacosConfigContend(nacosConfigContent, v)
+	}
+
+	log.Info(fmt.Sprintf("%s config read result are %v", configFileName, v.AllSettings()))
+
+	return v
+}
+
+func parseAgentConfigFile(configFileName string, v *viper.Viper) *viper.Viper {
+
+	// 使用Viper框架读取
+	if v == nil {
+		v = viper.New()
+	}
+
+	// 设置配置文件路径和名称
+	v.SetConfigName(configFileName)
+	v.AddConfigPath(".")
+	v.SetConfigType("yaml")
+
+	// 读取默认的总配置文件
+	err := v.ReadInConfig()
+	if err != nil {
+		panic(fmt.Errorf("fatal error config file: %s", err))
+	}
+
+	return v
+}
+
+func parseNacosConfigContend(configContent string, v *viper.Viper) *viper.Viper {
+
+	v.SetConfigType("yaml")
+
+	// use merge
+
+	err := v.MergeConfig(bytes.NewBuffer([]byte(configContent)))
+	if err != nil {
+		log.Error("nacos config contend read error !", zap.Error(err))
+	}
+
+	return v
+}
+func startNacosConnection(v *viper.Viper) config_client.IConfigClient {
+
+	serverAddr := v.GetString("spring.cloud.nacos.config.server-addr")
+
+	clientConfig := constant.ClientConfig{
+		//Endpoint: serverAddr,
+		NamespaceId:          "",
+		TimeoutMs:            v.GetUint64("spring.cloud.nacos.config.timeout"),
+		NotLoadCacheAtStart:  true,
+		AppendToStdout:       true,
+		UpdateCacheWhenEmpty: true,
+		//LogDir:              "/tmp/nacos/log",
+		//CacheDir:            "/tmp/nacos/cache",
+		Username: "nacos",
+		Password: "Superwmm.23",
+	}
+
+	split := strings.Split(serverAddr, ":")
+	if len(split) != 2 {
+		log.Error("nacos server addr error!")
+	}
+
+	port, _ := strconv.ParseUint(split[1], 10, 64)
+	serverConfigs := []constant.ServerConfig{
+		{
+			IpAddr:   split[0],
+			Port:     port,
+			GrpcPort: port + 1000,
+		},
+	}
+
+	// Another way of create config client for dynamic configuration (recommend)
+	configClient, err := clients.NewConfigClient(
+		vo.NacosClientParam{
+			ClientConfig:  &clientConfig,
+			ServerConfigs: serverConfigs,
+		},
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	return configClient
+}
+
+func getAllNacosConfig(v *viper.Viper, group string, configClient config_client.IConfigClient) []string {
+
+	result := make([]string, 0)
+
+	// main nacos configs
+	mainNacosConfigFileName := v.GetString("spring.application.name") + "-" + v.GetString("spring.profiles.active") + "." + v.GetString("spring.cloud.nacos.config.file-extension")
+
+	log.Debug(fmt.Sprintf("main nacos config file name is %s", mainNacosConfigFileName))
+	configContent := getConfig(mainNacosConfigFileName, group, configClient)
+	result = append(result, configContent)
+
+	// additional nacos config
+	additionalNacosConfig := v.Get("spring.cloud.nacos.config.extension-configs")
+	// 增加断言，判定map的类型
+	m, ok := additionalNacosConfig.([]interface{})
+	if !ok {
+		fmt.Println("additionalNacosConfig is not a slice")
+		return nil
+	}
+
+	for _, addConfigMap := range m {
+
+		realMap, _ := addConfigMap.(map[string]interface{})
+
+		// 拿到配置的Key
+		dataId := realMap["data-id"].(string)
+		group := realMap["group"].(string)
+
+		// 查询
+		config := getConfig(dataId, group, configClient)
+		result = append(result, config)
+	}
+
+	return result
+}
+
+// getConfig 从Nacos中获取相应的
+func getConfig(dataId string, group string, configClient config_client.IConfigClient) string {
+
+	log.Debug(fmt.Sprintf("nacos config get method dataID is %s, group is %s", dataId, group))
+
+	content, err := configClient.GetConfig(vo.ConfigParam{
+		DataId: dataId,
+		Group:  group,
+	})
+	if err != nil {
+		log.Error("nacos config get error !", zap.Error(err))
+	}
+
+	log.Debug(fmt.Sprintf("dataId %s , group %s, nacos config content is %s", dataId, group, content))
+
+	return content
+}