toutoukan/controllers/article/getarticle.go

package article

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

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

// ArticleList 数据库文章记录结构体
// 添加了与 ArticleOption 的关联
type ArticleList struct {
	ArticleID      int64     `gorm:"column:articleId;primaryKey;autoIncrement"`
	PublishUserID  string    `gorm:"column:publish_user_id"`
	Title          string    `gorm:"column:title"`
	VoteType       string    `gorm:"column:vote_type"`
	EndTime        time.Time `gorm:"column:end_time"`
	IsEnded        bool      `gorm:"column:is_ended"`
	TotalVotersNum int       `gorm:"column:total_voters_num"`
	CreateTime     time.Time `gorm:"column:create_time"`
	// 这是关键：定义 Has Many 关联，表示一篇文章有多个选项
	Options []ArticleOption `gorm:"foreignKey:VoteArticleID;references:ArticleID"`
}

// ArticleOption 文评选项结构
type ArticleOption struct {
	ID             int64  `gorm:"column:id"`
	VoteArticleID  int64  `gorm:"column:vote_article_id"` // 外键，关联 ArticleList
	OptionContent  string `gorm:"column:option_content"`
	OptionVotesNum int    `gorm:"column:option_votes_num"`
	SortOrder      int    `gorm:"column:sort_order"`
}

// UserVote 用户投票记录表结构

// ArticleOptionResp 格式化后的选项响应结构
type ArticleOptionResp struct {
	ID            int64  `json:"id"`
	Name          string `json:"name"`
	Votes         int    `json:"votes"`
	IsVotedByUser bool   `json:"is_voted"`
}

// ArticleResponse 单个文评的响应结构
type ArticleResponse struct {
	ID             int64               `json:"article_id"`
	Title          string              `json:"article_title"`
	VoteType       string              `json:"vote_type"`
	TotalVoters    int                 `json:"total_voters"`
	EndTime        string              `json:"end_time"`
	IsEnded        bool                `json:"is_ended"`
	PublisherID    string              `json:"publisher_id"`
	CreateTime     string              `json:"create_time"`
	Options        []ArticleOptionResp `json:"options"`
	UserHasVoted   bool                `json:"user_has_voted"`
	VotedOptionIDs []int64             `json:"user_voted_option_ids"`
}

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

// 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. 使用 Preload 一次性查询所有文章和它们的选项
	var articles []ArticleList
	if err := databaseInit.UserDB.Model(&ArticleList{}).
		Preload("Options", func(db *gorm.DB) *gorm.DB {
			return db.Order("sort_order ASC")
		}).
		Find(&articles).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "查询文评失败: " + err.Error(),
		})
		return
	}

	// 3. 一次性查询当前用户的所有投票记录，并构建一个 map 方便查找
	var userAllVotes []UserVote
	if err := databaseInit.UserDB.Table("user_votes").Where("user_id = ?", req.Uid).Find(&userAllVotes).Error; err != nil && err != gorm.ErrRecordNotFound {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "查询用户投票记录失败: " + err.Error(),
		})
		return
	}
	// 投票记录 map：key: articleId, value: []optionId
	userVotesMap := make(map[int64][]int64)
	for _, vote := range userAllVotes {
		userVotesMap[vote.VoteArticleID] = append(userVotesMap[vote.VoteArticleID], vote.OptionID)
	}

	// 4. 组装最终响应数据
	var responseList []ArticleResponse
	for _, article := range articles {
		votedOptionIDs, userHasVoted := userVotesMap[article.ArticleID]

		// 格式化选项数据
		var optionList []ArticleOptionResp
		for _, opt := range article.Options {
			isVoted := false
			for _, votedID := range votedOptionIDs {
				if opt.ID == votedID {
					isVoted = true
					break
				}
			}
			optionList = append(optionList, ArticleOptionResp{
				ID:            opt.ID,
				Name:          opt.OptionContent,
				Votes:         opt.OptionVotesNum,
				IsVotedByUser: isVoted,
			})
		}

		articleData := ArticleResponse{
			ID:             article.ArticleID,
			Title:          article.Title,
			VoteType:       article.VoteType,
			TotalVoters:    article.TotalVotersNum,
			EndTime:        article.EndTime.Format("2006-01-02 15:04:05"),
			IsEnded:        article.IsEnded,
			PublisherID:    article.PublishUserID,
			CreateTime:     article.CreateTime.Format("2006-01-02 15:04:05"),
			Options:        optionList,
			UserHasVoted:   userHasVoted,
			VotedOptionIDs: votedOptionIDs,
		}
		responseList = append(responseList, articleData)
	}

	c.JSON(http.StatusOK, gin.H{
		"success": true,
		"message": fmt.Sprintf("查询成功，共 %d 篇文章", len(responseList)),
		"data":    responseList,
	})
}