From 357289c49190adc62c12c8fd0811091e185f9655 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 25 Apr 2025 15:49:20 +0800
Subject: [PATCH] =?UTF-8?q?style:=20=20UDM=E6=95=B0=E6=8D=AE=E5=AF=BC?=
 =?UTF-8?q?=E5=87=BA=E4=BB=A3=E7=A0=81=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_data/controller/udm_auth.go      | 4 +---
 src/modules/network_data/controller/udm_voip.go      | 6 ++----
 src/modules/network_data/controller/udm_volte_ims.go | 8 +++-----
 src/modules/network_data/repository/udm_volte_ims.go | 2 +-
 4 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/src/modules/network_data/controller/udm_auth.go b/src/modules/network_data/controller/udm_auth.go
index 5677adb9..4c12d3ee 100644
--- a/src/modules/network_data/controller/udm_auth.go
+++ b/src/modules/network_data/controller/udm_auth.go
@@ -497,8 +497,7 @@ func (s *UDMAuthController) Export(c *gin.Context) {
 			data = append(data, []string{v.IMSI, v.Ki, v.AlgoIndex, v.Amf, opc})
 		}
 		// 输出到文件
-		err := file.WriterFileCSV(data, filePath)
-		if err != nil {
+		if err := file.WriterFileCSV(data, filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
 		}
@@ -515,7 +514,6 @@ func (s *UDMAuthController) Export(c *gin.Context) {
 			data = append(data, []string{v.IMSI, v.Ki, v.AlgoIndex, v.Amf, opc})
 		}
 		// 输出到文件
-
 		if err := file.WriterFileTXT(data, ",", filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
diff --git a/src/modules/network_data/controller/udm_voip.go b/src/modules/network_data/controller/udm_voip.go
index c2000b48..c9f14ddc 100644
--- a/src/modules/network_data/controller/udm_voip.go
+++ b/src/modules/network_data/controller/udm_voip.go
@@ -30,7 +30,7 @@ var NewUDMVOIP = &UDMVOIPController{
 //
 // PATH /udm/voip
 type UDMVOIPController struct {
-	udmVOIPService *neDataService.UDMVOIPUser // UDM鉴权信息服务
+	udmVOIPService *neDataService.UDMVOIPUser // UDMVOIP信息服务
 	neInfoService  *neService.NeInfo          // 网元信息服务
 }
 
@@ -431,8 +431,7 @@ func (s *UDMVOIPController) Export(c *gin.Context) {
 			data = append(data, []string{v.UserName, v.Password})
 		}
 		// 输出到文件
-		err := file.WriterFileCSV(data, filePath)
-		if err != nil {
+		if err := file.WriterFileCSV(data, filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
 		}
@@ -445,7 +444,6 @@ func (s *UDMVOIPController) Export(c *gin.Context) {
 			data = append(data, []string{v.UserName, v.Password})
 		}
 		// 输出到文件
-
 		if err := file.WriterFileTXT(data, ",", filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
diff --git a/src/modules/network_data/controller/udm_volte_ims.go b/src/modules/network_data/controller/udm_volte_ims.go
index d6d09ce0..f0247f1f 100644
--- a/src/modules/network_data/controller/udm_volte_ims.go
+++ b/src/modules/network_data/controller/udm_volte_ims.go
@@ -30,7 +30,7 @@ var NewUDMVolteIMS = &UDMVolteIMSController{
 //
 // PATH /udm/volte-ims
 type UDMVolteIMSController struct {
-	udmVolteIMSService *neDataService.UDMVolteIMSUser // UDM鉴权信息服务
+	udmVolteIMSService *neDataService.UDMVolteIMSUser // UDMVolteIMS信息服务
 	neInfoService      *neService.NeInfo              // 网元信息服务
 }
 
@@ -454,13 +454,12 @@ func (s *UDMVolteIMSController) Export(c *gin.Context) {
 	if fileType == "csv" {
 		// 转换数据
 		data := [][]string{}
-		data = append(data, []string{"IMSI", "MSISDN", "Tag", "VNI"})
+		data = append(data, []string{"IMSI", "MSISDN", "TAG", "VNI"})
 		for _, v := range rows {
 			data = append(data, []string{v.IMSI, v.MSISDN, v.Tag, v.VNI})
 		}
 		// 输出到文件
-		err := file.WriterFileCSV(data, filePath)
-		if err != nil {
+		if err := file.WriterFileCSV(data, filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
 		}
@@ -473,7 +472,6 @@ func (s *UDMVolteIMSController) Export(c *gin.Context) {
 			data = append(data, []string{v.IMSI, v.MSISDN, v.Tag, v.VNI})
 		}
 		// 输出到文件
-
 		if err := file.WriterFileTXT(data, ",", filePath); err != nil {
 			c.JSON(200, resp.ErrMsg(err.Error()))
 			return
diff --git a/src/modules/network_data/repository/udm_volte_ims.go b/src/modules/network_data/repository/udm_volte_ims.go
index 0a6d9128..b4fc935d 100644
--- a/src/modules/network_data/repository/udm_volte_ims.go
+++ b/src/modules/network_data/repository/udm_volte_ims.go
@@ -103,7 +103,7 @@ func (r UDMVolteIMSUser) SelectList(u model.UDMVolteIMSUser) []model.UDMVolteIMS
 
 	// 查询数据
 	arr := []model.UDMVolteIMSUser{}
-	if err := tx.Order("user_name asc").Find(&arr).Error; err != nil {
+	if err := tx.Order("imsi asc").Find(&arr).Error; err != nil {
 		logger.Errorf("query err => %v", err)
 	}
 	return arr