feat: 网元数据模块添加UDM鉴权和签约

src/modules/network_data/controller/udm_sub.go

@@ -0,0 +1,603 @@
+package controller
+
+import (
+	"fmt"
+	"path/filepath"
+	"strconv"
+	"strings"
+	"time"
+
+	"be.ems/src/framework/constants/uploadsubpath"
+	"be.ems/src/framework/i18n"
+	"be.ems/src/framework/utils/ctx"
+	"be.ems/src/framework/utils/file"
+	"be.ems/src/framework/utils/parse"
+	"be.ems/src/framework/utils/telnet"
+	"be.ems/src/framework/vo/result"
+	"be.ems/src/modules/network_data/model"
+	neDataService "be.ems/src/modules/network_data/service"
+	neService "be.ems/src/modules/network_element/service"
+	"github.com/gin-gonic/gin"
+	"github.com/gin-gonic/gin/binding"
+)
+
+// 实例化控制层 UDMSubController 结构体
+var NewUDMSub = &UDMSubController{
+	udmSubService: neDataService.NewUDMSubImpl,
+	neInfoService: neService.NewNeInfoImpl,
+}
+
+// UDM签约用户
+//
+// PATH /udm/sub
+type UDMSubController struct {
+	// UDM签约信息服务
+	udmSubService neDataService.IUDMSub
+	// 网元信息服务
+	neInfoService neService.INeInfo
+}
+
+// UDM签约用户重载数据
+//
+// POST /resetData/:neId
+func (s *UDMSubController) ResetData(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	if neId == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	neId = ""
+	data := s.udmSubService.ResetData(neId)
+	c.JSON(200, result.OkData(data))
+}
+
+// UDM签约用户列表
+//
+// GET /list
+func (s *UDMSubController) List(c *gin.Context) {
+	querys := ctx.QueryMap(c)
+	querys["neId"] = ""
+	data := s.udmSubService.SelectPage(querys)
+	c.JSON(200, result.Ok(data))
+}
+
+// UDM签约用户信息
+//
+// GET /:neId/:imsi
+func (s *UDMSubController) Info(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	imsi := c.Param("imsi")
+	if neId == "" || imsi == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("dsp udmuser:imsi=%s", imsi)
+	data, err := telnet.ConvertToMap(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	if len(data) == 0 {
+		c.JSON(200, result.ErrMsg("No Subs Data"))
+		return
+	}
+
+	// 解析返回的数据
+	cnType, _ := strconv.ParseInt(data["CNType"][:4], 0, 64)
+	rat, _ := strconv.ParseInt(data["RAT"][:4], 0, 64)
+	msisdn := data["MSISDN"]
+	imsMsisdnLen := strings.Index(msisdn, ",")
+	if imsMsisdnLen != -1 {
+		msisdn = msisdn[:imsMsisdnLen]
+	}
+	neId = ""
+	u := model.UDMSub{
+		IMSI:   imsi,
+		Msisdn: msisdn,
+		Ambr:   data["AMBR"],
+		Arfb:   data["AreaForbidden"],
+		Cn:     fmt.Sprint(cnType),
+		SmData: data["SM-Data(snssai+dnn[1..n])"],
+		Sar:    data["ServiceAreaRestriction"],
+		Nssai:  data["NSSAI"],
+		SmfSel: data["Smf-Selection"],
+		Rat:    fmt.Sprint(rat),
+		NeId:   neId,
+	}
+	// 1,64,24,65,def_eps,1,2,010200000000,-
+	if v, ok := data["EPS-Data"]; ok {
+		u.EpsDat = v
+		arr := strings.Split(v, ",")
+		u.EpsFlag = arr[0]
+		u.EpsOdb = arr[1]
+		u.HplmnOdb = arr[2]
+		u.Ard = arr[3]
+		u.Epstpl = arr[4]
+		u.ContextId = arr[5]
+		u.ApnContext = arr[7]
+		// [6] 是不要的，导入和导出不用
+		u.StaticIp = arr[8]
+	}
+
+	// 查询imsi存在赋予id用于更新
+	list := s.udmSubService.SelectList(u)
+	if len(list) > 0 {
+		item := list[0]
+		if item.ID != "" {
+			u.ID = item.ID
+		}
+	}
+	go s.udmSubService.Insert(neId, u)
+
+	c.JSON(200, result.OkData(u))
+}
+
+// UDM签约用户新增
+//
+// POST /:neId
+func (s *UDMSubController) Add(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	if neId == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	var body model.UDMSub
+	err := c.ShouldBindBodyWith(&body, binding.JSON)
+	if err != nil || body.IMSI == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("add udmuser:imsi=%s,msisdn=%s,ambr=%s,nssai=%s,arfb=%s,sar=%s,rat=%s,cn=%s,smf_sel=%s,sm_data=%s,eps_flag=%s,eps_odb=%s,hplmn_odb=%s,ard=%s,epstpl=%s,context_id=%s,apn_context=%s",
+		body.IMSI, body.Msisdn, body.Ambr, body.Nssai, body.Arfb, body.Sar, body.Rat, body.Cn, body.SmfSel, body.SmData, body.EpsFlag, body.EpsOdb, body.HplmnOdb, body.Ard, body.Epstpl, body.ContextId, body.ApnContext)
+	// static_ip指给4G UE分配的静态IP，没有可不带此字段名，批量添加IP会自动递增
+	if body.StaticIp != "" {
+		cmd += fmt.Sprintf(",static_ip=%s", body.StaticIp)
+	}
+	data, err := telnet.ConvertToStr(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 命令ok时
+	if strings.Contains(data, "ok") {
+		neId = ""
+		go s.udmSubService.Insert(neId, body)
+	}
+	c.JSON(200, result.OkData(data))
+}
+
+// UDM签约用户批量新增
+//
+// POST /:neId/:num
+func (s *UDMSubController) Adds(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	num := c.Param("num")
+	if neId == "" || num == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	var body model.UDMSub
+	err := c.ShouldBindBodyWith(&body, binding.JSON)
+	if err != nil || body.IMSI == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("baa udmuser:start_imsi=%s,start_msisdn=%s,sub_num=%s,ambr=%s,nssai=%s,arfb=%s,sar=%s,rat=%s,cn=%s,smf_sel=%s,sm_data=%s,eps_flag=%s,eps_odb=%s,hplmn_odb=%s,ard=%s,epstpl=%s,context_id=%s,apn_context=%s",
+		body.IMSI, body.Msisdn, num, body.Ambr, body.Nssai, body.Arfb, body.Sar, body.Rat, body.Cn, body.SmfSel, body.SmData, body.EpsFlag, body.EpsOdb, body.HplmnOdb, body.Ard, body.Epstpl, body.ContextId, body.ApnContext)
+	// static_ip指给4G UE分配的静态IP，没有可不带此字段名，批量添加IP会自动递增
+	if body.StaticIp != "" {
+		cmd += fmt.Sprintf(",static_ip=%s", body.StaticIp)
+	}
+	data, err := telnet.ConvertToStr(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 命令ok时
+	if strings.Contains(data, "ok") {
+		neId = ""
+		go s.udmSubService.LoadData(neId, body.IMSI, num)
+	}
+	c.JSON(200, result.OkData(data))
+}
+
+// UDM签约用户修改
+//
+// PUT /:neId
+func (s *UDMSubController) Edit(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	if neId == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	var body model.UDMSub
+	err := c.ShouldBindBodyWith(&body, binding.JSON)
+	if err != nil || body.IMSI == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("mod udmuser:imsi=%s", body.IMSI)
+	// 修改的参数名称
+	if body.Msisdn != "" {
+		cmd += fmt.Sprintf(",msisdn=%s", body.Msisdn)
+	}
+	if body.Ambr != "" {
+		cmd += fmt.Sprintf(",ambr=%s", body.Ambr)
+	}
+	if body.Nssai != "" {
+		cmd += fmt.Sprintf(",nssai=%s", body.Nssai)
+	}
+	if body.Arfb != "" {
+		cmd += fmt.Sprintf(",arfb=%s", body.Arfb)
+	}
+	if body.Sar != "" {
+		cmd += fmt.Sprintf(",sar=%s", body.Sar)
+	}
+	if body.Rat != "" {
+		cmd += fmt.Sprintf(",rat=%s", body.Rat)
+	}
+	if body.Cn != "" {
+		cmd += fmt.Sprintf(",cn=%s", body.Cn)
+	}
+	if body.SmfSel != "" {
+		cmd += fmt.Sprintf(",smf_sel=%s", body.SmfSel)
+	}
+	if body.SmData != "" {
+		cmd += fmt.Sprintf(",sm_data=%s", body.SmData)
+	}
+	if body.EpsDat != "" {
+		cmd += fmt.Sprintf(",eps_dat=%s", body.EpsDat)
+	}
+	if body.EpsFlag != "" {
+		cmd += fmt.Sprintf(",eps_flag=%s", body.EpsFlag)
+	}
+	if body.EpsOdb != "" {
+		cmd += fmt.Sprintf(",eps_odb=%s", body.EpsOdb)
+	}
+	if body.HplmnOdb != "" {
+		cmd += fmt.Sprintf(",hplmn_odb=%s", body.HplmnOdb)
+	}
+	if body.Epstpl != "" {
+		cmd += fmt.Sprintf(",epstpl=%s", body.Epstpl)
+	}
+	if body.Ard != "" {
+		cmd += fmt.Sprintf(",ard=%s", body.Ard)
+	}
+	if body.ContextId != "" {
+		cmd += fmt.Sprintf(",context_id=%s", body.ContextId)
+	}
+	if body.ApnContext != "" {
+		cmd += fmt.Sprintf(",apn_context=%s", body.ApnContext)
+	}
+	if body.StaticIp != "" {
+		cmd += fmt.Sprintf(",static_ip=%s", body.StaticIp)
+	}
+	data, err := telnet.ConvertToStr(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 命令ok时
+	if strings.Contains(data, "ok") {
+		neId = ""
+		go s.udmSubService.Insert(neId, body)
+	}
+	c.JSON(200, result.OkData(data))
+}
+
+// UDM签约用户删除
+//
+// DELETE /:neId/:imsi
+func (s *UDMSubController) Remove(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	imsi := c.Param("imsi")
+	if neId == "" || imsi == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 处理字符转id数组后去重
+	imsiArr := strings.Split(imsi, ",")
+	uniqueIDs := parse.RemoveDuplicates(imsiArr)
+	if len(uniqueIDs) <= 0 {
+		c.JSON(200, result.Err(nil))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	resultData := map[string]string{}
+	for _, imsi := range uniqueIDs {
+		// 发送MML
+		cmd := fmt.Sprintf("del udmuser:imsi=%s", imsi)
+		data, err := telnet.ConvertToStr(telnetClient, cmd)
+		if err != nil {
+			resultData[imsi] = err.Error()
+			continue
+		}
+		// 命令ok时
+		if strings.Contains(data, "ok") {
+			neId = ""
+			go s.udmSubService.Delete(neId, imsi)
+		}
+		resultData[imsi] = data
+	}
+
+	c.JSON(200, result.OkData(resultData))
+}
+
+// UDM签约用户批量删除
+//
+// DELETE /:neId/:imsi/:num
+func (s *UDMSubController) Removes(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	neId := c.Param("neId")
+	imsi := c.Param("imsi")
+	num := c.Param("num")
+	if neId == "" || imsi == "" || num == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", neId)
+	if neInfo.NeId != neId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("bde udmuser:start_imsi=%s,sub_num=%s", imsi, num)
+	data, err := telnet.ConvertToStr(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 命令ok时
+	if strings.Contains(data, "ok") {
+		neId = ""
+		go s.udmSubService.LoadData(neId, imsi, num)
+	}
+	c.JSON(200, result.OkData(data))
+}
+
+// UDM签约用户导出
+//
+// POST /export
+func (s *UDMSubController) Export(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body struct {
+		NeId string `json:"neId" binding:"required"`
+		Type string `json:"type" binding:"required"`
+	}
+	err := c.ShouldBindBodyWith(&body, binding.JSON)
+	if err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	if !(body.Type == "csv" || body.Type == "txt") {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "ne.udm.errImportUserSubFileFormat")))
+		return
+	}
+
+	neId := ""
+	list := s.udmSubService.SelectList(model.UDMSub{NeId: neId})
+	// 文件名
+	fileName := fmt.Sprintf("udm_sub_user_export_%s_%d.%s", neId, time.Now().UnixMilli(), body.Type)
+	filePath := fmt.Sprintf("%s/%s", file.ParseUploadFileDir(uploadsubpath.EXPORT), fileName)
+
+	if body.Type == "csv" {
+		// 转换数据
+		data := [][]string{}
+		data = append(data, []string{"imsi", "msisdn", "ambr", "nssai", "arfb", "sar", "rat", "cn", "smf_sel", "sm_dat", "eps_dat"})
+		for _, v := range list {
+			epsDat := fmt.Sprintf("%s,%s,%s,%s,%s,%s,%s,%s", v.EpsFlag, v.EpsOdb, v.HplmnOdb, v.Ard, v.Epstpl, v.ContextId, v.ApnContext, v.StaticIp)
+			data = append(data, []string{v.IMSI, v.Msisdn, v.Ambr, v.Nssai, v.Arfb, v.Sar, v.Rat, v.Cn, v.SmfSel, v.SmData, epsDat})
+		}
+		// 输出到文件
+		err = file.WriterFileCSV(data, filePath)
+		if err != nil {
+			c.JSON(200, result.ErrMsg(err.Error()))
+			return
+		}
+	}
+
+	if body.Type == "txt" {
+		// 转换数据
+		data := [][]string{}
+		for _, v := range list {
+			epsDat := fmt.Sprintf("%s,%s,%s,%s,%s,%s,%s,%s", v.EpsFlag, v.EpsOdb, v.HplmnOdb, v.Ard, v.Epstpl, v.ContextId, v.ApnContext, v.StaticIp)
+			data = append(data, []string{v.IMSI, v.Msisdn, v.Ambr, v.Nssai, v.Arfb, v.Sar, v.Rat, v.Cn, v.SmfSel, v.SmData, epsDat})
+		}
+		// 输出到文件
+		err = file.WriterFileTXT(data, ",", filePath)
+		if err != nil {
+			c.JSON(200, result.ErrMsg(err.Error()))
+			return
+		}
+	}
+
+	c.FileAttachment(filePath, fileName)
+}
+
+// UDM签约用户导入
+//
+// POST /import
+func (s *UDMSubController) Import(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body struct {
+		NeId       string `json:"neId" binding:"required"`
+		UploadPath string `json:"uploadPath" binding:"required"`
+	}
+	if err := c.ShouldBindBodyWith(&body, binding.JSON); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 判断文件名
+	if !(strings.HasSuffix(body.UploadPath, ".csv") || strings.HasSuffix(body.UploadPath, ".txt")) {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "ne.udm.errImportUserSubFileFormat")))
+		return
+	}
+
+	// 查询网元获取IP
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UDM", body.NeId)
+	if neInfo.NeId != body.NeId || neInfo.IP == "" {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
+		return
+	}
+
+	// 网元主机的SSH客户端
+	sshClient, err := s.neInfoService.NeRunSSHClient(neInfo.NeType, neInfo.NeId)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sshClient.Close()
+	// 网元主机的SSH客户端进行文件传输
+	sftpClient, err := sshClient.NewClientSFTP()
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sftpClient.Close()
+
+	// 本地文件
+	localFilePath := file.ParseUploadFilePath(body.UploadPath)
+	neFilePath := fmt.Sprintf("/tmp/%s", filepath.Base(localFilePath))
+	// 复制到远程
+	if err = sftpClient.CopyFileLocalToRemote(localFilePath, neFilePath); err != nil {
+		c.JSON(200, result.ErrMsg("error uploading file"))
+		return
+	}
+
+	// 网元主机的Telnet客户端
+	telnetClient, err := s.neInfoService.NeRunTelnetClient(neInfo.NeType, neInfo.NeId, 1)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer telnetClient.Close()
+
+	// 发送MML
+	cmd := fmt.Sprintf("import udmuser:path=%s", neFilePath)
+	data, err := telnet.ConvertToStr(telnetClient, cmd)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 命令ok时
+	if strings.Contains(data, "ok") {
+		if strings.HasSuffix(body.UploadPath, ".csv") {
+			data := file.ReadFileCSV(localFilePath)
+			neId := ""
+			go s.udmSubService.InsertData(neId, "csv", data)
+		}
+		if strings.HasSuffix(body.UploadPath, ".txt") {
+			data := file.ReadFileTXT(",", localFilePath)
+			neId := ""
+			go s.udmSubService.InsertData(neId, "txt", data)
+		}
+	}
+	c.JSON(200, result.OkMsg(data))
+}