From 1d30dd214d022f40eb55779fc9dd24b525455c50 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Sun, 30 Mar 2025 13:22:40 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=94=A8=E6=88=B7=E7=9A=84?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=E7=BB=9F=E8=AE=A1=E6=95=B0=E6=8D=AE=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/user.go            | 15 +++++++++++++++
 handler/user.go        |  9 +++++++++
 service/userService.go |  4 ++++
 3 files changed, 28 insertions(+)

diff --git a/dao/user.go b/dao/user.go
index 7b23165..01f8cf4 100644
--- a/dao/user.go
+++ b/dao/user.go
@@ -173,3 +173,18 @@ func FindUsersDefault() []User {
 	DB.Limit(20).Find(&users)
 	return users
 }
+
+// 用户的信息统计数据
+type UserStatistics struct {
+	SessionCount int64 `json:"session_count" form:"session_count"` //会话数量
+	FileCount    int64 `json:"file_count" form:"file_count"`       //文件数量
+	MessageCount int64 `json:"message_count" form:"message_count"` //消息数量,提问数量
+}
+
+func UserStatisticsData(userID int) UserStatistics {
+	userStatistics := UserStatistics{}
+	DB.Model(&Session{}).Group("user_id").Count(&userStatistics.SessionCount)
+	DB.Model(&File{}).Group("user_id").Count(&userStatistics.FileCount)
+	DB.Model(&Message{}).Where("from_id = ?", userID).Group("from_id").Count(&userStatistics.MessageCount)
+	return userStatistics
+}
diff --git a/handler/user.go b/handler/user.go
index 0ce1c40..5749f4e 100644
--- a/handler/user.go
+++ b/handler/user.go
@@ -29,6 +29,7 @@ func SetUpUserGroup(router *gin.Engine) {
 	userGroup.POST("/sync", GetSyncUserInfo)
 	userGroup.POST("/delete", DeleteUser)
 	userGroup.POST("/reset", ResetPassword)
+	userGroup.POST("/statistic", GetUserStatistic)
 }
 
 type RLReq struct {
@@ -61,6 +62,14 @@ type ResetPasswordReq struct {
 	Code        string `json:"code" form:"code"` //验证码
 }
 
+func GetUserStatistic(c *gin.Context) {
+	id, _ := c.Get("id")
+	userId := int(id.(float64))
+	//获取数据
+	us := service.GetUserStatistics(userId)
+	c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "data": us})
+}
+
 func ResetPassword(c *gin.Context) {
 	var req_data ResetPasswordReq
 	if err := c.ShouldBind(&req_data); err == nil {
diff --git a/service/userService.go b/service/userService.go
index 63deb57..49a29de 100644
--- a/service/userService.go
+++ b/service/userService.go
@@ -417,3 +417,7 @@ func CreateTokenAndSave(user dao.User) (string, error) {
 	// 返回令牌
 	return tokenString, err
 }
+
+func GetUserStatistics(userID int) dao.UserStatistics {
+	return dao.UserStatisticsData(userID)
+}