feat: 网元直连接口补充AMF/SMF/MME/IMS

src/modules/network_data/controller/ims.go

@@ -16,6 +16,7 @@ import (
 	"be.ems/src/framework/vo/result"
 	"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"
 	sysService "be.ems/src/modules/system/service"
 	"github.com/gin-gonic/gin"
@@ -215,3 +216,68 @@ func (s *IMSController) CDRExport(c *gin.Context) {
 
 	c.FileAttachment(saveFilePath, fileName)
 }
+
+// 在线会话用户数量
+//
+// GET /session/num
+func (s *IMSController) UeSessionNum(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var query struct {
+		NeId string `form:"neId" binding:"required"`
+	}
+	if err := c.ShouldBindQuery(&query); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元信息
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("IMS", query.NeId)
+	if neInfo.NeId != query.NeId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	// 网元直连
+	num, err := neFetchlink.IMSUeSessionNum(neInfo)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	c.JSON(200, result.OkData(num))
+}
+
+// 在线会话用户列表信息
+//
+// GET /session/list
+func (s *IMSController) UeSessionList(c *gin.Context) {
+	language := ctx.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 {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元信息
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("IMS", query.NeId)
+	if neInfo.NeId != query.NeId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	// 网元直连
+	data, err := neFetchlink.IMSUeSessionList(neInfo, map[string]string{
+		"imsi":   query.IMSI,
+		"msisdn": query.MSISDN,
+	})
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	c.JSON(200, result.OkData(data))
+}