329 lines
11 KiB
Go
329 lines
11 KiB
Go
package controller
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"be.ems/src/framework/i18n"
|
||
"be.ems/src/framework/reqctx"
|
||
"be.ems/src/framework/resp"
|
||
neFetchlink "be.ems/src/modules/network_element/fetch_link"
|
||
neService "be.ems/src/modules/network_element/service"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// 实例化控制层 PCFController 结构体
|
||
var NewPCF = &PCFController{
|
||
neInfoService: neService.NewNeInfo,
|
||
}
|
||
|
||
// 网元PCF
|
||
//
|
||
// PATH /pcf
|
||
type PCFController struct {
|
||
neInfoService *neService.NeInfo // 网元信息服务
|
||
}
|
||
|
||
// 策略配置列表
|
||
//
|
||
// GET /rule/list
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param neId query string true "NE ID" default(001)
|
||
// @Param imsi query string false "IMSI"
|
||
// @Param msisdn query string false "MSISDN"
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration List
|
||
// @Description Policy Configuration List
|
||
// @Router /neData/pcf/rule/list [get]
|
||
func (s PCFController) RuleInfoList(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var query struct {
|
||
NeId string `form:"neId" binding:"required"`
|
||
IMSI string `form:"imsi"`
|
||
MSISDN string `form:"msisdn"`
|
||
}
|
||
if err := c.ShouldBindQuery(&query); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", query.NeId)
|
||
if neInfo.NeId != query.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
data, err := neFetchlink.PCFRuleInfo(neInfo, map[string]string{
|
||
"imsi": query.IMSI,
|
||
"msisdn": query.MSISDN,
|
||
})
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
|
||
c.JSON(200, resp.OkData(data))
|
||
}
|
||
|
||
// 策略配置添加
|
||
//
|
||
// POST /rule
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param data body object true "Request Param"
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration Additions
|
||
// @Description Policy Configuration Additions
|
||
// @Router /neData/pcf/rule [post]
|
||
func (s PCFController) RuleInfoAdd(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var body struct {
|
||
NeId string `json:"neId" binding:"required"` // 网元ID
|
||
Num int64 `json:"num"` // 批量添加，默认0单条，大于1时imsi/msisdn会累加数值
|
||
ParamData map[string]any `json:"paramData" binding:"required"` // 参数数据
|
||
// Imsi string `json:"imsi" binding:"required"`
|
||
// Msisdn string `json:"msisdn" binding:"required"`
|
||
// Sar string `json:"sar"` // 根据PCF参数配置Service Area Restriction -> Name
|
||
// PccRules string `json:"pccRules"` // 根据PCF参数配置PCC Rules -> Rule ID
|
||
// QosAudio string `json:"qosAudio"` // 根据PCF参数配置QoS Template -> QoS ID
|
||
// QosVideo string `json:"qosVideo"` // 根据PCF参数配置QoS Template -> QoS ID
|
||
// SessRules string `json:"sessRules"` // 根据PCF参数配置Session Rules -> Rule ID
|
||
// HdrEnrich string `json:"hdrEnrich"` // 根据PCF参数配置Header Enrich Template -> Template Name
|
||
// UePolicy string `json:"uePolicy"` // UE策略模板(样例: uep_001)
|
||
// Rfsp int64 `json:"rfsp"` // 无线频率选择优先级
|
||
}
|
||
if err := c.ShouldBindBodyWithJSON(&body); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", body.NeId)
|
||
if neInfo.NeId != body.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
var err error
|
||
if body.Num > 0 { // 批量添加
|
||
err = neFetchlink.PCFRuleAddBatch(neInfo, body.ParamData, body.Num)
|
||
} else { // 单条添加
|
||
err = neFetchlink.PCFRuleAdd(neInfo, body.ParamData)
|
||
}
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
c.JSON(200, resp.Ok(nil))
|
||
}
|
||
|
||
// 策略配置更新
|
||
//
|
||
// PUT /rule
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param data body object true "Request Param"
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration Updates
|
||
// @Description Policy Configuration Updates
|
||
// @Router /neData/pcf/rule [put]
|
||
func (s PCFController) RuleInfoEdit(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var body struct {
|
||
NeId string `json:"neId" binding:"required"` // 网元ID
|
||
Num int64 `json:"num"` // 更新数量
|
||
ParamData map[string]any `json:"paramData" binding:"required"` // 参数数据
|
||
// Imsi string `json:"imsi" binding:"required"`
|
||
// Msisdn string `json:"msisdn" binding:"required"`
|
||
// Sar string `json:"sar"` // 根据PCF参数配置Service Area Restriction -> Name
|
||
// PccRules string `json:"pccRules"` // 根据PCF参数配置PCC Rules -> Rule ID
|
||
// QosAudio string `json:"qosAudio"` // 根据PCF参数配置QoS Template -> QoS ID
|
||
// QosVideo string `json:"qosVideo"` // 根据PCF参数配置QoS Template -> QoS ID
|
||
// SessRules string `json:"sessRules"` // 根据PCF参数配置Session Rules -> Rule ID
|
||
// HdrEnrich string `json:"hdrEnrich"` // 根据PCF参数配置Header Enrich Template -> Template Name
|
||
// UePolicy string `json:"uePolicy"` // UE策略模板(样例: uep_001)
|
||
// Rfsp int64 `json:"rfsp"` // 无线频率选择优先级
|
||
}
|
||
if err := c.ShouldBindBodyWithJSON(&body); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", body.NeId)
|
||
if neInfo.NeId != body.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
var err error
|
||
if body.Num > 0 { // 批量更新
|
||
err = neFetchlink.PCFRuleUpdateBatch(neInfo, body.ParamData, body.Num)
|
||
} else { // 单条更新
|
||
err = neFetchlink.PCFRuleUpdate(neInfo, body.ParamData)
|
||
}
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
c.JSON(200, resp.Ok(nil))
|
||
}
|
||
|
||
// 策略配置删除
|
||
//
|
||
// DELETE /rule
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param neId query string true "NE ID" default(001)
|
||
// @Param imsi query string true "IMSi, batch deletion with quantity"
|
||
// @Param num query number false "Number of deletions"
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration Deletion
|
||
// @Description Policy Configuration Deletion
|
||
// @Router /neData/pcf/rule [delete]
|
||
func (s PCFController) RuleInfoRemove(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var query struct {
|
||
NeId string `form:"neId" binding:"required"` // 网元ID
|
||
IMSI string `form:"imsi" binding:"required"` // IMSi, 带数量时为批量删除
|
||
Num int64 `form:"num"` // 删除数量
|
||
}
|
||
if err := c.ShouldBindQuery(&query); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", query.NeId)
|
||
if neInfo.NeId != query.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
var err error
|
||
if query.Num > 0 { // 批量删除
|
||
err = neFetchlink.PCFRuleDeleteBatch(neInfo, query.IMSI, query.Num)
|
||
} else { // 单条删除
|
||
err = neFetchlink.PCFRuleDelete(neInfo, query.IMSI)
|
||
}
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
c.JSON(200, resp.Ok(nil))
|
||
}
|
||
|
||
// 策略配置导出
|
||
//
|
||
// GET /rule/export
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param neId query string true "NE ID" default(001)
|
||
// @Param fileType query string true "File Type" default(txt)
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration Export
|
||
// @Description Policy Configuration Export
|
||
// @Router /neData/pcf/rule/export [get]
|
||
func (s PCFController) RuleInfoExport(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var query struct {
|
||
NeId string `form:"neId" binding:"required"`
|
||
FileType string `form:"fileType" binding:"required,oneof=txt"` // 文件类型
|
||
}
|
||
if err := c.ShouldBindQuery(&query); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", query.NeId)
|
||
if neInfo.NeId != query.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
data, err := neFetchlink.PCFRuleExport(neInfo, map[string]string{
|
||
"fileType": query.FileType,
|
||
})
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
c.Writer.Header().Set("Content-Disposition", `attachment; filename="pcf_rule_export.txt"`)
|
||
c.Data(200, "application/octet-stream", data)
|
||
}
|
||
|
||
// 策略配置导入
|
||
//
|
||
// PUT /rule/import
|
||
//
|
||
// @Tags network_data/pcf
|
||
// @Accept json
|
||
// @Produce json
|
||
// @Param neId query string true "NE ID" default(001)
|
||
// @Param fileType query string true "File Type" default(txt)
|
||
// @Param filePath query string true "File Path" default(/tmp/pcfuser.txt)
|
||
// @Success 200 {object} object "Response Results"
|
||
// @Security TokenAuth
|
||
// @Summary Policy Configuration Import
|
||
// @Description Policy Configuration Import
|
||
// @Router /neData/pcf/rule/import [put]
|
||
func (s PCFController) RuleInfoImport(c *gin.Context) {
|
||
language := reqctx.AcceptLanguage(c)
|
||
var body struct {
|
||
NeId string `json:"neId" binding:"required"`
|
||
FileType string `json:"fileType" binding:"required,oneof=txt"` // 文件类型
|
||
FilePath string `json:"filePath" binding:"required"` // 网元端文件所在绝对路径
|
||
}
|
||
if err := c.ShouldBindBodyWithJSON(&body); err != nil {
|
||
errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
|
||
c.JSON(422, resp.CodeMsg(422001, errMsgs))
|
||
return
|
||
}
|
||
|
||
// 查询网元信息
|
||
neInfo := s.neInfoService.FindByNeTypeAndNeID("PCF", body.NeId)
|
||
if neInfo.NeId != body.NeId || neInfo.IP == "" {
|
||
c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
|
||
return
|
||
}
|
||
|
||
// 网元直连
|
||
output, err := neFetchlink.PCFRuleImport(neInfo, map[string]any{
|
||
"type": body.FileType,
|
||
"filePath": body.FilePath,
|
||
})
|
||
if err != nil {
|
||
c.JSON(200, resp.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
c.JSON(200, resp.OkMsg(output))
|
||
}
|