feat: custom func

config/custom_config.go

@@ -27,14 +27,18 @@ import (
 
 type CustomConfig struct {
 	ConfigMap map[string]interface{} `yaml:"config-map" json:"config-map,omitempty" property:"config-map"`
+	Functions []CustomFunc
 }
 
 func (*CustomConfig) Prefix() string {
 	return constant.CustomConfigPrefix
 }
 
 func (c *CustomConfig) Init() error {
-	return c.check()
+	if err := c.check(); err != nil {
+		return err
+	}
+	return c.runCustomFunc()
 }
 
 func (c *CustomConfig) check() error {
@@ -51,6 +55,15 @@ func (c *CustomConfig) GetDefineValue(key string, default_value interface{}) int
 	return default_value
 }
 
+func (c *CustomConfig) runCustomFunc() error {
+	for _, f := range c.Functions {
+		if err := f(rootConfig); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 func GetDefineValue(key string, default_value interface{}) interface{} {
 	rt := GetRootConfig()
 	if rt.Custom == nil {
@@ -75,6 +88,14 @@ func (ccb *CustomConfigBuilder) SetDefineConfig(key string, val interface{}) *Cu
 	return ccb
 }
 
+// CustomFunc must not block
+type CustomFunc func(config *RootConfig) error
+
+func (ccb *CustomConfigBuilder) AddCustomFunc(fc CustomFunc) *CustomConfigBuilder {
+	ccb.customConfig.Functions = append(ccb.customConfig.Functions, fc)
+	return ccb
+}
+
 func (ccb *CustomConfigBuilder) Build() *CustomConfig {
 	return ccb.customConfig
 }

config/custom_config_test.go

@@ -86,3 +86,22 @@ func TestConfigUtils(t *testing.T) {
 	element := removeDuplicateElement([]string{"nacos", "nacos"})
 	assert.Equal(t, len(element), 1)
 }
+
+func TestCustomFunc(t *testing.T) {
+	t.Run("custom func", func(t *testing.T) {
+		customConfigBuilder := NewCustomConfigBuilder()
+		customConfigBuilder.SetDefineConfig("password", "encrypt")
+		customConfigBuilder.AddCustomFunc(func(rootConfig *RootConfig) error {
+			//decrypt the password
+			rootConfig.Custom.ConfigMap["password"] = "decrypt"
+			return nil
+		})
+		customConfig := customConfigBuilder.Build()
+		rootConfig.Custom = customConfig
+		err := customConfig.Init()
+		if err != nil {
+			return
+		}
+		assert.Equal(t, "decrypt", customConfig.GetDefineValue("password", ""))
+	})
+}