491 lines
14 KiB
Go
491 lines
14 KiB
Go
package controller
|
||
|
||
import (
|
||
"fmt"
|
||
"path/filepath"
|
||
"strings"
|
||
"time"
|
||
|
||
"be.ems/features/ue/model"
|
||
"be.ems/features/ue/service"
|
||
"be.ems/src/framework/constants/uploadsubpath"
|
||
"be.ems/src/framework/i18n"
|
||
"be.ems/src/framework/telnet"
|
||
"be.ems/src/framework/utils/ctx"
|
||
"be.ems/src/framework/utils/file"
|
||
"be.ems/src/framework/utils/parse"
|
||
"be.ems/src/framework/vo/result"
|
||
neService "be.ems/src/modules/network_element/service"
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/gin-gonic/gin/binding"
|
||
)
|
||
|
||
// 实例化控制层 IMSUserController 结构体
|
||
var NewIMSUserController = &IMSUserController{
|
||
imsUserService: service.NewIMSUserService,
|
||
neInfoService: neService.NewNeInfo,
|
||
}
|
||
|
||
// IMS用户信息 控制层处理
|
||
//
|
||
// @Description IMS用户信息 控制层处理
|
||
type IMSUserController struct {
|
||
imsUserService *service.IMSUserService // IMS User信息服务
|
||
neInfoService *neService.NeInfo // 网元信息服务
|
||
}
|
||
|
||
func (s *IMSUserController) 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
|
||
}
|
||
|
||
data := s.imsUserService.ResetData(neId)
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
func (s *IMSUserController) List(c *gin.Context) {
|
||
querys := ctx.QueryMap(c)
|
||
// querys["userName"] = ctx.LoginUserToUserName(c)
|
||
// 判断租户角色
|
||
loginUser, _ := ctx.LoginUser(c)
|
||
if len(loginUser.User.Roles) > 0 {
|
||
for _, v := range loginUser.User.Roles {
|
||
if v.RoleKey == "tenant" {
|
||
querys["tenantName"] = loginUser.User.UserName
|
||
break
|
||
}
|
||
}
|
||
}
|
||
data := s.imsUserService.SelectPage(querys)
|
||
c.JSON(200, result.Ok(data))
|
||
}
|
||
|
||
func (s *IMSUserController) 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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser: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("Not found IMS User data"))
|
||
return
|
||
}
|
||
|
||
// 解析返回的数据
|
||
u := s.imsUserService.ParseInfo(imsi, neId, data)
|
||
s.imsUserService.Insert(neId, u)
|
||
c.JSON(200, result.OkData(u))
|
||
}
|
||
|
||
func (s *IMSUserController) 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.IMSUser
|
||
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
||
if err != nil || (len(body.IMSI) < model.IMSI_MAX_LENGTH && body.Tag == model.TAG_VoLTE) {
|
||
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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser:imsi=%s,", body.IMSI)
|
||
cmd += s.imsUserService.ParseCommandParams(body)
|
||
data, err := telnet.ConvertToStr(telnetClient, cmd)
|
||
if err != nil {
|
||
c.JSON(200, result.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
|
||
// 命令ok时
|
||
if strings.Contains(data, "ok") {
|
||
body.NeId = neId
|
||
s.imsUserService.Insert(neId, body)
|
||
}
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
func (s *IMSUserController) 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.IMSUser
|
||
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
||
if err != nil || (len(body.IMSI) < model.IMSI_MAX_LENGTH && body.Tag == model.TAG_VoLTE) {
|
||
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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser:start_imsi=%s,start_msisdn=%s,sub_num=%s,", body.IMSI, body.MSISDN, num)
|
||
cmd += s.imsUserService.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, result.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
// 命令ok时
|
||
if strings.Contains(data, "ok") {
|
||
s.imsUserService.LoadData(neId, body.IMSI, num)
|
||
}
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
func (s *IMSUserController) 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.IMSUser
|
||
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
||
if err != nil || (len(body.IMSI) < model.IMSI_MAX_LENGTH && body.Tag == model.TAG_VoLTE) {
|
||
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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser:imsi=%s,", body.IMSI)
|
||
cmd += s.imsUserService.ParseCommandParams(body)
|
||
data, err := telnet.ConvertToStr(telnetClient, cmd)
|
||
if err != nil {
|
||
c.JSON(200, result.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
|
||
// 命令ok时
|
||
if strings.Contains(data, "ok") {
|
||
body.NeId = neId
|
||
s.imsUserService.Insert(neId, body)
|
||
}
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
func (s *IMSUserController) 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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser:imsi=%s", imsi)
|
||
data, err := telnet.ConvertToStr(telnetClient, cmd)
|
||
if err != nil {
|
||
resultData[imsi] = err.Error()
|
||
continue
|
||
}
|
||
// 命令ok时
|
||
if strings.Contains(data, "ok") {
|
||
s.imsUserService.Delete(neId, imsi)
|
||
}
|
||
resultData[imsi] = data
|
||
}
|
||
|
||
c.JSON(200, result.OkData(resultData))
|
||
}
|
||
|
||
func (s *IMSUserController) 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
|
||
}
|
||
// 网元主机的Telnet客户端
|
||
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 imsuser: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") {
|
||
s.imsUserService.LoadData(neId, imsi, num)
|
||
}
|
||
c.JSON(200, result.OkData(data))
|
||
}
|
||
|
||
func (s *IMSUserController) Export(c *gin.Context) {
|
||
language := ctx.AcceptLanguage(c)
|
||
// 查询结果，根据查询条件结果，单页最大值限制
|
||
querys := ctx.BodyJSONMap(c)
|
||
neId := querys["neId"].(string)
|
||
fileType := querys["type"].(string)
|
||
if neId == "" || fileType == "" {
|
||
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
||
return
|
||
}
|
||
if !(fileType == "csv" || fileType == "txt") {
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "ne.udm.errImportUserSubFileFormat")))
|
||
return
|
||
}
|
||
|
||
// 判断租户角色
|
||
loginUser, _ := ctx.LoginUser(c)
|
||
if len(loginUser.User.Roles) > 0 {
|
||
for _, v := range loginUser.User.Roles {
|
||
if v.RoleKey == "tenant" {
|
||
querys["tenantName"] = loginUser.User.UserName
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
querys["pageNum"] = 1
|
||
querys["pageSize"] = 10000
|
||
data := s.imsUserService.SelectPage(querys)
|
||
if parse.Number(data["total"]) == 0 {
|
||
// 导出数据记录为空
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.exportEmpty")))
|
||
return
|
||
}
|
||
rows := data["rows"].([]model.IMSUser)
|
||
|
||
// rows := s.imsUserService.SelectList(model.IMSUser{NeId: neId})
|
||
if len(rows) <= 0 {
|
||
// 导出数据记录为空
|
||
c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.exportEmpty")))
|
||
return
|
||
}
|
||
|
||
// 文件名
|
||
fileName := fmt.Sprintf("u_ims_user_export_%s_%d.%s", neId, time.Now().UnixMilli(), fileType)
|
||
filePath := filepath.Join(file.ParseUploadFileDir(uploadsubpath.EXPORT), fileName)
|
||
|
||
if fileType == "csv" {
|
||
// 转换数据
|
||
data := [][]string{}
|
||
data = append(data, []string{"IMSI", "MSISDN", "Tag", "VNI"})
|
||
for _, v := range rows {
|
||
data = append(data, []string{v.IMSI, v.MSISDN, v.Tag.String(), v.VNI})
|
||
}
|
||
// 输出到文件
|
||
if err := file.WriterFileCSV(data, filePath); err != nil {
|
||
c.JSON(200, result.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
}
|
||
|
||
if fileType == "txt" {
|
||
// 转换数据
|
||
data := [][]string{}
|
||
for _, v := range rows {
|
||
data = append(data, []string{v.IMSI, v.MSISDN, v.Tag.String(), v.VNI})
|
||
}
|
||
// 输出到文件
|
||
if err := file.WriterFileTXT(data, ",", filePath); err != nil {
|
||
c.JSON(200, result.ErrMsg(err.Error()))
|
||
return
|
||
}
|
||
}
|
||
|
||
c.FileAttachment(filePath, fileName)
|
||
}
|
||
|
||
func (s *IMSUserController) 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 imsuser: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)
|
||
go s.imsUserService.InsertData(neInfo.NeId, "csv", data)
|
||
}
|
||
if strings.HasSuffix(body.UploadPath, ".txt") {
|
||
data := file.ReadFileTXT(",", localFilePath)
|
||
go s.imsUserService.InsertData(neInfo.NeId, "txt", data)
|
||
}
|
||
}
|
||
c.JSON(200, result.OkMsg(data))
|
||
}
|