From e59f666541f00bb73141d7663daeb8c2ef9c4a53 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 19 Sep 2024 11:45:38 +0800
Subject: [PATCH] =?UTF-8?q?style:=20=E4=BC=98=E5=8C=96=E6=B3=A8=E9=87=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_data/controller/upf.go    | 7 +++----
 src/modules/network_element/fetch_link/udm.go | 3 +--
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/modules/network_data/controller/upf.go b/src/modules/network_data/controller/upf.go
index aa3ef337..366fc15d 100644
--- a/src/modules/network_data/controller/upf.go
+++ b/src/modules/network_data/controller/upf.go
@@ -32,9 +32,8 @@ type UPFController struct {
 func (s *UPFController) TotalFlow(c *gin.Context) {
 	language := ctx.AcceptLanguage(c)
 	var querys struct {
-		NeType string `json:"neType" form:"neType" binding:"required"`
-		NeID   string `form:"neId" binding:"required"`
-		Day    int    `form:"day" binding:"required"`
+		NeID string `form:"neId" binding:"required"`
+		Day  int    `form:"day" binding:"required"`
 	}
 	if err := c.ShouldBindQuery(&querys); querys.Day < 0 || err != nil {
 		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
@@ -42,7 +41,7 @@ func (s *UPFController) TotalFlow(c *gin.Context) {
 	}
 
 	// 查询网元获取IP
-	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID(querys.NeType, querys.NeID)
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UPF", querys.NeID)
 	if neInfo.NeId != querys.NeID || neInfo.IP == "" {
 		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
 		return
diff --git a/src/modules/network_element/fetch_link/udm.go b/src/modules/network_element/fetch_link/udm.go
index 905bbb5b..d027f9f0 100644
--- a/src/modules/network_element/fetch_link/udm.go
+++ b/src/modules/network_element/fetch_link/udm.go
@@ -28,8 +28,7 @@ func UDMImportAuth(udmIP string, data map[string]any) (string, error) {
 	}
 
 	// 序列化结果
-	err = json.Unmarshal(resBytes, &resData)
-	if err != nil {
+	if err = json.Unmarshal(resBytes, &resData); err != nil {
 		logger.Errorf("UDMImportAuth Unmarshal %s", err.Error())
 		return "", err
 	}