feat: 跟踪任务 网元HLR功能接口

src/modules/trace/controller/trace_task_hlr.go

@@ -0,0 +1,178 @@
+package controller
+
+import (
+	"strings"
+
+	"be.ems/src/framework/i18n"
+	"be.ems/src/framework/utils/ctx"
+	"be.ems/src/framework/utils/parse"
+	"be.ems/src/framework/vo/result"
+	neService "be.ems/src/modules/network_element/service"
+	"be.ems/src/modules/trace/model"
+	traceService "be.ems/src/modules/trace/service"
+	"github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin/binding"
+)
+
+// 实例化控制层 TraceTaskHlrController 结构体
+var NewTraceTaskHlr = &TraceTaskHlrController{
+	neInfoService:       neService.NewNeInfoImpl,
+	traceTaskHlrService: traceService.NewTraceTaskHlrImpl,
+}
+
+// 网元SMF
+//
+// PATH /task/hlr
+type TraceTaskHlrController struct {
+	// 网元信息服务
+	neInfoService neService.INeInfo
+	// 跟踪_任务给HRL网元信息服务
+	traceTaskHlrService traceService.ITraceTaskHlr
+}
+
+// 跟踪任务列表
+//
+// GET /list
+func (s *TraceTaskHlrController) List(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var querys model.TraceTaskHlrQuery
+	if err := c.ShouldBindQuery(&querys); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询数据
+	data := s.traceTaskHlrService.SelectPage(querys)
+	c.JSON(200, result.Ok(data))
+}
+
+// 跟踪任务删除
+//
+// DELETE /:ids
+func (s *TraceTaskHlrController) Remove(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	rowIds := c.Param("ids")
+	if rowIds == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+	// 处理字符转id数组后去重
+	ids := strings.Split(rowIds, ",")
+	uniqueIDs := parse.RemoveDuplicates(ids)
+	if len(uniqueIDs) <= 0 {
+		c.JSON(200, result.Err(nil))
+		return
+	}
+	rows, err := s.traceTaskHlrService.DeleteByIds(uniqueIDs)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
+		return
+	}
+	msg := i18n.TTemplate(language, "app.common.deleteSuccess", map[string]any{"num": rows})
+	c.JSON(200, result.OkMsg(msg))
+}
+
+// 跟踪任务创建
+//
+// POST /start
+func (s *TraceTaskHlrController) Start(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body struct {
+		IMSI      string `json:"imsi"`      // IMSI
+		MSISDN    string `json:"msisdn"`    // MSISDN
+		StartTime int64  `json:"startTime"` // 开始时间
+		EndTime   int64  `json:"endTime"`   // 结束时间
+		Remark    string `json:"remark"`    // 备注说明
+	}
+	if err := c.ShouldBindBodyWith(&body, binding.JSON); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	if body.IMSI == "" && body.MSISDN == "" {
+		c.JSON(400, result.CodeMsg(400, "imsi amd msisdn is empty"))
+		return
+	}
+
+	task := model.TraceTaskHlr{
+		IMSI:      body.IMSI,
+		MSISDN:    body.MSISDN,
+		StartTime: body.StartTime,
+		EndTime:   body.EndTime,
+		Remark:    body.Remark,
+		CreateBy:  ctx.LoginUserToUserName(c),
+	}
+	id, err := s.traceTaskHlrService.Start(task)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	c.JSON(200, result.OkData(id))
+}
+
+// 跟踪任务停止
+//
+// POST /stop
+func (s *TraceTaskHlrController) Stop(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body struct {
+		ID string `json:"id"` // 任务ID
+	}
+	if err := c.ShouldBindBodyWith(&body, binding.JSON); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 处理字符转id数组后去重
+	ids := strings.Split(body.ID, ",")
+	uniqueIDs := parse.RemoveDuplicates(ids)
+	if len(uniqueIDs) <= 0 {
+		c.JSON(200, result.Err(nil))
+		return
+	}
+
+	errArr := []map[string]any{}
+	for _, id := range uniqueIDs {
+		task := s.traceTaskHlrService.SelectById(id)
+		if task.ID != id || task.ID == "" {
+			errArr = append(errArr, map[string]any{"id": id, "err": "task not found"})
+			continue
+		}
+
+		task.UpdateBy = ctx.LoginUserToUserName(c)
+		err := s.traceTaskHlrService.Stop(task)
+		if err != nil {
+			errArr = append(errArr, map[string]any{"id": id, "err": err.Error()})
+			continue
+		}
+	}
+	c.JSON(200, result.OkData(errArr))
+}
+
+// 跟踪任务文件
+//
+// POST /file
+func (s *TraceTaskHlrController) File(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body struct {
+		ID string `json:"id"` // 任务ID
+	}
+	if err := c.ShouldBindBodyWith(&body, binding.JSON); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	task := s.traceTaskHlrService.SelectById(body.ID)
+	if task.ID != body.ID || task.ID == "" {
+		c.JSON(200, result.CodeMsg(400, "task not found"))
+		return
+	}
+
+	task.UpdateBy = ctx.LoginUserToUserName(c)
+	list, err := s.traceTaskHlrService.File(task)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	c.JSON(200, result.OkData(list))
+}