From 141d593a40d503d5eb1959921fa8097869cd08d0 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Mon, 23 Dec 2024 10:00:09 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UE=E4=BA=8B=E4=BB=B6=E6=9F=A5=E8=AF=A2?=
 =?UTF-8?q?=E6=96=B9=E5=BC=8F=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/ws/processor/ue_connect.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/modules/ws/processor/ue_connect.go b/src/modules/ws/processor/ue_connect.go
index f6903819..7734e41f 100644
--- a/src/modules/ws/processor/ue_connect.go
+++ b/src/modules/ws/processor/ue_connect.go
@@ -20,10 +20,10 @@ func GetUEConnectByAMF(requestID string, data any) ([]byte, error) {
 		return nil, fmt.Errorf("query data structure error")
 	}
 
-	dataMap := neDataService.NewUEEventAMF.SelectPage(query)
+	rows, total := neDataService.NewUEEventAMF.SelectPage(query)
 	resultByte, err := json.Marshal(result.Ok(map[string]any{
 		"requestId": requestID,
-		"data":      dataMap,
+		"data":      map[string]any{"rows": rows, "total": total},
 	}))
 	return resultByte, err
 }
@@ -38,10 +38,10 @@ func GetUEConnectByMME(requestID string, data any) ([]byte, error) {
 		return nil, fmt.Errorf("query data structure error")
 	}
 
-	dataMap := neDataService.NewUEEventMME.SelectPage(query)
+	rows, total := neDataService.NewUEEventMME.SelectPage(query)
 	resultByte, err := json.Marshal(result.Ok(map[string]any{
 		"requestId": requestID,
-		"data":      dataMap,
+		"data":      map[string]any{"rows": rows, "total": total},
 	}))
 	return resultByte, err
 }