From 642143502249f7f23fa51be66c5b5ff6409a2aeb Mon Sep 17 00:00:00 2001 From: mayiming <1627832236@qq.com> Date: Wed, 24 Sep 2025 01:48:12 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=8A=95=E7=A5=A8=E5=8A=9F?= =?UTF-8?q?=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- controllers/article/getarticle.go | 123 +++++++++++++++++++++--------- controllers/user/userLogin.go | 5 +- 2 files changed, 87 insertions(+), 41 deletions(-) diff --git a/controllers/article/getarticle.go b/controllers/article/getarticle.go index 233628c..cc444fb 100644 --- a/controllers/article/getarticle.go +++ b/controllers/article/getarticle.go @@ -6,31 +6,50 @@ import ( "toutoukan/init/databaseInit" "github.com/gin-gonic/gin" + "gorm.io/gorm" ) -// ArticleOption 文评选项结构（增加了ID字段） +// ArticleOption 文评选项结构 type ArticleOption struct { - ID int64 `json:"id"` // 选项ID - Name string `json:"name"` // 选项名称 - Votes int `json:"votes"` // 该选项的投票数 + ID int64 `json:"id"` // 选项ID + Name string `json:"name"` // 选项名称 + Votes int `json:"votes"` // 该选项的投票数 + IsVotedByUser bool `json:"is_voted"` // 当前用户是否投了该选项 } -// ArticleResponse 单个文评的响应结构（保持不变） +// ArticleResponse 单个文评的响应结构 type ArticleResponse struct { - ID int64 `json:"文评ID"` // 文评唯一ID - Title string `json:"文评标题"` // 文评标题 - VoteType string `json:"投票类型"` // 投票类型 - TotalVoters int `json:"总投票人数"` // 总投票人数 - EndTime string `json:"结束时间"` // 结束时间 - IsEnded bool `json:"是否结束"` // 是否结束 - PublisherID string `json:"发布者ID"` // 发布者ID - CreateTime string `json:"创建时间"` // 创建时间 - Options []ArticleOption `json:"选项"` // 按顺序排列的选项列表 + ID int64 `json:"文评ID"` // 文评唯一ID + Title string `json:"文评标题"` // 文评标题 + VoteType string `json:"投票类型"` // 投票类型 + TotalVoters int `json:"总投票人数"` // 总投票人数 + EndTime string `json:"结束时间"` // 结束时间 + IsEnded bool `json:"是否结束"` // 是否结束 + PublisherID string `json:"发布者ID"` // 发布者ID + CreateTime string `json:"创建时间"` // 创建时间 + Options []ArticleOption `json:"选项"` // 按顺序排列的选项列表 + UserHasVoted bool `json:"用户是否已投票"` // 当前用户是否投过票 + VotedOptionID *int64 `json:"用户投票的选项ID"` // 若已投票，记录选项ID（未投票为null） } -// ArticleList 获取所有文评及选项信息（带排序） +type UserReq struct { + Uid string `json:"uid"` +} + +// ArticleListget 获取所有文评及选项信息（包含用户投票状态） func ArticleListget(c *gin.Context) { - // 1. 查询所有文评，获取表中所有字段 + var req UserReq + + // 1. 解析并验证请求参数 + if err := c.ShouldBindJSON(&req); err != nil { + c.JSON(http.StatusBadRequest, gin.H{ + "error": "参数解析失败", + "detail": err.Error(), + }) + return + } + + // 2. 查询所有文评 var articles []struct { ArticleID int64 `gorm:"column:articleId"` Title string `gorm:"column:title"` @@ -41,7 +60,7 @@ func ArticleListget(c *gin.Context) { PublishUserID string `gorm:"column:publish_user_id"` CreateTime string `gorm:"column:create_time"` } - // 查询文评主表，选择所有字段 + if err := databaseInit.UserDB.Table("article_list"). Select("articleId, title, vote_type, total_voters_num, end_time, is_ended, publish_user_id, create_time"). Find(&articles).Error; err != nil { @@ -53,22 +72,22 @@ func ArticleListget(c *gin.Context) { fmt.Println("所有文评列表：", articles) - // 2. 构建结果映射 + // 3. 构建结果映射 result := make(map[string]interface{}) - // 3. 逐个查询每个文评的选项 + // 4. 逐个处理文评 for _, article := range articles { + // 4.1 查询当前文评的选项 var options []struct { - ID int64 `gorm:"column:id"` // 新增：获取选项ID + ID int64 `gorm:"column:id"` OptionName string `gorm:"column:option_content"` VoteCount int `gorm:"column:option_votes_num"` - SortOrder int `gorm:"column:sort_order"` // 用于排序的字段 + SortOrder int `gorm:"column:sort_order"` } - // 查询当前文评的所有选项，并按sort_order排序 if err := databaseInit.UserDB.Table("article_options"). Where("vote_article_id = ?", article.ArticleID). - Select("id, option_content, option_votes_num, sort_order"). // 新增：选择id字段 + Select("id, option_content, option_votes_num, sort_order"). Order("sort_order ASC"). Find(&options).Error; err != nil { c.JSON(http.StatusInternalServerError, gin.H{ @@ -77,30 +96,58 @@ func ArticleListget(c *gin.Context) { return } - // 4. 格式化选项数据为有序切片（包含ID） + // 4.2 查询当前用户对该文评的投票记录 + var userVote struct { + OptionID int64 `gorm:"column:option_id"` + } + voteErr := databaseInit.UserDB.Table("user_votes"). + Where("user_id = ? AND vote_article_id = ?", req.Uid, article.ArticleID). + First(&userVote).Error + + // 标记用户是否投票及投票的选项ID + userHasVoted := false + var votedOptionID *int64 = nil + if voteErr == nil { + userHasVoted = true + votedOptionID = &userVote.OptionID + } else if voteErr != gorm.ErrRecordNotFound { + // 处理查询错误（非"未找到"的错误） + c.JSON(http.StatusInternalServerError, gin.H{ + "error": "查询用户投票记录失败: " + voteErr.Error(), + }) + return + } + + // 4.3 格式化选项数据（标记用户是否投了该选项） var optionList []ArticleOption for _, opt := range options { + // 检查当前选项是否是用户投票的选项 + isVoted := userHasVoted && (opt.ID == *votedOptionID) + optionList = append(optionList, ArticleOption{ - ID: opt.ID, // 新增：赋值选项ID - Name: opt.OptionName, - Votes: opt.VoteCount, + ID: opt.ID, + Name: opt.OptionName, + Votes: opt.VoteCount, + IsVotedByUser: isVoted, // 标记当前用户是否投了这个选项 }) } - // 5. 组装单个文评的响应数据 + // 4.4 组装单个文评的响应数据 articleData := ArticleResponse{ - ID: article.ArticleID, - Title: article.Title, - VoteType: article.VoteType, - TotalVoters: article.TotalVotersNum, - EndTime: article.EndTime, - IsEnded: article.IsEnded, - PublisherID: article.PublishUserID, - CreateTime: article.CreateTime, - Options: optionList, + ID: article.ArticleID, + Title: article.Title, + VoteType: article.VoteType, + TotalVoters: article.TotalVotersNum, + EndTime: article.EndTime, + IsEnded: article.IsEnded, + PublisherID: article.PublishUserID, + CreateTime: article.CreateTime, + Options: optionList, + UserHasVoted: userHasVoted, // 整体标记用户是否投过票 + VotedOptionID: votedOptionID, // 记录用户投票的选项ID（未投票则为null） } - // 6. 加入结果集 + // 4.5 加入结果集 result[article.Title] = articleData } diff --git a/controllers/user/userLogin.go b/controllers/user/userLogin.go index ee7b29c..762c5b1 100644 --- a/controllers/user/userLogin.go +++ b/controllers/user/userLogin.go @@ -89,14 +89,14 @@ func UserLogin(c *gin.Context) { // 业务逻辑实现 var user UserInfo - var username string + username := "" // 查找用户是否存在 result := databaseInit.UserDB.Where("uid = ?", openid).First(&user) if result.Error != nil { if errors.Is(result.Error, gorm.ErrRecordNotFound) { // 用户不存在，创建新用户 - username := generateUsername() + username = generateUsername() now := time.Now() newUser := UserInfo{ Uid: openid, @@ -117,7 +117,6 @@ func UserLogin(c *gin.Context) { }) return } - username = newUser.Username } else { // 数据库查询错误 c.JSON(http.StatusInternalServerError, gin.H{