mayiming/toutoukan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改投票和get逻辑

Browse Source
This commit is contained in:
JACKYMYPERSON
2025-09-24 15:34:09 +08:00
parent 4a0e61cac5
commit f66aa3a715
3 changed files with 140 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
controllers/article/getarticle.go
View File

@@ -19,17 +19,17 @@ type ArticleOption struct {
// 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:"选项"` // 按顺序排列的选项列表
UserHasVoted bool `json:"用户是否已投票"` // 当前用户是否投过票
VotedOptionID *int64 `json:"用户投票的选项ID"` // 若已投票，记录选项ID（未投票为null）
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:"用户是否已投票"` // 当前用户是否投过票
VotedOptionIDs []int64 `json:"用户投票的选项ID"` // 若已投票，记录所有选项ID（未投票为空数组）
}
type UserReq struct {
@@ -96,21 +96,25 @@ func ArticleListget(c *gin.Context) {
return
}
// 4.2 查询当前用户对该文评的投票记录
var userVote struct {
// 4.2 查询当前用户对该文评的所有投票记录
var userVotes []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
Find(&userVotes).Error
// 标记用户是否投票及投票的选项ID
// 标记用户是否投票及所有投票的选项ID
userHasVoted := false
var votedOptionID *int64 = nil
if voteErr == nil {
var votedOptionIDs []int64 = []int64{}
if voteErr == nil && len(userVotes) > 0 {
userHasVoted = true
votedOptionID = &userVote.OptionID
} else if voteErr != gorm.ErrRecordNotFound {
// 收集所有投票的选项ID
for _, vote := range userVotes {
votedOptionIDs = append(votedOptionIDs, vote.OptionID)
}
} else if voteErr != nil && voteErr != gorm.ErrRecordNotFound {
// 处理查询错误（非"未找到"的错误）
c.JSON(http.StatusInternalServerError, gin.H{
"error": "查询用户投票记录失败: " + voteErr.Error(),
@@ -121,8 +125,14 @@ func ArticleListget(c *gin.Context) {
// 4.3 格式化选项数据（标记用户是否投了该选项）
var optionList []ArticleOption
for _, opt := range options {
// 检查当前选项是否是用户投票的选项
isVoted := userHasVoted && (opt.ID == *votedOptionID)
// 检查当前选项是否是用户投票的选项之一
isVoted := false
for _, votedID := range votedOptionIDs {
if opt.ID == votedID {
isVoted = true
break
}
}
optionList = append(optionList, ArticleOption{
ID: opt.ID,
@@ -134,17 +144,17 @@ func ArticleListget(c *gin.Context) {
// 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,
UserHasVoted: userHasVoted, // 整体标记用户是否投过票
VotedOptionID: votedOptionID, // 记录用户投票的选项ID（未投票则为null）
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, // 整体标记用户是否投过票
VotedOptionIDs: votedOptionIDs, // 记录用户投票的所有选项ID（未投票则为空数组）
}
// 4.5 加入结果集

174
controllers/article/votearticle.go
View File

@@ -1,7 +1,6 @@
package article
import (
"fmt"
"net/http"
"time"
"toutoukan/init/databaseInit"
@@ -10,10 +9,20 @@ import (
"gorm.io/gorm"
)
// 投票请求结构体，同时支持单选(optionId)和多选(optionIds)
type VoteArticleRequest struct {
Uid string `json:"uid" binding:"required,min=1,max=50"` // 用户ID，非空且长度限制
ArticleID int64 `json:"articleId" binding:"required,min=1"` // 文评ID，非空且为正整数
OptionID int64 `json:"optionId" binding:"required,min=1"` // 选项ID，投票必需
Uid string `json:"uid" binding:"required,min=1,max=50"` // 用户ID
ArticleID int64 `json:"articleId" binding:"required,min=1"` // 文章ID
OptionID int64 `json:"optionId" binding:"omitempty,min=1"` // 单个选项ID（单选时使用）
OptionIDs []int64 `json:"optionIds" binding:"omitempty,dive,min=1"` // 多个选项ID（多选时使用）
}
// 用户投票记录表结构
type UserVote struct {
UserID string `gorm:"column:user_id"`
VoteArticleID int64 `gorm:"column:vote_article_id"`
OptionID int64 `gorm:"column:option_id"`
VoteTime time.Time `gorm:"column:vote_time"`
}
func VoteArticle(c *gin.Context) {
@@ -42,31 +51,11 @@ func VoteArticle(c *gin.Context) {
}
}()
// 3. 检查用户是否已投票（防重复投票）
var existingVote struct{ ID int64 }
voteCheckErr := tx.Table("user_votes").
Where("user_id = ? AND vote_article_id = ?", req.Uid, req.ArticleID).
First(&existingVote).Error
if voteCheckErr == nil {
// 已存在投票记录
tx.Rollback()
c.JSON(http.StatusForbidden, gin.H{
"error": "您已对该文评投过票，无法重复投票",
})
return
} else if voteCheckErr != gorm.ErrRecordNotFound {
// 其他查询错误
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{
"error": "检查投票记录失败: " + voteCheckErr.Error(),
})
return
}
// 3. 检查文章状态和投票类型
var articleStatus struct {
IsEnded bool `gorm:"column:is_ended"`
EndTime time.Time `gorm:"column:end_time"` // 新增：获取截止时间
IsEnded bool `gorm:"column:is_ended"`
EndTime time.Time `gorm:"column:end_time"`
VoteType string `gorm:"column:vote_type"` // 投票类型：single-单选，multiple-多选
}
if err := tx.Table("article_list").
Where("articleId = ?", req.ArticleID).
@@ -80,70 +69,95 @@ func VoteArticle(c *gin.Context) {
return
}
// 双重检查：已标记结束 或 已过截止时间，均禁止投票
// 3.1 检查投票是否已结束
now := time.Now()
if articleStatus.IsEnded || articleStatus.EndTime.Before(now) {
tx.Rollback()
// 区分提示信息（可选，增强用户体验）
var errMsg string
if articleStatus.IsEnded {
errMsg = "该文评投票已结束，无法投票"
} else {
errMsg = "该文评投票已过截止时间，无法投票"
}
c.JSON(http.StatusForbidden, gin.H{"error": errMsg})
c.JSON(http.StatusForbidden, gin.H{"error": "该文评投票已结束或已过截止时间，无法投票"})
return
}
// 额外优化：如果已过截止时间但is_ended未更新，主动更新状态（可选）
if !articleStatus.IsEnded && articleStatus.EndTime.Before(now) {
if err := tx.Table("article_list").
Where("articleId = ?", req.ArticleID).
Update("is_ended", true).Error; err != nil {
// 此处不回滚，仅记录警告，不影响主要逻辑
fmt.Printf("警告：更新过期文评状态失败，articleId=%d, err=%v\n", req.ArticleID, err)
// 4. 处理投票选项，统一转换为选项ID列表
var selectedOptions []int64
if articleStatus.VoteType == "single" {
if req.OptionID == 0 || len(req.OptionIDs) > 0 {
tx.Rollback()
c.JSON(http.StatusForbidden, gin.H{
"error": "该文评为单选投票，请使用optionId提交单个选项",
})
return
}
selectedOptions = []int64{req.OptionID}
} else {
if len(req.OptionIDs) == 0 || req.OptionID != 0 {
tx.Rollback()
c.JSON(http.StatusForbidden, gin.H{
"error": "该文评为多选投票，请使用optionIds提交多个选项",
})
return
}
selectedOptions = req.OptionIDs
}
// 5. 检查选项是否属于该文评
var optionCheck struct{ ID int64 }
if err := tx.Table("article_options").
Where("id = ? AND vote_article_id = ?", req.OptionID, req.ArticleID).
First(&optionCheck).Error; err != nil {
// 5. 检查用户是否已投票
var voteCount int64
if err := tx.Table("user_votes").
Where("user_id = ? AND vote_article_id = ?", req.Uid, req.ArticleID).
Count(&voteCount).Error; err != nil {
tx.Rollback()
if err == gorm.ErrRecordNotFound {
c.JSON(http.StatusNotFound, gin.H{"error": "选项不存在或不属于该文评"})
} else {
c.JSON(http.StatusInternalServerError, gin.H{"error": "查询选项失败: " + err.Error()})
}
c.JSON(http.StatusInternalServerError, gin.H{"error": "检查投票记录失败: " + err.Error()})
return
}
// 6. 执行投票事务（三表操作）
// 6.1 记录用户投票记录
if err := tx.Table("user_votes").Create(map[string]interface{}{
"user_id": req.Uid,
"vote_article_id": req.ArticleID,
"option_id": req.OptionID,
"vote_time": time.Now(),
}).Error; err != nil {
if voteCount > 0 {
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{"error": "记录投票失败: " + err.Error()})
c.JSON(http.StatusForbidden, gin.H{"error": "您已对该文评投过票，无法重复投票"})
return
}
// 6.2 更新选项票数
// 6. 批量检查所有选项是否属于该文评
var existingOptionsCount int64
if err := tx.Table("article_options").
Where("id = ?", req.OptionID).
Where("vote_article_id = ? AND id IN ?", req.ArticleID, selectedOptions).
Count(&existingOptionsCount).Error; err != nil {
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{"error": "查询选项失败: " + err.Error()})
return
}
if existingOptionsCount != int64(len(selectedOptions)) {
tx.Rollback()
c.JSON(http.StatusNotFound, gin.H{"error": "部分选项ID不存在或不属于该文评"})
return
}
// 7. 执行批量投票事务
voteTime := time.Now()
// 7.1 批量插入用户投票记录
userVotes := make([]UserVote, len(selectedOptions))
for i, optionID := range selectedOptions {
userVotes[i] = UserVote{
UserID: req.Uid,
VoteArticleID: req.ArticleID,
OptionID: optionID,
VoteTime: voteTime,
}
}
if err := tx.Table("user_votes").Create(&userVotes).Error; err != nil {
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{"error": "批量记录投票失败: " + err.Error()})
return
}
// 7.2 批量更新选项票数
if err := tx.Table("article_options").
Where("id IN ?", selectedOptions).
Update("option_votes_num", gorm.Expr("option_votes_num + 1")).Error; err != nil {
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{"error": "更新选项票数失败: " + err.Error()})
c.JSON(http.StatusInternalServerError, gin.H{"error": "批量更新选项票数失败: " + err.Error()})
return
}
// 6.3 更新文评总投票人数
// 7.3 更新文评总投票人数
if err := tx.Table("article_list").
Where("articleId = ?", req.ArticleID).
Update("total_voters_num", gorm.Expr("total_voters_num + 1")).Error; err != nil {
@@ -152,21 +166,27 @@ func VoteArticle(c *gin.Context) {
return
}
// 7. 提交事务
// 8. 提交事务
if err := tx.Commit().Error; err != nil {
tx.Rollback()
c.JSON(http.StatusInternalServerError, gin.H{"error": "提交投票失败: " + err.Error()})
return
}
// 8. 返回成功响应
// 9. 返回成功响应
responseData := gin.H{
"articleId": req.ArticleID,
"voteTime": voteTime.Format("2006-01-02 15:04:05"),
}
if articleStatus.VoteType == "single" {
responseData["optionId"] = req.OptionID
} else {
responseData["optionIds"] = req.OptionIDs
}
c.JSON(http.StatusOK, gin.H{
"success": true,
"message": "投票成功",
"data": gin.H{
"articleId": req.ArticleID,
"optionId": req.OptionID,
"voteTime": time.Now().Format("2006-01-02 15:04:05"),
},
"data": responseData,
})
}