From f375959e9134532446b10399f5755f46e4e422b7 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Wed, 6 Mar 2024 15:06:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=BD=91=E5=85=83=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E7=BC=93=E5=AD=98redis?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../network_element/service/ne_info.impl.go   | 25 +++++++++++++++----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/src/modules/network_element/service/ne_info.impl.go b/src/modules/network_element/service/ne_info.impl.go
index 33554fe8..8949ae77 100644
--- a/src/modules/network_element/service/ne_info.impl.go
+++ b/src/modules/network_element/service/ne_info.impl.go
@@ -28,7 +28,7 @@ type NeInfoImpl struct {
 // SelectNeInfoByNeTypeAndNeID 通过ne_type和ne_id查询网元信息
 func (r *NeInfoImpl) SelectNeInfoByNeTypeAndNeID(neType, neID string) model.NeInfo {
 	var neInfo model.NeInfo
-	key := fmt.Sprintf("%s%s.%s", cachekey.NE_KEY, neType, neID)
+	key := fmt.Sprintf("%s%s:%s", cachekey.NE_KEY, neType, neID)
 	jsonStr, _ := redis.Get("", key)
 	if len(jsonStr) > 7 {
 		err := json.Unmarshal([]byte(jsonStr), &neInfo)
@@ -49,7 +49,7 @@ func (r *NeInfoImpl) SelectNeInfoByNeTypeAndNeID(neType, neID string) model.NeIn
 // RefreshByNeTypeAndNeID 通过ne_type和ne_id刷新redis中的缓存
 func (r *NeInfoImpl) RefreshByNeTypeAndNeID(neType, neID string) model.NeInfo {
 	var neInfo model.NeInfo
-	key := fmt.Sprintf("%s%s.%s", cachekey.NE_KEY, neType, neID)
+	key := fmt.Sprintf("%s%s:%s", cachekey.NE_KEY, neType, neID)
 	redis.Del("", key)
 	neInfo = r.neInfoRepository.SelectNeInfoByNeTypeAndNeID(neType, neID)
 	if neInfo.NeId == neID {
@@ -188,7 +188,13 @@ func (r *NeInfoImpl) Insert(neInfo model.NeInfo) string {
 		}
 		neInfo.HostIDs = strings.Join(hostIDs, ",")
 	}
-	return r.neInfoRepository.Insert(neInfo)
+
+	insertId := r.neInfoRepository.Insert(neInfo)
+	if insertId != "" {
+		// 刷新缓存
+		r.RefreshByNeTypeAndNeID(neInfo.NeType, neInfo.NeId)
+	}
+	return insertId
 }
 
 // Update 修改信息
@@ -203,7 +209,13 @@ func (r *NeInfoImpl) Update(neInfo model.NeInfo) int64 {
 			}
 		}
 	}
-	return r.neInfoRepository.Update(neInfo)
+
+	num := r.neInfoRepository.Update(neInfo)
+	if num > 0 {
+		// 刷新缓存
+		r.RefreshByNeTypeAndNeID(neInfo.NeType, neInfo.NeId)
+	}
+	return num
 }
 
 // DeleteByIds 批量删除信息
@@ -215,12 +227,15 @@ func (r *NeInfoImpl) DeleteByIds(infoIds []string) (int64, error) {
 	}
 
 	if len(infos) == len(infoIds) {
-		// 主机信息删除
 		for _, v := range infos {
+			// 主机信息删除
 			if v.HostIDs != "" {
 				hostIds := strings.Split(v.HostIDs, ",")
 				r.neHostRepository.DeleteByIds(hostIds)
 			}
+			// 缓存信息删除
+			key := fmt.Sprintf("%s%s:%s", cachekey.NE_KEY, v.NeType, v.NeId)
+			redis.Del("", key)
 		}
 		rows := r.neInfoRepository.DeleteByIds(infoIds)
 		return rows, nil