TsMask
e7ae390f6e
Refactor network element state management - Removed the NE state endpoint and related service logic from the network_data module. - Introduced a new NEStateController to handle network element state records. - Implemented NEState service and repository for managing state records in the database. - Updated NEInfo and NeLicense controllers to utilize the new NEState service for fetching and saving state information. - Enhanced state handling in the websocket processor to reflect the latest state of network elements. - Added caching logic for network element states using Redis. - Improved error handling and response formatting for state queries.
535 lines
16 KiB
Go
535 lines
16 KiB
Go
package controller
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"be.ems/src/framework/constants"
|
|
"be.ems/src/framework/database/redis"
|
|
"be.ems/src/framework/i18n"
|
|
"be.ems/src/framework/reqctx"
|
|
"be.ems/src/framework/resp"
|
|
"be.ems/src/framework/utils/parse"
|
|
neFetchlink "be.ems/src/modules/network_element/fetch_link"
|
|
"be.ems/src/modules/network_element/model"
|
|
neService "be.ems/src/modules/network_element/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// 实例化控制层 NeInfoController 结构体
|
|
var NewNeInfo = &NeInfoController{
|
|
neInfoService: neService.NewNeInfo,
|
|
neLicenseService: neService.NewNeLicense,
|
|
neVersionService: neService.NewNeVersion,
|
|
neStateService: neService.NewNEState,
|
|
}
|
|
|
|
// 网元信息请求
|
|
//
|
|
// PATH /info
|
|
type NeInfoController struct {
|
|
neInfoService *neService.NeInfo // 网元信息服务
|
|
neLicenseService *neService.NeLicense // 网元授权激活信息服务
|
|
neVersionService *neService.NeVersion // 网元版本信息服务
|
|
neStateService *neService.NEState // 网元状态服务
|
|
}
|
|
|
|
// 网元信息状态
|
|
//
|
|
// GET /state
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param neType query string true "NE Type" Enums(IMS,AMF,AUSF,UDM,SMF,PCF,NSSF,NRF,UPF,MME,CBC,OMC,SGWC,SMSC)
|
|
// @Param neId query string true "NE ID" default(001)
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information state
|
|
// @Description Network element information state
|
|
// @Router /ne/info/state [get]
|
|
func (s NeInfoController) State(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var querys struct {
|
|
NeType string `form:"neType" binding:"required"`
|
|
NeId string `form:"neId" binding:"required"`
|
|
}
|
|
if err := c.ShouldBindQuery(&querys); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
neInfo := s.neInfoService.FindByNeTypeAndNeID(querys.NeType, querys.NeId)
|
|
if neInfo.NeId != querys.NeId || neInfo.IP == "" {
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
|
return
|
|
}
|
|
state := s.neStateService.Last(neInfo.NeType, neInfo.NeId)
|
|
result := s.neStateService.ParseState(neInfo, state)
|
|
|
|
// 没状态标记，视为离线
|
|
key := fmt.Sprintf("%s:%s:%s_status", constants.CACHE_NE_INFO, neInfo.NeType, neInfo.NeId)
|
|
keyNum, err := redis.Has("", key)
|
|
if keyNum == 0 || err != nil {
|
|
result["online"] = false
|
|
} else {
|
|
result["online"] = true
|
|
}
|
|
c.JSON(200, resp.OkData(result))
|
|
}
|
|
|
|
// 网元neType和neID查询
|
|
//
|
|
// GET /byTypeAndID
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param neType query string true "NE Type" Enums(IMS,AMF,AUSF,UDM,SMF,PCF,NSSF,NRF,UPF,MME,CBC,OMC,SGWC,SMSC)
|
|
// @Param neId query string true "NE ID" default(001)
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element neType and neID queries
|
|
// @Description Network element neType and neID queries
|
|
// @Router /ne/info/byTypeAndID [get]
|
|
func (s NeInfoController) NeTypeAndID(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var querys struct {
|
|
NeType string `form:"neType" binding:"required"`
|
|
NeID string `form:"neId" binding:"required"`
|
|
}
|
|
if err := c.ShouldBindQuery(&querys); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
neInfo := s.neInfoService.FindByNeTypeAndNeID(querys.NeType, querys.NeID)
|
|
if neInfo.NeId != querys.NeID || neInfo.IP == "" {
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
|
return
|
|
}
|
|
c.JSON(200, resp.OkData(neInfo))
|
|
}
|
|
|
|
// 网元信息列表全部无分页
|
|
//
|
|
// GET /listAll
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param neType query string true "NE Type"
|
|
// @Param neId query string true "NE ID" default(001)
|
|
// @Param bandStatus query boolean true "With status information"
|
|
// @Param bandHost query boolean true "With host information"
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary The list of network element information is all unpaginated
|
|
// @Description The list of network element information is all unpaginated
|
|
// @Router /ne/info/listAll [get]
|
|
func (s NeInfoController) ListAll(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var querys struct {
|
|
NeType string `form:"neType"`
|
|
NeId string `form:"neId"`
|
|
BandStatus bool `form:"bandStatus"`
|
|
BandHost bool `form:"bandHost"`
|
|
}
|
|
if err := c.ShouldBindQuery(&querys); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
// 查询实体参数
|
|
ne := model.NeInfo{}
|
|
if querys.NeType != "" {
|
|
ne.NeType = querys.NeType
|
|
}
|
|
if querys.NeId != "" {
|
|
ne.NeId = querys.NeId
|
|
}
|
|
neList := s.neInfoService.Find(ne, querys.BandStatus, querys.BandHost)
|
|
if len(neList) == 0 {
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
|
return
|
|
}
|
|
c.JSON(200, resp.OkData(neList))
|
|
}
|
|
|
|
// 网元端Para5G配置文件读取
|
|
//
|
|
// GET /para5GFile
|
|
func (s NeInfoController) Para5GFileRead(c *gin.Context) {
|
|
data, err := s.neInfoService.NeConfPara5GRead()
|
|
if err != nil {
|
|
c.JSON(200, resp.ErrMsg(err.Error()))
|
|
return
|
|
}
|
|
c.JSON(200, resp.OkData(data))
|
|
}
|
|
|
|
// 网元端Para5G配置文件写入
|
|
//
|
|
// PUT /para5GFile
|
|
func (s NeInfoController) Para5GFileWrite(c *gin.Context) {
|
|
var body struct {
|
|
Content map[string]any `json:"content" binding:"required"` // 内容
|
|
SyncNE []string `json:"syncNe"` // 同步到网元
|
|
}
|
|
if err := c.ShouldBindBodyWithJSON(&body); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
err := s.neInfoService.NeConfPara5GWirte(body.Content, body.SyncNE)
|
|
if err != nil {
|
|
c.JSON(200, resp.ErrMsg(err.Error()))
|
|
return
|
|
}
|
|
c.JSON(200, resp.Ok(nil))
|
|
}
|
|
|
|
// 网元端OAM配置文件读取
|
|
//
|
|
// GET /oamFile
|
|
func (s NeInfoController) OAMFileRead(c *gin.Context) {
|
|
var querys struct {
|
|
NeType string `form:"neType" binding:"required"`
|
|
NeID string `form:"neId" binding:"required"`
|
|
}
|
|
if err := c.ShouldBindQuery(&querys); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
data, err := s.neInfoService.NeConfOAMReadSync(querys.NeType, querys.NeID)
|
|
if err != nil {
|
|
c.JSON(200, resp.ErrMsg(err.Error()))
|
|
return
|
|
}
|
|
c.JSON(200, resp.OkData(data))
|
|
}
|
|
|
|
// 网元端OAM配置文件写入
|
|
//
|
|
// PUT /oamFile
|
|
func (s NeInfoController) OAMFileWrite(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var body struct {
|
|
NeType string `json:"neType" binding:"required"`
|
|
NeID string `json:"neId" binding:"required"`
|
|
Content map[string]any `json:"content" binding:"required"` // 内容
|
|
Sync bool `json:"sync"` // 同步到网元
|
|
}
|
|
if err := c.ShouldBindBodyWithJSON(&body); err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
|
|
// 查询网元获取IP
|
|
neInfo := s.neInfoService.FindByNeTypeAndNeID(body.NeType, body.NeID)
|
|
if neInfo.NeId != body.NeID || neInfo.IP == "" {
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
|
return
|
|
}
|
|
|
|
err := s.neInfoService.NeConfOAMWirteSync(neInfo, body.Content, body.Sync)
|
|
if err != nil {
|
|
c.JSON(200, resp.ErrMsg(err.Error()))
|
|
return
|
|
}
|
|
c.JSON(200, resp.Ok(nil))
|
|
}
|
|
|
|
// 网元信息列表
|
|
//
|
|
// GET /list
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param bandStatus query boolean false "The result carries the state of the network element"
|
|
// @Param neId query string false "NE ID"
|
|
// @Param neType query string false "Ne Type"
|
|
// @Param pageNum query number true "pageNum" default(1)
|
|
// @Param pageSize query number true "pageSize" default(10)
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information list
|
|
// @Description Network element information list
|
|
// @Router /ne/info/list [get]
|
|
func (s NeInfoController) List(c *gin.Context) {
|
|
query := reqctx.QueryMap(c)
|
|
bandStatus := false
|
|
if v, ok := query["bandStatus"]; ok {
|
|
bandStatus = parse.Boolean(v)
|
|
}
|
|
rows, total := s.neInfoService.FindByPage(query, bandStatus)
|
|
c.JSON(200, resp.OkData(map[string]any{"rows": rows, "total": total}))
|
|
}
|
|
|
|
// 网元信息
|
|
//
|
|
// GET /:id
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param value path string true "Row ID"
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information
|
|
// @Description Network element information
|
|
// @Router /ne/info/{value} [get]
|
|
func (s NeInfoController) Info(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
id := parse.Number(c.Param("id"))
|
|
if id <= 0 {
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: id is empty"))
|
|
return
|
|
}
|
|
|
|
neHost := s.neInfoService.FindById(id, true)
|
|
if neHost.ID != id {
|
|
// 没有可访问网元信息数据！
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "neInfo.noData")))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp.OkData(neHost))
|
|
}
|
|
|
|
// 网元信息新增
|
|
//
|
|
// POST /
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param data body object true "Request Param"
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information addition
|
|
// @Description Network element information addition
|
|
// @Router /ne/info [post]
|
|
func (s NeInfoController) Add(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var body model.NeInfo
|
|
err := c.ShouldBindBodyWithJSON(&body)
|
|
if err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
if body.ID != 0 {
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: id not is empty"))
|
|
return
|
|
}
|
|
|
|
// 检查属性值唯一
|
|
uniqueInfo := s.neInfoService.CheckUniqueNeTypeAndNeId(body.NeType, body.NeId, 0)
|
|
if !uniqueInfo {
|
|
// 网元信息操作【%s】失败，同类型下标识已存在
|
|
msg := i18n.TTemplate(language, "neInfo.errKeyExists", map[string]any{"key": body.NeId})
|
|
c.JSON(200, resp.ErrMsg(msg))
|
|
return
|
|
}
|
|
|
|
loginUserName := reqctx.LoginUserToUserName(c)
|
|
// 新增Version信息
|
|
neVersion := model.NeVersion{
|
|
NeType: body.NeType,
|
|
NeId: body.NeId,
|
|
CreateBy: loginUserName,
|
|
}
|
|
// 新增License信息
|
|
neLicense := model.NeLicense{
|
|
NeType: body.NeType,
|
|
NeId: body.NeId,
|
|
CreateBy: loginUserName,
|
|
}
|
|
|
|
// 获取网元状态是否正常
|
|
key := fmt.Sprintf("%s:%s:%s_status", constants.CACHE_NE_INFO, body.NeType, body.NeId)
|
|
state, err := neFetchlink.NeState(body)
|
|
if err != nil {
|
|
redis.Del("", key)
|
|
body.Status = 0
|
|
} else {
|
|
redis.Set("", key, body.RmUID, 0)
|
|
// 网元状态设置为在线
|
|
body.Status = 1
|
|
if parse.Boolean(state["standby"]) {
|
|
body.Status = 3
|
|
}
|
|
// 下发网管配置信息给网元
|
|
if _, err = neFetchlink.NeConfigOMC(body); err != nil {
|
|
body.Status = 2
|
|
}
|
|
// 网元状态记录
|
|
neState := s.neStateService.SaveState(state)
|
|
body.ServerState = s.neStateService.ParseState(body, neState)
|
|
if neState.ID > 0 {
|
|
body.NeVersion = neState.NeVersion
|
|
neVersion.Version = neState.NeVersion
|
|
neLicense.SerialNum = neState.SerialNum
|
|
neLicense.ExpiryDate = neState.ExpiryDate
|
|
neLicense.Status = "1"
|
|
neLicense.UeNumber = neState.UENumber
|
|
neLicense.NbNumber = neState.NbNumber
|
|
}
|
|
}
|
|
|
|
s.neVersionService.Insert(neVersion)
|
|
s.neLicenseService.Insert(neLicense)
|
|
body.CreateBy = loginUserName
|
|
insertId := s.neInfoService.Insert(body)
|
|
if insertId > 0 {
|
|
c.JSON(200, resp.OkData(insertId))
|
|
return
|
|
}
|
|
c.JSON(200, resp.Err(nil))
|
|
}
|
|
|
|
// 网元信息修改
|
|
//
|
|
// PUT /
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param data body object true "Request Param"
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information modification
|
|
// @Description Network element information modification
|
|
// @Router /ne/info [put]
|
|
func (s NeInfoController) Edit(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
var body model.NeInfo
|
|
err := c.ShouldBindBodyWithJSON(&body)
|
|
if err != nil {
|
|
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
|
|
return
|
|
}
|
|
if body.ID <= 0 {
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: id is empty"))
|
|
return
|
|
}
|
|
|
|
// 检查属性值唯一
|
|
uniqueInfo := s.neInfoService.CheckUniqueNeTypeAndNeId(body.NeType, body.NeId, body.ID)
|
|
if !uniqueInfo {
|
|
// 网元信息操作【%s】失败，同类型下标识已存在
|
|
msg := i18n.TTemplate(language, "neInfo.errKeyExists", map[string]any{"key": body.NeId})
|
|
c.JSON(200, resp.ErrMsg(msg))
|
|
return
|
|
}
|
|
|
|
// 检查是否存在
|
|
neInfo := s.neInfoService.FindById(body.ID, false)
|
|
if neInfo.ID != body.ID {
|
|
// 没有可访问网元信息数据！
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "neInfo.noData")))
|
|
return
|
|
}
|
|
// 赋予主机ID
|
|
if neInfo.HostIDs != "" && len(body.Hosts) > 0 {
|
|
hostIDs := strings.Split(neInfo.HostIDs, ",")
|
|
for index, id := range hostIDs {
|
|
body.Hosts[index].ID = parse.Number(id)
|
|
}
|
|
}
|
|
|
|
loginUserName := reqctx.LoginUserToUserName(c)
|
|
neLicense := s.neLicenseService.FindByNeTypeAndNeID(neInfo.NeType, neInfo.NeId)
|
|
neVersion := s.neVersionService.FindByNeTypeAndNeID(neInfo.NeType, neInfo.NeId)
|
|
|
|
// 获取网元状态是否正常
|
|
key := fmt.Sprintf("%s:%s:%s_status", constants.CACHE_NE_INFO, body.NeType, body.NeId)
|
|
state, err := neFetchlink.NeState(body)
|
|
if err != nil {
|
|
redis.Del("", key)
|
|
body.Status = 0
|
|
} else {
|
|
redis.Set("", key, body.RmUID, 0)
|
|
// 网元状态设置为在线
|
|
body.Status = 1
|
|
if parse.Boolean(state["standby"]) {
|
|
body.Status = 3
|
|
}
|
|
// 下发网管配置信息给网元
|
|
if _, err = neFetchlink.NeConfigOMC(body); err != nil {
|
|
body.Status = 2
|
|
}
|
|
// 网元状态记录
|
|
neState := s.neStateService.SaveState(state)
|
|
body.ServerState = s.neStateService.ParseState(body, neState)
|
|
if neState.ID > 0 {
|
|
body.NeVersion = neState.NeVersion
|
|
neVersion.Version = neState.NeVersion
|
|
s.neVersionService.Update(neVersion)
|
|
neLicense.SerialNum = neState.SerialNum
|
|
neLicense.ExpiryDate = neState.ExpiryDate
|
|
neLicense.Status = "1"
|
|
neLicense.UeNumber = neState.UENumber
|
|
neLicense.NbNumber = neState.NbNumber
|
|
s.neLicenseService.Update(neLicense)
|
|
}
|
|
}
|
|
|
|
body.UpdateBy = loginUserName
|
|
rows := s.neInfoService.Update(body)
|
|
if rows > 0 {
|
|
c.JSON(200, resp.Ok(nil))
|
|
return
|
|
}
|
|
c.JSON(200, resp.Err(nil))
|
|
}
|
|
|
|
// 网元信息删除
|
|
//
|
|
// DELETE /:id
|
|
//
|
|
// @Tags network_element/info
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param value path string true "Row ID"
|
|
// @Success 200 {object} object "Response Results"
|
|
// @Security TokenAuth
|
|
// @Summary Network element information deletion
|
|
// @Description Network element information deletion
|
|
// @Router /ne/info [delete]
|
|
func (s NeInfoController) Remove(c *gin.Context) {
|
|
language := reqctx.AcceptLanguage(c)
|
|
id := c.Param("id")
|
|
if id == "" {
|
|
c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: id is empty"))
|
|
return
|
|
}
|
|
|
|
// 处理字符转id数组后去重
|
|
uniqueIDs := parse.RemoveDuplicatesToArray(id, ",")
|
|
// 转换成int64数组类型
|
|
ids := make([]int64, 0)
|
|
for _, v := range uniqueIDs {
|
|
ids = append(ids, parse.Number(v))
|
|
}
|
|
|
|
rows, err := s.neInfoService.DeleteByIds(ids)
|
|
if err != nil {
|
|
c.JSON(200, resp.ErrMsg(i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
msg := i18n.TTemplate(language, "app.common.deleteSuccess", map[string]any{"num": rows})
|
|
c.JSON(200, resp.OkMsg(msg))
|
|
}
|