feat: udmk4和网元配置yaml测试函数
This commit is contained in:
TsMask
@@ -6,7 +6,7 @@ import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
// 加密
|
||||
// UDM K4加密
|
||||
func encrypt(origData, key []byte) ([]byte, error) {
|
||||
if len(origData) < 1 || len(key) < 1 {
|
||||
return nil, errors.New("wrong data or key")
|
||||
@@ -36,11 +36,11 @@ func TestEncrypt(t *testing.T) {
|
||||
// ki := []byte{0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef}
|
||||
// 0123456789abcdef0123456789abcdef
|
||||
|
||||
// 密码
|
||||
// k4 password
|
||||
key := []byte{0x12, 0x34, 0x12, 0x34, 0x12, 0x34, 0x12, 0x34}
|
||||
// 1234123412341234
|
||||
|
||||
// 要加密的ki
|
||||
// k4 crypt ki
|
||||
ki := []byte{0x80, 0x5D, 0xAD, 0xC6, 0xE8, 0xA5, 0x4A, 0x0D, 0x59, 0xD6, 0x22, 0xC7, 0xA0, 0x4D, 0x08, 0xE0}
|
||||
// 805DADC6E8A54A0D59D622C7A04D08E0
|
||||
|
||||
230
src/modules/network_element/ne_config_test.go
Normal file
230
src/modules/network_element/ne_config_test.go
Normal file
@@ -0,0 +1,230 @@
|
||||
package networkelement
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"be.ems/src/modules/network_element/model"
|
||||
"gopkg.in/yaml.v3"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
const (
|
||||
// 数据库
|
||||
DbHost = "192.168.8.58"
|
||||
DbPort = 33066
|
||||
DbUser = "root"
|
||||
DbPassswd = "1000omc@kp!"
|
||||
DbName = "omc_db"
|
||||
// 配置文件路径
|
||||
configParamDir = "../../../config/param"
|
||||
// configParamFile = "*" // 目录下全部更新
|
||||
configParamFile = "upf_param_config.yaml" // 单文件更新
|
||||
)
|
||||
|
||||
func TestEncrypt(t *testing.T) {
|
||||
fileNameList, err := getDirFileNameList(configParamDir)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
return
|
||||
}
|
||||
|
||||
if configParamFile == "*" {
|
||||
for _, v := range fileNameList {
|
||||
params := parseData(filepath.Join(configParamDir, v))
|
||||
if params == nil {
|
||||
return
|
||||
}
|
||||
saveData(params)
|
||||
}
|
||||
} else {
|
||||
params := parseData(filepath.Join(configParamDir, configParamFile))
|
||||
if params == nil {
|
||||
return
|
||||
}
|
||||
saveData(params)
|
||||
}
|
||||
}
|
||||
|
||||
// ========= Main =============
|
||||
|
||||
// parseData 文件转map数据
|
||||
func parseData(filePaht string) []map[string]string {
|
||||
data, err := parseStrToMap(filePaht)
|
||||
if err != nil {
|
||||
log.Printf("parseStrToMap => %s", err.Error())
|
||||
return nil
|
||||
}
|
||||
params, err := parseParamConfig(data)
|
||||
if err != nil {
|
||||
log.Printf("parseParamConfig => %s", err.Error())
|
||||
return nil
|
||||
}
|
||||
return params
|
||||
}
|
||||
|
||||
// saveData 保存数据
|
||||
func saveData(params []map[string]string) {
|
||||
// 定义排序函数
|
||||
sort.Slice(params, func(i, j int) bool {
|
||||
paramSortI := params[i]["paramSort"]
|
||||
if len(paramSortI) == 0 || paramSortI == "" {
|
||||
paramSortI = "0"
|
||||
}
|
||||
paramSortJ := params[j]["paramSort"]
|
||||
if len(paramSortJ) == 0 || paramSortJ == "" {
|
||||
paramSortJ = "0"
|
||||
}
|
||||
// 将 age 字段转换为整数进行比较
|
||||
si, _ := strconv.Atoi(paramSortI)
|
||||
sj, _ := strconv.Atoi(paramSortJ)
|
||||
return si < sj
|
||||
})
|
||||
// 遍历插入
|
||||
for _, v := range params {
|
||||
paramSort := v["paramSort"]
|
||||
if len(paramSort) == 0 || paramSort == "" {
|
||||
paramSort = "0"
|
||||
}
|
||||
sort, err := strconv.ParseInt(paramSort, 10, 64)
|
||||
if err != nil {
|
||||
sort = 0
|
||||
}
|
||||
|
||||
neConfig := model.NeConfig{
|
||||
NeType: v["neType"],
|
||||
ParamName: v["paramName"],
|
||||
ParamDisplay: v["paramDisplay"],
|
||||
ParamType: v["paramType"],
|
||||
ParamJson: v["paramJson"],
|
||||
ParamPerms: v["paramPerms"],
|
||||
ParamSort: sort,
|
||||
}
|
||||
neConfig.ID = saveDB(neConfig)
|
||||
log.Println(neConfig.ID, neConfig.NeType, neConfig.ParamDisplay)
|
||||
}
|
||||
}
|
||||
|
||||
// ========= DB =============
|
||||
|
||||
var gdb *gorm.DB
|
||||
|
||||
// connDB 连接到数据库
|
||||
func connDB() *gorm.DB {
|
||||
if gdb != nil {
|
||||
return gdb
|
||||
}
|
||||
|
||||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", DbUser, DbPassswd, DbHost, DbPort, DbName)
|
||||
newLogger := logger.New(
|
||||
log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer
|
||||
logger.Config{
|
||||
SlowThreshold: time.Minute, // Slow SQL threshold
|
||||
LogLevel: logger.Error, // Log level
|
||||
IgnoreRecordNotFoundError: true, // Ignore ErrRecordNotFound error for logger
|
||||
ParameterizedQueries: true, // Don't include params in the SQL log
|
||||
Colorful: false, // Disable color
|
||||
},
|
||||
)
|
||||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{Logger: newLogger})
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
gdb = db
|
||||
return gdb
|
||||
}
|
||||
|
||||
// saveDB 表插入或更新
|
||||
func saveDB(s model.NeConfig) string {
|
||||
db := connDB()
|
||||
// 检查是否存在
|
||||
var id string
|
||||
db.Raw("SELECT id FROM ne_config WHERE ne_type = ? AND param_name = ?", s.NeType, s.ParamName).Scan(&id)
|
||||
// 更新时间
|
||||
s.UpdateTime = time.Now().UnixMilli()
|
||||
if id != "" {
|
||||
s.ID = id
|
||||
db.Save(s)
|
||||
} else {
|
||||
db.Create(s)
|
||||
}
|
||||
return s.ID
|
||||
}
|
||||
|
||||
// ========= Utils =============
|
||||
|
||||
// getDirFileNameList 获取文件目录下所有文件名称，不含目录名称
|
||||
func getDirFileNameList(dirPath string) ([]string, error) {
|
||||
fileNames := []string{}
|
||||
|
||||
dir, err := os.Open(dirPath)
|
||||
if err != nil {
|
||||
return fileNames, nil
|
||||
}
|
||||
defer dir.Close()
|
||||
|
||||
fileInfos, err := dir.Readdir(-1)
|
||||
if err != nil {
|
||||
return fileNames, err
|
||||
}
|
||||
|
||||
for _, fileInfo := range fileInfos {
|
||||
if fileInfo.Mode().IsRegular() {
|
||||
fileNames = append(fileNames, fileInfo.Name())
|
||||
}
|
||||
}
|
||||
|
||||
return fileNames, nil
|
||||
}
|
||||
|
||||
// parseStrToMap 解析内容string到map
|
||||
func parseStrToMap(filePath string) (map[string]any, error) {
|
||||
// 读取文件内容
|
||||
bytes, err := os.ReadFile(filePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
content := string(bytes)
|
||||
var configMap map[string]any
|
||||
err = yaml.Unmarshal([]byte(content), &configMap)
|
||||
|
||||
return configMap, err
|
||||
}
|
||||
|
||||
// parseParamConfig 解析内容文件数据
|
||||
func parseParamConfig(data map[string]any) ([]map[string]string, error) {
|
||||
paramMapArr := make([]map[string]string, 0)
|
||||
for k, v := range data {
|
||||
for ik, iv := range v.(map[string]any) {
|
||||
itemMap := make(map[string]string)
|
||||
itemMap["neType"] = strings.ToUpper(k)
|
||||
itemMap["paramName"] = ik
|
||||
for iik, iiv := range iv.(map[string]any) {
|
||||
switch iik {
|
||||
case "display":
|
||||
itemMap["paramDisplay"] = iiv.(string)
|
||||
case "sort":
|
||||
itemMap["paramSort"] = fmt.Sprint(iiv)
|
||||
case "perms", "method":
|
||||
itemMap["paramPerms"] = iiv.(string)
|
||||
case "data", "list", "array":
|
||||
itemMap["paramType"] = iik
|
||||
strByte, _ := json.Marshal(iiv)
|
||||
itemMap["paramJson"] = string(strByte)
|
||||
}
|
||||
}
|
||||
paramMapArr = append(paramMapArr, itemMap)
|
||||
}
|
||||
}
|
||||
return paramMapArr, nil
|
||||
}
|
||||
Reference in New Issue
Block a user