From 0c7de5fac96cdbadf79ac755778c57affb8f12e2 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Wed, 13 Sep 2023 11:56:35 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E5=AF=BC=E5=85=A5=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E5=90=8E=E7=BC=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 features/udm_user/api_udm_user.go | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/features/udm_user/api_udm_user.go b/features/udm_user/api_udm_user.go
index 0d73f2b1..be9de478 100644
--- a/features/udm_user/api_udm_user.go
+++ b/features/udm_user/api_udm_user.go
@@ -519,8 +519,15 @@ func (s *UdmUserApi) UdmAuthUserImport(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// 获取文件名
+	_, fileHeader, err := r.FormFile("file")
+	if err != nil {
+		ctx.JSON(w, 200, result.ErrMsg(err.Error()))
+		return
+	}
+
 	// 文件名
-	fileName := fmt.Sprintf("OMC_AUTH_USER_IMPORT_%s_%d.csv", neId, time.Now().UnixMilli())
+	fileName := fmt.Sprintf("OMC_AUTH_USER_IMPORT_%s_%s", neId, fileHeader.Filename)
 	filePath := fmt.Sprintf("%s/upload/mml/%s", conf.Get("ne.omcdir"), fileName)
 	dstPath := conf.Get("mml.upload").(string)
 	// 输出保存文件
@@ -1037,8 +1044,15 @@ func (s *UdmUserApi) UdmSubUserImport(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	// 获取文件名
+	_, fileHeader, err := r.FormFile("file")
+	if err != nil {
+		ctx.JSON(w, 200, result.ErrMsg(err.Error()))
+		return
+	}
+
 	// 文件名
-	fileName := fmt.Sprintf("OMC_SUB_USER_IMPORT_%s_%d.csv", neId, time.Now().UnixMilli())
+	fileName := fmt.Sprintf("OMC_SUB_USER_IMPORT_%s_%s", neId, fileHeader.Filename)
 	filePath := fmt.Sprintf("%s/upload/mml/%s", conf.Get("ne.omcdir"), fileName)
 	dstPath := conf.Get("mml.upload").(string)
 	// 输出保存文件