更新投票功能
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
|
||||
@@ -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{
|
||||
|
||||
Reference in New Issue
Block a user