toutoukan/controllers/article/getarticle.go

package article

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

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

// ArticleList 数据库文章记录结构体
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"`
	Options        []ArticleOption `gorm:"foreignKey:VoteArticleID;references:ArticleID"`
}

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

// 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"`
	PublisherName  string              `json:"publisher_name"`
	CreateTime     string              `json:"create_time"`
	Options        []ArticleOptionResp `json:"options"`
	UserHasVoted   bool                `json:"user_has_voted"`
	VotedOptionIDs []int64             `json:"user_voted_option_ids"`
}

// UserReq 请求参数结构体，允许uid为空
type UserReq struct {
	Uid string `json:"uid" binding:"omitempty"` // 允许uid为空，空字符串也合法
}

// ArticleListget 获取所有文评及选项信息（支持空uid场景）
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 []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. 批量查询所有发布者的用户名（核心修改：获取uid对应的username）
	// 3.1 收集所有发布者的uid（去重）
	uniquePublishUids := make(map[string]struct{})
	for _, article := range articles {
		uniquePublishUids[article.PublishUserID] = struct{}{}
	}
	// 转换为切片便于查询
	publishUidList := make([]string, 0, len(uniquePublishUids))
	for uid := range uniquePublishUids {
		publishUidList = append(publishUidList, uid)
	}

	// 3.2 从用户表查询这些uid对应的用户名（假设用户表为`users`，字段为`user_id`和`username`）
	type userInfo struct {
		UserID   string `gorm:"column:uid"`      // 对应发布者的PublishUserID
		Username string `gorm:"column:username"` // 发布者用户名
	}
	var publishUsers []userInfo
	if len(publishUidList) > 0 {
		if err := databaseInit.UserDB.Table("user_info").
			Where("uid IN (?)", publishUidList).
			Find(&publishUsers).Error; err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{
				"error": "查询发布者信息失败: " + err.Error(),
			})
			return
		}
	}

	// 3.3 构建uid到username的映射（方便快速查询）
	uidToUsername := make(map[string]string)
	for _, user := range publishUsers {
		uidToUsername[user.UserID] = user.Username
	}

	// 4. 查询用户投票记录（原有逻辑，保持不变）
	userVotesMap := make(map[int64][]int64)
	uidValid := req.Uid != ""
	if uidValid {
		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
		}
		for _, vote := range userAllVotes {
			userVotesMap[vote.VoteArticleID] = append(userVotesMap[vote.VoteArticleID], vote.OptionID)
		}
	}

	// 5. 组装响应数据（添加用户名）
	var responseList []ArticleResponse
	for _, article := range articles {
		// 5.1 处理投票状态（原有逻辑）
		var votedOptionIDs []int64
		var userHasVoted bool
		if uidValid {
			votedOptionIDs, userHasVoted = userVotesMap[article.ArticleID]
		} else {
			votedOptionIDs = []int64{}
			userHasVoted = false
		}

		// 5.2 处理选项列表（原有逻辑）
		var optionList []ArticleOptionResp
		for _, opt := range article.Options {
			isVoted := false
			if uidValid {
				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,
			})
		}

		// 5.3 获取发布者用户名（核心修改：从映射中查询）
		publisherName := uidToUsername[article.PublishUserID]
		if publisherName == "" {
			publisherName = "未知用户" // 处理用户不存在的情况
		}

		// 5.4 组装最终响应
		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,
			PublisherName:  publisherName, // 填充用户名
			CreateTime:     article.CreateTime.Format("2006-01-02 15:04:05"),
			Options:        optionList,
			UserHasVoted:   userHasVoted,
			VotedOptionIDs: votedOptionIDs,
		}
		responseList = append(responseList, articleData)
	}

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