144 lines
3.6 KiB
Go
144 lines
3.6 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"be.ems/src/framework/constants/cachekey"
|
|
"be.ems/src/framework/redis"
|
|
"be.ems/src/framework/utils/parse"
|
|
"be.ems/src/modules/network_data/model"
|
|
"be.ems/src/modules/network_data/repository"
|
|
neModel "be.ems/src/modules/network_element/model"
|
|
)
|
|
|
|
// 实例化数据层 PerfKPI 结构体
|
|
var NewPerfKPI = &PerfKPI{
|
|
perfKPIRepository: repository.NewPerfKPI,
|
|
}
|
|
|
|
// PerfKPI 性能统计 服务层处理
|
|
type PerfKPI struct {
|
|
perfKPIRepository *repository.PerfKPI // 性能统计数据信息
|
|
}
|
|
|
|
// SelectGoldKPI 通过网元指标数据信息
|
|
func (r *PerfKPI) SelectGoldKPI(query model.GoldKPIQuery) []map[string]any {
|
|
// 获取数据指标id
|
|
var kpiIds []string
|
|
kpiTitles := r.perfKPIRepository.SelectGoldKPITitle(query.NeType)
|
|
for _, kpiId := range kpiTitles {
|
|
kpiIds = append(kpiIds, kpiId.KPIID)
|
|
}
|
|
|
|
data := r.perfKPIRepository.SelectGoldKPI(query, kpiIds)
|
|
if data == nil {
|
|
return []map[string]any{}
|
|
}
|
|
return data
|
|
}
|
|
|
|
// SelectGoldKPITitle 网元对应的指标名称
|
|
func (r *PerfKPI) SelectGoldKPITitle(neType string) []model.GoldKPITitle {
|
|
return r.perfKPIRepository.SelectGoldKPITitle(neType)
|
|
}
|
|
|
|
// UPFTodayFlowFind 查询UPF总流量 N3上行 N6下行
|
|
func (r PerfKPI) UPFTodayFlowFind(rmUID string, day int) (int64, int64) {
|
|
// 获取当前日期
|
|
now := time.Now()
|
|
var upTotal, downTotal int64
|
|
|
|
// 查询最近7天的数据
|
|
for i := 0; i <= day; i++ {
|
|
dateKey := now.AddDate(0, 0, -i).Format("2006-01-02")
|
|
key := fmt.Sprintf("%sUPF_FLOW:%s:%s", cachekey.NE_DATA_KEY, rmUID, dateKey)
|
|
|
|
// 读取缓存数据
|
|
up, err := redis.GetHash("", key, "up")
|
|
if err != nil || up == "" {
|
|
up = "0"
|
|
}
|
|
down, err := redis.GetHash("", key, "down")
|
|
if err != nil || down == "" {
|
|
down = "0"
|
|
}
|
|
|
|
upTotal += parse.Number(up)
|
|
downTotal += parse.Number(down)
|
|
}
|
|
|
|
return upTotal, downTotal
|
|
}
|
|
|
|
// UPFTodayFlow UPF流量今日统计
|
|
func (r PerfKPI) UPFTodayFlowUpdate(rmUID string, upValue, downValue int64, rest bool) error {
|
|
// 按日期存储统计数据
|
|
dateKey := time.Now().Format("2006-01-02")
|
|
key := fmt.Sprintf("%sUPF_FLOW:%s:%s", cachekey.NE_DATA_KEY, rmUID, dateKey)
|
|
|
|
// 重置数据
|
|
if rest {
|
|
err := redis.SetHash("", key, map[string]any{
|
|
"up": upValue,
|
|
"down": downValue,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 设置key的过期时间为30天，自动清理旧数据
|
|
err = redis.Expire("", key, 30*24*time.Hour)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// 使用HIncrBy实时累加统计值
|
|
if err := redis.IncrBy("", key, "up", upValue); err != nil {
|
|
return err
|
|
}
|
|
if err := redis.IncrBy("", key, "down", downValue); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// UPFTodayFlowLoad UPF上下行数据到redis
|
|
func (r PerfKPI) UPFTodayFlowLoad() {
|
|
cacheKeys, _ := redis.GetKeys("", cachekey.NE_KEY+"UPF:*")
|
|
if len(cacheKeys) == 0 {
|
|
return
|
|
}
|
|
|
|
// 今日流量
|
|
now := time.Now()
|
|
beginTime := now.Truncate(24 * time.Hour).UnixMilli()
|
|
endTime := beginTime + 24*60*60*1000 - 1
|
|
|
|
for _, key := range cacheKeys {
|
|
var v neModel.NeInfo
|
|
jsonStr, _ := redis.Get("", key)
|
|
if len(jsonStr) > 7 {
|
|
json.Unmarshal([]byte(jsonStr), &v)
|
|
}
|
|
if v.NeType == "UPF" && v.RmUID != "" {
|
|
// 查询历史数据
|
|
// down * 8 / 1000 / 1000 单位M
|
|
info := r.perfKPIRepository.SelectUPFTotalFlow("UPF", v.RmUID, fmt.Sprint(beginTime), fmt.Sprint(endTime))
|
|
if v, ok := info["up"]; ok && v == nil {
|
|
info["up"] = 0
|
|
}
|
|
if v, ok := info["down"]; ok && v == nil {
|
|
info["down"] = 0
|
|
}
|
|
|
|
upTotal := parse.Number(info["up"])
|
|
downTotal := parse.Number(info["down"])
|
|
|
|
// 将历史数据添加到Redis
|
|
r.UPFTodayFlowUpdate(v.RmUID, upTotal, downTotal, true)
|
|
}
|
|
}
|
|
}
|