feat: 新增函数定义

src/modules/network_element/fetch_link/alarm.go

@@ -0,0 +1,39 @@
+package fetchlink
+
+import (
+	"encoding/json"
+	"fmt"
+	"strings"
+
+	"be.ems/src/framework/logger"
+	"be.ems/src/framework/utils/fetch"
+	"be.ems/src/modules/network_element/model"
+)
+
+// AlarmHistory 告警历史记录
+func AlarmHistory(neInfo model.NeInfo) ([]map[string]any, error) {
+	// 网元参数配置信息
+	neUrl := fmt.Sprintf("http://%s:%d/api/rest/faultManagement/v1/elementType/%s/objectType/alarms", neInfo.IP, neInfo.Port, strings.ToLower(neInfo.NeType))
+	resBytes, err := fetch.Get(neUrl, nil, 60_000)
+	var resData []map[string]any
+	if err != nil {
+		errStr := err.Error()
+		logger.Warnf("AlarmHistory Get \"%s\"", neUrl)
+		if !(strings.HasPrefix(errStr, "201") || strings.HasPrefix(errStr, "202") || strings.HasPrefix(errStr, "204")) {
+			logger.Errorf("AlarmHistory %s", err.Error())
+			return nil, fmt.Errorf("NeService Alarm History Info API Error")
+		}
+	}
+
+	if len(resBytes) == 0 {
+		return resData, nil
+	}
+
+	// 序列化结果
+	err = json.Unmarshal(resBytes, &resData)
+	if err != nil {
+		logger.Errorf("AlarmHistory Unmarshal %s", err.Error())
+		return nil, err
+	}
+	return resData, nil
+}

src/modules/network_element/service/ne_info.go

@@ -129,6 +129,33 @@ func (r *NeInfo) SelectNeInfoByNeType(neType string) []model.NeInfo {
 	return neInfo
 }
 
+// FindByNeType 通过ne_type查询网元信息
+func (r NeInfo) FindByNeType(neType string) []model.NeInfo {
+	neInfo := make([]model.NeInfo, 0)
+	key := fmt.Sprintf("%s%s:*", cachekey.NE_KEY, strings.ToUpper(neType))
+	cacheKeys, _ := redis.GetKeys("", key)
+	if len(cacheKeys) > 0 {
+		for _, key := range cacheKeys {
+			var v model.NeInfo
+			jsonStr, _ := redis.Get("", key)
+			if len(jsonStr) > 7 {
+				json.Unmarshal([]byte(jsonStr), &v)
+			}
+			neInfo = append(neInfo, v)
+		}
+		return neInfo
+	} else {
+		neInfo = r.neInfoRepository.SelectList(model.NeInfo{NeType: neType})
+		for _, v := range neInfo {
+			key := fmt.Sprintf("%s%s:%s", cachekey.NE_KEY, strings.ToUpper(v.NeType), v.NeId)
+			redis.Del("", key)
+			values, _ := json.Marshal(v)
+			redis.Set("", key, string(values))
+		}
+	}
+	return neInfo
+}
+
 // SelectNeInfoByRmuid 通过rmUID查询网元信息
 func (r *NeInfo) SelectNeInfoByRmuid(rmUid string) model.NeInfo {
 	var neInfo model.NeInfo
@@ -160,6 +187,37 @@ func (r *NeInfo) SelectNeInfoByRmuid(rmUid string) model.NeInfo {
 	return neInfo
 }
 
+// FindByRmuid 通过rmUID查询网元信息
+func (r *NeInfo) FindByRmuid(rmUid string) model.NeInfo {
+	var neInfo model.NeInfo
+	cacheKeys, _ := redis.GetKeys("", cachekey.NE_KEY+"*")
+	if len(cacheKeys) > 0 {
+		for _, key := range cacheKeys {
+			var v model.NeInfo
+			jsonStr, _ := redis.Get("", key)
+			if len(jsonStr) > 7 {
+				json.Unmarshal([]byte(jsonStr), &v)
+			}
+			if v.RmUID == rmUid {
+				neInfo = v
+				break
+			}
+		}
+	} else {
+		neInfos := r.SelectList(neInfo, false, false)
+		for _, v := range neInfos {
+			key := fmt.Sprintf("%s%s:%s", cachekey.NE_KEY, strings.ToUpper(v.NeType), v.NeId)
+			redis.Del("", key)
+			values, _ := json.Marshal(v)
+			redis.Set("", key, string(values))
+			if v.RmUID == rmUid {
+				neInfo = v
+			}
+		}
+	}
+	return neInfo
+}
+
 // SelectPage 根据条件分页查询
 //
 // bandStatus 带状态信息
@@ -194,6 +252,26 @@ func (r *NeInfo) SelectList(ne model.NeInfo, bandStatus bool, bandHost bool) []m
 	return list
 }
 
+// Find 查询列表
+//
+// bandStatus 带状态信息
+// bandHost 带主机信息
+func (r NeInfo) Find(ne model.NeInfo, bandStatus bool, bandHost bool) []model.NeInfo {
+	list := r.neInfoRepository.SelectList(ne)
+
+	// 网元直连读取网元服务状态
+	if bandStatus {
+		r.bandNeStatus(&list)
+	}
+
+	// 网元主机信息
+	if bandHost {
+		r.bandNeHosts(&list)
+	}
+
+	return list
+}
+
 // bandNeStatus 网元列表项数据带网元服务状态
 func (r *NeInfo) bandNeStatus(arr *[]model.NeInfo) {
 	for i := range *arr {

src/modules/system/service/sys_dict_data.go

@@ -46,6 +46,11 @@ func (r *SysDictData) SelectDictDataByType(dictType string) []model.SysDictData
 	return r.sysDictTypeService.DictDataCache(dictType)
 }
 
+// FindByType 根据字典类型查询信息
+func (r *SysDictData) FindByType(dictType string) []model.SysDictData {
+	return r.sysDictTypeService.DictDataCache(dictType)
+}
+
 // CheckUniqueDictLabel 校验字典标签是否唯一
 func (r *SysDictData) CheckUniqueDictLabel(dictType, dictLabel, dictCode string) bool {
 	uniqueId := r.sysDictDataRepository.CheckUniqueDictData(model.SysDictData{