feat: 添加SMF在线订阅用户列表

src/modules/network_data/controller/smf.go

@@ -15,6 +15,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"
 	"github.com/gin-gonic/gin"
 	"github.com/gin-gonic/gin/binding"
@@ -22,8 +23,9 @@ import (
 
 // 实例化控制层 SMFController 结构体
 var NewSMFController = &SMFController{
-	neInfoService:   neService.NewNeInfoImpl,
-	cdrEventService: neDataService.NewCDREventSMFImpl,
+	neInfoService:      neService.NewNeInfoImpl,
+	cdrEventService:    neDataService.NewCDREventSMFImpl,
+	udmUserInfoService: *neDataService.NewUDMUserInfo,
 }
 
 // 网元SMF
@@ -34,6 +36,8 @@ type SMFController struct {
 	neInfoService neService.INeInfo
 	// CDR会话事件服务
 	cdrEventService neDataService.ICDREventSMF
+	// UDM用户信息服务
+	udmUserInfoService neDataService.UDMUserInfo
 }
 
 // CDR会话列表
@@ -263,3 +267,63 @@ PDU IPv6 Addres Swith Prefix: %s`, User_Identifier, SSC_Mode, RAT_Type, DNN_ID,
 
 	c.FileAttachment(saveFilePath, fileName)
 }
+
+// 在线订阅用户列表信息
+//
+// GET /subscribers
+func (s *SMFController) SubUserList(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var query struct {
+		NeId    string `form:"neId" binding:"required"`
+		IMSI    string `form:"imsi"`
+		MSISDN  string `form:"msisdn"`
+		Upstate string `form:"upstate"`
+		PageNum string `form:"pageNum"`
+	}
+	if err := c.ShouldBindQuery(&query); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元信息 rmUID
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("SMF", query.NeId)
+	if neInfo.NeId != query.NeId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	// 网元直连
+	data, err := neFetchlink.SMFSubInfo(neInfo, map[string]string{
+		"imsi":    query.IMSI,
+		"msisdn":  query.MSISDN,
+		"upstate": query.Upstate,
+		"pageNum": query.PageNum,
+	})
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 对数据进行处理，去掉前缀，并加入imsi拓展信息
+	rows := data["rows"].([]any)
+	arr := &rows
+	for i := range *arr {
+		item := (*arr)[i].(map[string]any)
+		if v, ok := item["imsi"]; ok && v != nil {
+			imsiStr := v.(string)
+			imsiStr = strings.TrimPrefix(imsiStr, "imsi-")
+			item["imsi"] = imsiStr
+			// 查UDM拓展信息
+			info := s.udmUserInfoService.SelectByIMSIAndNeID(imsiStr, "")
+			item["remark"] = info.Remark
+		}
+		if v, ok := item["msisdn"]; ok && v != nil {
+			item["msisdn"] = strings.TrimPrefix(v.(string), "msisdn-")
+		}
+	}
+
+	c.JSON(200, result.Ok(map[string]any{
+		"total": data["total"],
+		"rows":  data["rows"],
+	}))
+}