perf: 重构接收UE接入功能

src/modules/network_data/controller/mme.go

@@ -8,7 +8,6 @@ import (
 	"be.ems/src/framework/reqctx"
 	"be.ems/src/framework/resp"
 	"be.ems/src/framework/utils/parse"
-	"be.ems/src/modules/network_data/model"
 	neDataService "be.ems/src/modules/network_data/service"
 	neFetchlink "be.ems/src/modules/network_element/fetch_link"
 	neService "be.ems/src/modules/network_element/service"
@@ -19,15 +18,15 @@ import (
 // 实例化控制层 MMEController 结构体
 var NewMME = &MMEController{
 	neInfoService:  neService.NewNeInfo,
-	ueEventService: neDataService.NewUEEventMME,
+	ueEventService: neDataService.NewUEEvent,
 }
 
 // 网元MME
 //
 // PATH /mme
 type MMEController struct {
-	neInfoService  *neService.NeInfo         // 网元信息服务
-	ueEventService *neDataService.UEEventMME // UE会话事件服务
+	neInfoService  *neService.NeInfo      // 网元信息服务
+	ueEventService *neDataService.UEEvent // UE会话事件服务
 }
 
 // UE会话列表
@@ -49,23 +48,20 @@ type MMEController struct {
 //	@Router			/neData/mme/ue/list [get]
 func (s *MMEController) UEList(c *gin.Context) {
 	language := reqctx.AcceptLanguage(c)
-	var querys model.UEEventMMEQuery
-	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
+	query := reqctx.QueryMap(c)
+	// 限制导出数据集
+	pageSize := parse.Number(query["pageSize"])
+	if pageSize > 10000 {
+		query["pageSize"] = "10000"
 	}
-
-	// 查询网元获取IP
-	neInfo := s.neInfoService.FindByNeTypeAndNeID("MME", querys.NeID)
-	if neInfo.NeId != querys.NeID || neInfo.IP == "" {
+	// 查询网元信息 rmUID
+	neInfo := s.neInfoService.FindByNeTypeAndNeID(query["neType"], query["neId"])
+	if neInfo.NeType == "" {
 		c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
 		return
 	}
-	querys.RmUID = neInfo.RmUID
-
-	// 查询数据
-	rows, total := s.ueEventService.FindByPage(querys)
+	query["rmUID"] = neInfo.RmUID
+	rows, total := s.ueEventService.FindByPage(neInfo.NeType, query)
 	c.JSON(200, resp.OkData(map[string]any{"rows": rows, "total": total}))
 }
 
@@ -98,7 +94,7 @@ func (s *MMEController) UERemove(c *gin.Context) {
 		ids = append(ids, parse.Number(v))
 	}
 
-	rows, err := s.ueEventService.DeleteByIds(ids)
+	rows, err := s.ueEventService.DeleteByIds("MME", ids)
 	if err != nil {
 		c.JSON(200, resp.ErrMsg(i18n.TKey(language, err.Error())))
 		return
@@ -109,7 +105,7 @@ func (s *MMEController) UERemove(c *gin.Context) {
 
 // UE会话列表导出
 //
-// POST /ue/export
+// GET /ue/export
 //
 //	@Tags			network_data/mme
 //	@Accept			json
@@ -119,28 +115,23 @@ func (s *MMEController) UERemove(c *gin.Context) {
 //	@Security		TokenAuth
 //	@Summary		UE Session List Export
 //	@Description	UE Session List Export
-//	@Router			/neData/mme/ue/export [post]
+//	@Router			/neData/mme/ue/export [get]
 func (s *MMEController) UEExport(c *gin.Context) {
 	language := reqctx.AcceptLanguage(c)
-	// 查询结果，根据查询条件结果，单页最大值限制
-	var querys model.UEEventMMEQuery
-	if err := c.ShouldBindBodyWithJSON(&querys); err != nil {
-		errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err))
-		c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs))
-		return
-	}
+	query := reqctx.QueryMap(c)
 	// 限制导出数据集
-	if querys.PageSize > 10000 {
-		querys.PageSize = 10000
+	pageSize := parse.Number(query["pageSize"])
+	if pageSize > 10000 {
+		query["pageSize"] = "10000"
 	}
-	// 查询网元获取IP
-	neInfo := s.neInfoService.FindByNeTypeAndNeID("MME", querys.NeID)
-	if neInfo.NeId != querys.NeID || neInfo.IP == "" {
+	// 查询网元信息 rmUID
+	neInfo := s.neInfoService.FindByNeTypeAndNeID(query["neType"], query["neId"])
+	if neInfo.NeType == "" {
 		c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
 		return
 	}
-	querys.RmUID = neInfo.RmUID
-	rows, total := s.ueEventService.FindByPage(querys)
+	query["rmUID"] = neInfo.RmUID
+	rows, total := s.ueEventService.FindByPage(neInfo.NeType, query)
 	if total == 0 {
 		// 导出数据记录为空
 		c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.exportEmpty")))
@@ -150,7 +141,7 @@ func (s *MMEController) UEExport(c *gin.Context) {
 	// 导出文件名称
 	fileName := fmt.Sprintf("mme_ue_event_export_%d_%d.xlsx", len(rows), time.Now().UnixMilli())
 	// 导出数据表格
-	saveFilePath, err := s.ueEventService.ExportXlsx(rows, fileName, language)
+	saveFilePath, err := s.ueEventService.ExportMME(rows, fileName, language)
 	if err != nil {
 		c.JSON(200, resp.ErrMsg(err.Error()))
 		return