更新投票功能

controllers/article/getarticle.go

@@ -6,16 +6,18 @@ 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"`    // 该选项的投票数
+	IsVotedByUser bool   `json:"is_voted"` // 当前用户是否投了该选项
 }
 
-// ArticleResponse 单个文评的响应结构（保持不变）
+// ArticleResponse 单个文评的响应结构
 type ArticleResponse struct {
 	ID            int64           `json:"文评ID"`      // 文评唯一ID
 	Title         string          `json:"文评标题"`      // 文评标题
@@ -26,11 +28,28 @@ type ArticleResponse struct {
 	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,17 +96,43 @@ 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
+				ID:            opt.ID,
 				Name:          opt.OptionName,
 				Votes:         opt.VoteCount,
+				IsVotedByUser: isVoted, // 标记当前用户是否投了这个选项
 			})
 		}
 
-		// 5. 组装单个文评的响应数据
+		// 4.4 组装单个文评的响应数据
 		articleData := ArticleResponse{
 			ID:            article.ArticleID,
 			Title:         article.Title,
@@ -98,9 +143,11 @@ func ArticleListget(c *gin.Context) {
 			PublisherID:   article.PublishUserID,
 			CreateTime:    article.CreateTime,
 			Options:       optionList,
+			UserHasVoted:  userHasVoted,  // 整体标记用户是否投过票
+			VotedOptionID: votedOptionID, // 记录用户投票的选项ID（未投票则为null）
 		}
 
-		// 6. 加入结果集
+		// 4.5 加入结果集
 		result[article.Title] = articleData
 	}
 

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{