toutoukan/controllers/article/getarticle.go

package article

import (
	"fmt"
	"net/http"
	"toutoukan/init/databaseInit"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

// ArticleOption 文评选项结构
type ArticleOption struct {
	ID            int64  `json:"id"`       // 选项ID
	Name          string `json:"name"`     // 选项名称
	Votes         int    `json:"votes"`    // 该选项的投票数
	IsVotedByUser bool   `json:"is_voted"` // 当前用户是否投了该选项
}

// 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:"用户是否已投票"`   // 当前用户是否投过票
	VotedOptionIDs []int64         `json:"用户投票的选项ID"` // 若已投票，记录所有选项ID（未投票为空数组）
}

type UserReq struct {
	Uid string `json:"uid"`
}

// ArticleListget 获取所有文评及选项信息（包含用户投票状态）
func ArticleListget(c *gin.Context) {
	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"`
		VoteType       string `gorm:"column:vote_type"`
		TotalVotersNum int    `gorm:"column:total_voters_num"`
		EndTime        string `gorm:"column:end_time"`
		IsEnded        bool   `gorm:"column:is_ended"`
		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 {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "查询文评失败: " + err.Error(),
		})
		return
	}

	fmt.Println("所有文评列表：", articles)

	// 3. 构建结果映射
	result := make(map[string]interface{})

	// 4. 逐个处理文评
	for _, article := range articles {
		// 4.1 查询当前文评的选项
		var options []struct {
			ID         int64  `gorm:"column:id"`
			OptionName string `gorm:"column:option_content"`
			VoteCount  int    `gorm:"column:option_votes_num"`
			SortOrder  int    `gorm:"column:sort_order"`
		}

		if err := databaseInit.UserDB.Table("article_options").
			Where("vote_article_id = ?", article.ArticleID).
			Select("id, option_content, option_votes_num, sort_order").
			Order("sort_order ASC").
			Find(&options).Error; err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"error": "查询文评选项失败: " + err.Error(),
			})
			return
		}

		// 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).
			Find(&userVotes).Error

		// 标记用户是否投票及所有投票的选项ID
		userHasVoted := false
		var votedOptionIDs []int64 = []int64{}

		if voteErr == nil && len(userVotes) > 0 {
			userHasVoted = true
			// 收集所有投票的选项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(),
			})
			return
		}

		// 4.3 格式化选项数据（标记用户是否投了该选项）
		var optionList []ArticleOption
		for _, opt := range options {
			// 检查当前选项是否是用户投票的选项之一
			isVoted := false
			for _, votedID := range votedOptionIDs {
				if opt.ID == votedID {
					isVoted = true
					break
				}
			}

			optionList = append(optionList, ArticleOption{
				ID:            opt.ID,
				Name:          opt.OptionName,
				Votes:         opt.VoteCount,
				IsVotedByUser: isVoted, // 标记当前用户是否投了这个选项
			})
		}

		// 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,   // 整体标记用户是否投过票
			VotedOptionIDs: votedOptionIDs, // 记录用户投票的所有选项ID（未投票则为空数组）
		}

		// 4.5 加入结果集
		result[article.Title] = articleData
	}

	c.JSON(http.StatusOK, result)
}