更新投票功能

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
 	}