marge: 合并代码

src/modules/network_element/service/ne_info.impl.go

@@ -1,6 +1,11 @@
 package service
 
 import (
+	"encoding/json"
+	"fmt"
+
+	"ems.agt/src/framework/constants/cachekey"
+	"ems.agt/src/framework/redis"
 	"ems.agt/src/modules/network_element/model"
 	"ems.agt/src/modules/network_element/repository"
 )
@@ -18,7 +23,50 @@ type NeInfoImpl struct {
 
 // SelectNeInfoByNeTypeAndNeID 通过ne_type和ne_id查询网元信息
 func (r *NeInfoImpl) SelectNeInfoByNeTypeAndNeID(neType, neID string) model.NeInfo {
-	return r.neInfoRepository.SelectNeInfoByNeTypeAndNeID(neType, neID)
+	var neInfo model.NeInfo
+	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)
+		if err != nil {
+			neInfo = model.NeInfo{}
+		}
+	} else {
+		neInfo = r.neInfoRepository.SelectNeInfoByNeTypeAndNeID(neType, neID)
+		if neInfo.NeId == neID {
+			redis.Del("", key)
+			values, _ := json.Marshal(neInfo)
+			redis.Set("", key, string(values))
+		}
+	}
+	return neInfo
+}
+
+// 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)
+	redis.Del("", key)
+	neInfo = r.neInfoRepository.SelectNeInfoByNeTypeAndNeID(neType, neID)
+	if neInfo.NeId == neID {
+		values, _ := json.Marshal(neInfo)
+		redis.Set("", key, string(values))
+	}
+	return neInfo
+}
+
+// ClearNeCacheByNeType 清除网元类型缓存
+func (r *NeInfoImpl) ClearNeCacheByNeType(neType string) bool {
+	key := fmt.Sprintf("%s*", cachekey.NE_KEY)
+	if neType != "*" {
+		key = fmt.Sprintf("%s%s*", cachekey.NE_KEY, neType)
+	}
+	keys, err := redis.GetKeys("", key)
+	if err != nil {
+		return false
+	}
+	delOk, _ := redis.DelKeys("", keys)
+	return delOk
 }
 
 // SelectNeList 查询网元列表