package controller import ( "fmt" "path/filepath" "strings" "time" "be.ems/src/framework/constants" "be.ems/src/framework/i18n" "be.ems/src/framework/reqctx" "be.ems/src/framework/resp" "be.ems/src/framework/telnet" "be.ems/src/framework/utils/file" "be.ems/src/framework/utils/parse" "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" ) // 实例化控制层 UDMSubController 结构体 var NewUDMSub = &UDMSubController{ udmSubService: neDataService.NewUDMSubUser, neInfoService: neService.NewNeInfo, } // UDM签约用户 // // PATH /udm/sub type UDMSubController struct { udmSubService *neDataService.UDMSubUser // UDM签约信息服务 neInfoService *neService.NeInfo // 网元信息服务 } // UDM签约用户重载数据 // // PUT /resetData/:neId // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Reload Data // @Description UDM Subscriber User Reload Data // @Router /neData/udm/sub/resetData/{neId} [put] func (s *UDMSubController) ResetData(c *gin.Context) { neId := c.Param("neId") if neId == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId is empty")) return } data := s.udmSubService.ResetData(neId) c.JSON(200, resp.OkData(data)) } // UDM签约用户列表 // // GET /list // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId query string true "NE ID" default(001) // @Param imsi query string false "IMSI" // @Param msisdn query string false "Msisdn" // @Param pageNum query number true "pageNum" default(1) // @Param pageSize query number true "pageSize" default(10) // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User List // @Description UDM Subscriber User List // @Router /neData/udm/sub/list [get] func (s *UDMSubController) List(c *gin.Context) { query := reqctx.QueryMap(c) rows, total := s.udmSubService.FindByPage(query) c.JSON(200, resp.OkData(map[string]any{"rows": rows, "total": total})) } // UDM签约用户信息 // // GET /:neId/:imsi // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param value path string true "IMSI" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Information // @Description UDM Subscriber User Information // @Router /neData/udm/sub/{neId}/{value} [get] func (s *UDMSubController) Info(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") imsi := c.Param("imsi") if neId == "" || imsi == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId or imsi is empty")) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.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, resp.ErrMsg(err.Error())) return } if len(data) == 0 { c.JSON(200, resp.ErrMsg("No Subs Data")) return } // 解析返回的数据 u := s.udmSubService.ParseInfo(imsi, neId, data) s.udmSubService.Insert(neId, u) c.JSON(200, resp.OkData(u)) } // UDM签约用户新增 // // POST /:neId // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param data body object true "Request Param" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Added // @Description UDM Subscriber User Added // @Router /neData/udm/sub/{neId} [post] func (s *UDMSubController) Add(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") if neId == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId is empty")) return } var body model.UDMSubUser if err := c.ShouldBindBodyWithJSON(&body); err != nil { errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err)) c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs)) return } if len(body.IMSI) != 15 { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: IMSI length is not 15 bits")) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } defer telnetClient.Close() // 发送MML cmd := fmt.Sprintf("add udmuser:imsi=%s,", body.IMSI) cmd += s.udmSubService.ParseCommandParams(body) data, err := telnet.ConvertToStr(telnetClient, cmd) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } // 命令ok时 if strings.Contains(data, "ok") { body.NeId = neId s.udmSubService.Insert(neId, body) } c.JSON(200, resp.OkData(data)) } // UDM签约用户批量新增 // // POST /:neId/:num // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param value path number true "Number of releases, value includes start imsi" default(1) // @Param data body object true "Request Param" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Batch Add // @Description UDM Subscriber User Batch Add // @Router /neData/udm/sub/{neId}/{value} [post] func (s *UDMSubController) Adds(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") num := c.Param("num") if neId == "" || num == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId/num is empty")) return } var body model.UDMSubUser if err := c.ShouldBindBodyWithJSON(&body); err != nil { errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err)) c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs)) return } if len(body.IMSI) != 15 { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: IMSI length is not 15 bits")) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } defer telnetClient.Close() // 发送MML cmd := fmt.Sprintf("baa udmuser:start_imsi=%s,start_msisdn=%s,sub_num=%s,", body.IMSI, body.MSISDN, num) cmd += s.udmSubService.ParseCommandParams(body) // 去除msisdn参数,避免重复 omemsisdn := fmt.Sprintf(",msisdn=%s,", body.MSISDN) cmd = strings.Replace(cmd, omemsisdn, ",", 1) data, err := telnet.ConvertToStr(telnetClient, cmd) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } // 命令ok时 if strings.Contains(data, "ok") { s.udmSubService.LoadData(neId, body.IMSI, num, body.Remark) } c.JSON(200, resp.OkData(data)) } // UDM签约用户修改 // // PUT /:neId // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param data body object true "Request Param" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Modification // @Description UDM Subscriber User Modification // @Router /neData/udm/sub/{neId} [put] func (s *UDMSubController) Edit(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") if neId == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId is empty")) return } var body model.UDMSubUser if err := c.ShouldBindBodyWithJSON(&body); err != nil { errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err)) c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs)) return } if len(body.IMSI) != 15 { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: IMSI length is not 15 bits")) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } defer telnetClient.Close() // 发送MML cmd := fmt.Sprintf("mod udmuser:imsi=%s,", body.IMSI) cmd += s.udmSubService.ParseCommandParams(body) data, err := telnet.ConvertToStr(telnetClient, cmd) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } // 命令ok时 if strings.Contains(data, "ok") { body.NeId = neId s.udmSubService.Insert(neId, body) } c.JSON(200, resp.OkData(data)) } // UDM签约用户删除 // // DELETE /:neId/:imsi // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param value path string true "IMSI, multiple separated by a , sign" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Deletion // @Description UDM Subscriber User Deletion // @Router /neData/udm/sub/{neId}/{value} [delete] func (s *UDMSubController) Remove(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") imsi := c.Param("imsi") if neId == "" || len(imsi) < 15 { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId/imsi is empty")) return } // 处理字符转id数组后去重 imsiArr := strings.Split(imsi, ",") uniqueIDs := parse.RemoveDuplicates(imsiArr) if len(uniqueIDs) <= 0 { c.JSON(200, resp.Err(nil)) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.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") { s.udmSubService.Delete(neId, imsi) } resultData[imsi] = data } c.JSON(200, resp.OkData(resultData)) } // UDM签约用户批量删除 // // DELETE /:neId/:imsi/:num // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId path string true "NE ID" default(001) // @Param imsi path string true "IMSI" // @Param num path number true "Number of releases, value includes start imsi" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Batch Deletion // @Description UDM Subscriber User Batch Deletion // @Router /neData/udm/sub/{neId}/{imsi}/{num} [delete] func (s *UDMSubController) Removes(c *gin.Context) { language := reqctx.AcceptLanguage(c) neId := c.Param("neId") imsi := c.Param("imsi") num := c.Param("num") if neId == "" || len(imsi) < 15 || num == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId/imsi/num is empty")) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", neId) if neInfo.NeId != neId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient("UDM", neId, 1) if err != nil { c.JSON(200, resp.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, resp.ErrMsg(err.Error())) return } // 命令ok时 if strings.Contains(data, "ok") { s.udmSubService.LoadData(neId, imsi, num, "") } c.JSON(200, resp.OkData(data)) } // UDM签约用户导出 // // GET /export // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param neId query string true "NE ID" default(001) // @Param type query string true "File Type" Enums(csv,txt) default(txt) // @Param imsi query string false "IMSI" // @Param msisdn query string false "Msisdn" // @Param pageNum query number true "pageNum" default(1) // @Param pageSize query number true "pageSize" default(10) // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Export // @Description UDM Subscriber User Export // @Router /neData/udm/sub/export [post] func (s *UDMSubController) Export(c *gin.Context) { language := reqctx.AcceptLanguage(c) // 查询结果,根据查询条件结果,单页最大值限制 neId := c.Query("neId") fileType := c.Query("type") if neId == "" || fileType == "" { c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_CHEACK, "bind err: neId or type is empty")) return } if !(fileType == "csv" || fileType == "txt") { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "ne.udm.errImportUserSubFileFormat"))) return } query := reqctx.QueryMap(c) rows, total := s.udmSubService.FindByPage(query) if total == 0 { // 导出数据记录为空 c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.exportEmpty"))) return } // rows := s.udmSubService.SelectList(model.UDMSubUser{NeId: neId}) if len(rows) <= 0 { // 导出数据记录为空 c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.exportEmpty"))) return } // 文件名 fileName := fmt.Sprintf("udm_sub_user_export_%s_%d.%s", neId, time.Now().UnixMilli(), fileType) filePath := filepath.Join(file.ParseUploadFileDir(constants.UPLOAD_EXPORT), fileName) if fileType == "csv" { // 转换数据 data := [][]string{} data = append(data, []string{"IMSI", "MSISDN", "UeAmbrTpl", "NssaiTpl", "AreaForbiddenTpl", "ServiceAreaRestrictionTpl", "RatRestrictions", "CnTypeRestrictions", "SmfSel", "SmData", "EPSDat", "CreateTime"}) for _, v := range rows { 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) createTime := "" if v.CreateTime == 0 { createTime = time.Now().Format(time.RFC3339) } else { createTime = time.UnixMilli(v.CreateTime).Format(time.RFC3339) } data = append(data, []string{v.IMSI, v.MSISDN, v.UeAmbrTpl, v.NssaiTpl, v.AreaForbiddenTpl, v.ServiceAreaRestrictionTpl, v.RatRestrictions, v.CnTypeRestrictions, v.SmfSel, v.SmData, epsDat, createTime}) } // 输出到文件 if err := file.WriterFileCSV(data, filePath); err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } } if fileType == "txt" { // 转换数据 data := [][]string{} for _, v := range rows { 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) createTime := "" if v.CreateTime == 0 { createTime = time.Now().Format(time.RFC3339) } else { createTime = time.UnixMilli(v.CreateTime).Format(time.RFC3339) } data = append(data, []string{v.IMSI, v.MSISDN, v.UeAmbrTpl, v.NssaiTpl, v.AreaForbiddenTpl, v.ServiceAreaRestrictionTpl, v.RatRestrictions, v.CnTypeRestrictions, v.SmfSel, v.SmData, epsDat, createTime}) } // 输出到文件 if err := file.WriterFileTXTLine(data, ",", filePath); err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } } c.FileAttachment(filePath, fileName) } // UDM签约用户导入 // // POST /import // // @Tags network_data/udm/sub // @Accept json // @Produce json // @Param data body object true "Request Param" // @Success 200 {object} object "Response Results" // @Security TokenAuth // @Summary UDM Subscriber User Import // @Description UDM Subscriber User Import // @Router /neData/udm/sub/import [post] func (s *UDMSubController) Import(c *gin.Context) { language := reqctx.AcceptLanguage(c) var body struct { NeId string `json:"neId" binding:"required"` UploadPath string `json:"uploadPath" binding:"required"` } if err := c.ShouldBindBodyWithJSON(&body); err != nil { errMsgs := fmt.Sprintf("bind err: %s", resp.FormatBindError(err)) c.JSON(422, resp.CodeMsg(resp.CODE_PARAM_PARSER, errMsgs)) return } // 判断文件名 if !(strings.HasSuffix(body.UploadPath, ".csv") || strings.HasSuffix(body.UploadPath, ".txt")) { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "ne.udm.errImportUserSubFileFormat"))) return } // 查询网元获取IP neInfo := s.neInfoService.FindByNeTypeAndNeID("UDM", body.NeId) if neInfo.NeId != body.NeId || neInfo.IP == "" { c.JSON(200, resp.ErrMsg(i18n.TKey(language, "app.common.noNEInfo"))) return } // 网元主机的SSH客户端 sshClient, err := s.neInfoService.NeRunSSHClient(neInfo.NeType, neInfo.NeId) if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } defer sshClient.Close() // 网元主机的SSH客户端进行文件传输 sftpClient, err := sshClient.NewClientSFTP() if err != nil { c.JSON(200, resp.ErrMsg(err.Error())) return } defer sftpClient.Close() // 本地文件 localFilePath := file.ParseUploadFileAbsPath(body.UploadPath) neFilePath := fmt.Sprintf("/tmp/%s", filepath.Base(localFilePath)) // 复制到远程 if err = sftpClient.CopyFileLocalToRemote(localFilePath, neFilePath); err != nil { c.JSON(200, resp.ErrMsg("error uploading file")) return } // 网元主机的Telnet客户端 telnetClient, err := s.neInfoService.NeRunTelnetClient(neInfo.NeType, neInfo.NeId, 1) if err != nil { c.JSON(200, resp.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, resp.ErrMsg(err.Error())) return } // 命令ok时 if strings.Contains(data, "ok") { if strings.HasSuffix(body.UploadPath, ".csv") { data := file.ReadFileCSV(localFilePath) go s.udmSubService.InsertData(neInfo.NeId, "csv", data) } if strings.HasSuffix(body.UploadPath, ".txt") { data := file.ReadFileTXTLine(",", localFilePath) go s.udmSubService.InsertData(neInfo.NeId, "txt", data) } } c.JSON(200, resp.OkMsg(data)) }