add: parameter config support alarm forward config

restagent/config/config.go

@@ -1,14 +1,18 @@
 package config
 
 import (
+	"bufio"
 	"fmt"
 	"os"
+	"reflect"
 	"strings"
 
 	"be.ems/lib/global"
 	"be.ems/lib/log"
 
 	"gopkg.in/yaml.v3"
+	//"github.com/go-yaml-comment/yaml"
+	//"github.com/goccy/go-yaml"
 )
 
 // Yaml struct of config
@@ -147,29 +151,34 @@ type DbConfig struct {
 }
 
 type AlarmConfig struct {
-	SplitEventAlarm bool   `yaml:"splitEventAlarm"`
-	ForwardAlarm    bool   `yaml:"forwardAlarm"`
-	SMProxy         string `yaml:"smProxy"`
-	Email           struct {
-		Smtp          string `yaml:"smtp"`
-		Port          uint16 `yaml:"port"`
-		User          string `yaml:"user"`
-		Password      string `yaml:"password"`
-		TlsSkipVerify bool   `yaml:"tlsSkipVerify"`
-	} `yaml:"email"`
+	SplitEventAlarm bool `yaml:"splitEventAlarm"`
+	//ForwardAlarm    bool `yaml:"forwardAlarm"`
+
+	EmailForward struct {
+		Enable        bool   `yaml:"enable" json:"enable"`
+		EmailList     string `yaml:"emailList" json:"emailList"`
+		SMTP          string `yaml:"smtp" json:"smtp"`
+		Port          uint16 `yaml:"port" json:"port"`
+		User          string `yaml:"user" json:"user"`
+		Password      string `yaml:"password" json:"password"`
+		TLSSkipVerify bool   `yaml:"tlsSkipVerify" json:"tlsSkipVerify"`
+	} `yaml:"alarmEmailForward"`
+	SMSCForward struct {
+		Enable     bool   `yaml:"enable" json:"enable"`
+		MobileList string `yaml:"mobileList" json:"mobileList"`
+		SMSCAddr   string `yaml:"smscAddr" json:"smscAddr"`
+		SystemID   string `yaml:"systemID" json:"systemID"`
+		Password   string `yaml:"password" json:"password"`
+		SystemType string `yaml:"systemType" json:"systemType"`
+	} `yaml:"alarmSMSForward"`
 	SMS struct {
 		ApiURL          string `yaml:"apiURL"`
 		AccessKeyID     string `yaml:"AccessKeyID"`
 		AccessKeySecret string `yaml:"accessKeySecret"`
 		SignName        string `yaml:"signName"`
 		TemplateCode    string `yaml:"templateCode"`
-	} `yaml:"sms"`
-	SMSC struct {
-		Addr       string `yaml:"addr"`
-		SystemID   string `yaml:"systemID"`
-		Password   string `yaml:"password"`
-		SystemType string `yaml:"systemType"`
-	} `yaml:"smsc"`
+	} `yaml:"smsForward"`
+	SMProxy string `yaml:"smProxy"`
 }
 
 type MMLParam struct {
@@ -218,6 +227,16 @@ type TestDataMap struct {
 
 var yamlConfig YamlConfig = NewYamlConfig()
 
+type YamlConfigFile struct {
+	FilePath     string     `json:"filePath"`
+	ConfigLines  YamlConfig `json:"configLines"`
+	OrignalLines []string   `json:"orignalLines"`
+}
+
+var YamlConfigInfo YamlConfigFile = YamlConfigFile{
+	ConfigLines: NewYamlConfig(),
+}
+
 // set default value for yaml config
 func NewYamlConfig() YamlConfig {
 	return YamlConfig{
@@ -237,6 +256,8 @@ func NewYamlConfig() YamlConfig {
 }
 
 func ReadConfig(configFile string) {
+	YamlConfigInfo.FilePath = configFile
+
 	yamlFile, err := os.ReadFile(configFile)
 	if err != nil {
 		fmt.Println("Read yaml config file error:", err)
@@ -244,25 +265,85 @@ func ReadConfig(configFile string) {
 	}
 	//	fmt.Println("yamlfile:", string(yamlFile))
 
-	err = yaml.Unmarshal(yamlFile, &yamlConfig)
+	err = yaml.Unmarshal(yamlFile, &YamlConfigInfo.ConfigLines)
 	if err != nil {
 		fmt.Println("Unmarshal error:", err)
 		os.Exit(3)
 	}
+	yamlConfig = YamlConfigInfo.ConfigLines
+
+	ReadOriginalConfig(configFile)
 }
 
-func WriteYamlConfig(newConfigData YamlConfig, configFile string) {
+func ReadOriginalConfig(configFile string) {
+	// 读取原始YAML文件
+	inputFile, err := os.Open(configFile)
+	if err != nil {
+		fmt.Println("failed to open:", err)
+		os.Exit(3)
+	}
+	defer inputFile.Close()
+
+	scanner := bufio.NewScanner(inputFile)
+	for scanner.Scan() {
+		YamlConfigInfo.OrignalLines = append(YamlConfigInfo.OrignalLines, scanner.Text())
+	}
+
+	if err := scanner.Err(); err != nil {
+		fmt.Println("failed to scanner:", err)
+		os.Exit(3)
+	}
+}
+
+func WriteOrignalConfig(configFile string, paramName string, paramData map[string]any) error {
+	lines := YamlConfigInfo.OrignalLines
+	for i, line := range lines {
+		if strings.Contains(line, paramName) {
+			for k, v := range paramData {
+				// find the first line nearby the paramName
+				for j := i + 1; j < len(lines); j++ {
+					if strings.Contains(lines[j], k+":") {
+						index := strings.Index(lines[j], k)
+						lines[j] = lines[j][:index] + fmt.Sprintf("%s: %v", k, v)
+						break
+					}
+				}
+			}
+			break
+		}
+	}
+
+	// write back to yaml file
+	outputFile, err := os.Create(configFile)
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	defer outputFile.Close()
+
+	writer := bufio.NewWriter(outputFile)
+	for _, line := range YamlConfigInfo.OrignalLines {
+		writer.WriteString(line + "\n")
+	}
+	writer.Flush()
+	return nil
+}
+
+func WriteYamlConfig(newConfigData YamlConfig, configFile string) error {
 	// 将配置转换回YAML数据
 	newYamlData, err := yaml.Marshal(&newConfigData)
 	if err != nil {
 		log.Errorf("Failed to marshal YAML: %v", err)
+		return err
 	}
 
 	// 将新的YAML数据写入文件
 	err = os.WriteFile(configFile, newYamlData, 0644)
 	if err != nil {
 		log.Errorf("Failed to write YAML file: %v", err)
+		return err
 	}
+	return nil
 }
 
 var mapYaml map[string]interface{}
@@ -284,8 +365,28 @@ func ReadParamConfig(fileName string) *map[string]interface{} {
 	return &mapYaml
 }
 
+func UpdateStructFromMap(s any, updates map[string]any) {
+	v := reflect.ValueOf(s).Elem()
+	t := v.Type()
+
+	for key, value := range updates {
+		for i := 0; i < t.NumField(); i++ {
+			field := t.Field(i)
+			if field.Tag.Get("json") == key {
+				structField := v.FieldByName(field.Name)
+				if structField.IsValid() && structField.CanSet() {
+					if structField.Type() == reflect.TypeOf(value) {
+						structField.Set(reflect.ValueOf(value))
+					}
+				}
+				break
+			}
+		}
+	}
+}
+
 func GetYamlConfig() *YamlConfig {
-	return &yamlConfig
+	return &YamlConfigInfo.ConfigLines
 }
 
 func GetAuthFromConfig() interface{} {

restagent/etc/restconf.yaml

@@ -51,13 +51,13 @@ redis:
     # OMC系统使用库
     default:
       port: 6379 # Redis port
-      host: "192.168.8.58" # Redis host
+      host: "127.0.0.1" # Redis host
       password: "helloearth"
       db: 10 # Redis db_num
     # UDM网元用户库
     udmuser:
       port: 6379 # Redis port
-      host: "192.168.8.58"
+      host: "127.0.0.1"
       password: "helloearth"
       db: 0 # Redis db_num
   # 多个数据源时可以用这个指定默认的数据源
@@ -123,29 +123,32 @@ omc:
 
 # Alarm module setting
 # Forward interface:
+# TLS Skip verify: true/false
 # email/sms
 # smProxy: sms(Short Message Service)/smsc(SMS Centre)  
 alarm:
-  forwardAlarm: false
-  email:
+  alarmEmailForward:
+    enable: true
+    emailList: 
     smtp: mail.agrandtech.com
     port: 25
     user: smtpext@agrandtech.com
     password: "1000smtp@omc!"
-    # TLS skip verify: true/false
     tlsSkipVerify: true
-  smProxy: smsc
+  alarmSMSForward:
+    enable: true
+    mobileList: 
+    smscAddr: "192.168.13.114:2775"
+    systemID: "omc"
+    password: "omc123"
+    systemType: "UTRAN"
   sms:
     apiURL: http://smsc.xxx.com/
     accessKeyID: xxxx
     accessKeySecret: xxxx
     signName: xxx SMSC
     templateCode: 1000
-  smsc:
-    addr: "192.168.13.114:2775"
-    systemID: "omc"
-    password: "omc123"
-    systemType: "UTRAN"
+  smProxy: smsc
     
 #User authorized information
 # crypt: mysql/md5/bcrypt