gdxz/vote-api
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初始化提交

Browse Source
This commit is contained in:
admin
2024-08-28 18:31:18 +08:00
parent 9c1e56d5af
commit 70bac9077e
27 changed files with 1608 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/controller/Article.go
+289
View File
@@ -0,0 +1,289 @@
package controller
import (
"fmt"
"github.com/gin-gonic/gin"
"net/http"
"strconv"
"time"
"vote-api/api/dao"
"vote-api/api/dto"
"vote-api/api/model"
"vote-api/api/requests"
"vote-api/api/responses"
"vote-api/api/service"
"vote-api/consts"
"vote-api/global"
"vote-api/utils"
)
type Article struct{}
// GetArticlePage 获取图文列表-分页
func (r *Article) GetArticlePage(c *gin.Context) {
articleRequest := requests.ArticleRequest{}
req := articleRequest.GetArticlePage
if err := c.ShouldBind(&req); err != nil {
responses.FailWithMessage(err.Error(), c)
return
}
// 参数验证
if err := global.Validate.Struct(req); err != nil {
responses.FailWithMessage(utils.Translate(err), c)
return
}
if req.Page == 0 {
req.Page = 1
}
if req.PageSize == 0 {
req.PageSize = 20
}
// 获取数据
articleDao := dao.ArticleDao{}
articles, total, err := articleDao.GetArticlePageSearch(req, req.Page, req.PageSize)
if err != nil {
responses.FailWithMessage(err.Error(), c)
return
}
// 处理返回值
g := dto.GetArticleListDto(articles)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
for _, articleDto := range g {
articleId, err := strconv.ParseInt(articleDto.ArticleId, 10, 64)
if err != nil {
continue
}
articleDto.IsVote = userService.CheckUserVoteDay(userId, articleId, 1)
}
}
result := make(map[string]interface{})
result["page"] = req.Page
result["page_size"] = req.PageSize
result["total"] = total
result["data"] = g
responses.OkWithData(result, c)
}
// GetArticle 获取图文详情
func (r *Article) GetArticle(c *gin.Context) {
id := c.Param("article_id")
if id == "" {
responses.FailWithMessage("缺少参数", c)
return
}
// 将 id 转换为 int64 类型
articleId, err := strconv.ParseInt(id, 10, 64)
if err != nil {
responses.Fail(c)
return
}
// 获取数据
articleDao := dao.ArticleDao{}
article, err := articleDao.GetArticlePreloadById(articleId)
if err != nil {
responses.FailWithMessage("文章错误", c)
return
}
// 获取排名
rank, _ := articleDao.GetArticleRank(article.ArticleId)
// 处理返回值
g := dto.GetArticleDto(article)
// 加载数据-作者
g.LoadArticleAuthor(article.ArticleAuthor)
// 加载数据-作者排名
g.LoadRank(rank)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
isVote := userService.CheckUserVoteDay(userId, articleId, 1)
// 加载数据-投票状态
g.LoadVoteStatus(isVote)
}
responses.OkWithData(g, c)
}
// GetArticleRankList 获取文章排名列表
func (r *Article) GetArticleRankList(c *gin.Context) {
// 检测投票有效期
systemTimeService := service.SystemTimeService{}
isValid := systemTimeService.CheckVoteValidStatus()
if isValid == false {
responses.OkWithData(nil, c)
return
}
// 获取数据
articleDao := dao.ArticleDao{}
maps := make(map[string]interface{})
maps["article_status"] = 1
articles, err := articleDao.GetArticleOrderList(maps, "vote_num desc", 15)
if err != nil {
responses.OkWithData(nil, c)
return
}
// 处理返回值
g := dto.GetArticleListDto(articles)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
for _, articleDto := range g {
articleId, err := strconv.ParseInt(articleDto.ArticleId, 10, 64)
if err != nil {
continue
}
articleDto.IsVote = userService.CheckUserVoteDay(userId, articleId, 1)
}
}
responses.OkWithData(g, c)
}
// AddArticleVote 文章投票
func (r *Article) AddArticleVote(c *gin.Context) {
id := c.Param("article_id")
if id == "" {
responses.FailWithMessage("缺少参数", c)
return
}
// 将 id 转换为 int64 类型
articleId, err := strconv.ParseInt(id, 10, 64)
if err != nil {
responses.Fail(c)
return
}
userId := c.GetInt64("UserId")
if userId == 0 {
c.JSON(http.StatusUnauthorized, gin.H{
"message": "请登录后投票",
"code": consts.TokenError,
"data": "",
})
return
}
// 检测并发请求
redisKey := "AddArticleVote" + fmt.Sprintf("%d", userId) + fmt.Sprintf("%d", articleId)
res, _ := global.Redis.Get(c, redisKey).Result()
if res != "" {
responses.FailWithMessage("请勿重复操作", c)
return
}
defer func(redisKey string) {
global.Redis.Del(c, redisKey)
}(redisKey)
// 添加缓存
_, err = global.Redis.Set(c, redisKey, "1", (1)*time.Second).Result()
if err != nil {
responses.FailWithMessage("投票失败", c)
return
}
// 获取数据
articleDao := dao.ArticleDao{}
article, err := articleDao.GetArticleById(articleId)
if err != nil || article == nil {
responses.FailWithMessage("非法数据", c)
return
}
// 检测投票有效期
systemTimeService := service.SystemTimeService{}
isValid := systemTimeService.CheckVoteValidStatus()
if isValid == false {
responses.FailWithMessage("投票已结束", c)
return
}
// 检测用户今日是否投票
userService := service.UserService{}
isVote := userService.CheckUserVoteDay(userId, articleId, 1)
if isVote == true {
responses.FailWithMessage("请勿重复投票", c)
return
}
// 开始事务
tx := global.Db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
utils.LogJsonErr("投票失败", r)
responses.FailWithMessage("投票失败", c)
return
}
}()
// 增加投票数
err = articleDao.Inc(tx, articleId, "vote_num", 1)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 增加投票记录
nowDay, _ := time.Parse("2006-01-02", time.Now().Format("2006-01-02"))
votedAt := model.LocalTime(nowDay)
articleVoteDay := &model.ArticleVoteDay{
ArticleId: articleId,
UserId: userId,
VotedAt: &votedAt,
}
articleVoteDayDao := dao.ArticleVoteDayDao{}
articleVoteDay, err = articleVoteDayDao.AddArticleVoteDay(tx, articleVoteDay)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 增加数据表-投票数量
dataDao := dao.DataDao{}
err = dataDao.Inc(tx, 1, "vote_num", 1)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 新增投票缓存
result := userService.AddUserVoteDayCache(userId, articleId, 1)
if result == false {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
tx.Commit()
responses.Ok(c)
}
api/controller/Base.go
+4 -1
View File
@@ -2,5 +2,8 @@ package controller
// Api api接口
type Api struct {
Login // 登录
Login // 登录
Article // 图文
Video // 视频
Public // 公共方法
}
api/controller/Public.go
+38
View File
@@ -0,0 +1,38 @@
package controller
import (
"github.com/gin-gonic/gin"
"vote-api/api/dao"
"vote-api/api/responses"
"vote-api/global"
"vote-api/utils"
)
type Public struct {
}
// AddBrowse 增加浏览数量
func (r *Public) AddBrowse(c *gin.Context) {
// 开始事务
tx := global.Db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
utils.LogJsonErr("增加浏览数量失败", r)
responses.Ok(c)
return
}
}()
// 增加数据表-投票数量
dataDao := dao.DataDao{}
err := dataDao.Inc(tx, 1, "view_num", 1)
if err != nil {
tx.Rollback()
responses.Ok(c)
return
}
tx.Commit()
responses.Ok(c)
}
api/controller/Video.go
+289
View File
@@ -0,0 +1,289 @@
package controller
import (
"fmt"
"github.com/gin-gonic/gin"
"net/http"
"strconv"
"time"
"vote-api/api/dao"
"vote-api/api/dto"
"vote-api/api/model"
"vote-api/api/requests"
"vote-api/api/responses"
"vote-api/api/service"
"vote-api/consts"
"vote-api/global"
"vote-api/utils"
)
type Video struct{}
// GetVideoPage 获取视频列表-分页
func (r *Video) GetVideoPage(c *gin.Context) {
videoRequest := requests.VideoRequest{}
req := videoRequest.GetVideoPage
if err := c.ShouldBind(&req); err != nil {
responses.FailWithMessage(err.Error(), c)
return
}
// 参数验证
if err := global.Validate.Struct(req); err != nil {
responses.FailWithMessage(utils.Translate(err), c)
return
}
if req.Page == 0 {
req.Page = 1
}
if req.PageSize == 0 {
req.PageSize = 20
}
// 获取数据
videoDao := dao.VideoDao{}
videos, total, err := videoDao.GetVideoPageSearch(req, req.Page, req.PageSize)
if err != nil {
responses.FailWithMessage(err.Error(), c)
return
}
// 处理返回值
g := dto.GetVideoListDto(videos)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
for _, videoDto := range g {
videoId, err := strconv.ParseInt(videoDto.VideoId, 10, 64)
if err != nil {
continue
}
videoDto.IsVote = userService.CheckUserVoteDay(userId, videoId, 2)
}
}
result := make(map[string]interface{})
result["page"] = req.Page
result["page_size"] = req.PageSize
result["total"] = total
result["data"] = g
responses.OkWithData(result, c)
}
// GetVideo 获取视频详情
func (r *Video) GetVideo(c *gin.Context) {
id := c.Param("video_id")
if id == "" {
responses.FailWithMessage("缺少参数", c)
return
}
// 将 id 转换为 int64 类型
videoId, err := strconv.ParseInt(id, 10, 64)
if err != nil {
responses.Fail(c)
return
}
// 获取数据
videoDao := dao.VideoDao{}
video, err := videoDao.GetVideoPreloadById(videoId)
if err != nil {
responses.FailWithMessage("文章错误", c)
return
}
// 获取排名
rank, _ := videoDao.GetVideoRank(video.VideoId)
// 处理返回值
g := dto.GetVideoDto(video)
// 加载数据-作者
g.LoadVideoAuthor(video.VideoAuthor)
// 加载数据-作者排名
g.LoadRank(rank)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
isVote := userService.CheckUserVoteDay(userId, videoId, 2)
// 加载数据-投票状态
g.LoadVoteStatus(isVote)
}
responses.OkWithData(g, c)
}
// GetVideoRankList 获取视频排名列表
func (r *Video) GetVideoRankList(c *gin.Context) {
// 检测投票状态
systemTimeService := service.SystemTimeService{}
isValid := systemTimeService.CheckVoteValidStatus()
if isValid == false {
responses.OkWithData(nil, c)
return
}
// 获取数据
videoDao := dao.VideoDao{}
maps := make(map[string]interface{})
maps["video_status"] = 1
videos, err := videoDao.GetVideoOrderList(maps, "vote_num desc", 15)
if err != nil {
responses.OkWithData(nil, c)
return
}
// 处理返回值
g := dto.GetVideoListDto(videos)
// 检测用户今日是否投票
userId := c.GetInt64("UserId")
if userId != 0 {
userService := service.UserService{}
for _, videoDto := range g {
videoId, err := strconv.ParseInt(videoDto.VideoId, 10, 64)
if err != nil {
continue
}
videoDto.IsVote = userService.CheckUserVoteDay(userId, videoId, 2)
}
}
responses.OkWithData(g, c)
}
// AddVideoVote 视频投票
func (r *Video) AddVideoVote(c *gin.Context) {
id := c.Param("video_id")
if id == "" {
responses.FailWithMessage("缺少参数", c)
return
}
// 将 id 转换为 int64 类型
videoId, err := strconv.ParseInt(id, 10, 64)
if err != nil {
responses.Fail(c)
return
}
userId := c.GetInt64("UserId")
if userId == 0 {
c.JSON(http.StatusUnauthorized, gin.H{
"message": "请登录后投票",
"code": consts.TokenError,
"data": "",
})
return
}
// 检测并发请求
redisKey := "AddVideoVote" + fmt.Sprintf("%d", userId) + fmt.Sprintf("%d", videoId)
res, _ := global.Redis.Get(c, redisKey).Result()
if res != "" {
responses.FailWithMessage("请勿重复操作", c)
return
}
defer func(redisKey string) {
global.Redis.Del(c, redisKey)
}(redisKey)
// 添加缓存
_, err = global.Redis.Set(c, redisKey, "1", (1)*time.Second).Result()
if err != nil {
responses.FailWithMessage("投票失败", c)
return
}
// 获取数据
videoDao := dao.VideoDao{}
video, err := videoDao.GetVideoById(videoId)
if err != nil || video == nil {
responses.FailWithMessage("非法数据", c)
return
}
// 检测投票有效期
systemTimeService := service.SystemTimeService{}
isValid := systemTimeService.CheckVoteValidStatus()
if isValid == false {
responses.FailWithMessage("投票已结束", c)
return
}
// 检测用户今日是否投票
userService := service.UserService{}
isVote := userService.CheckUserVoteDay(userId, videoId, 2)
if isVote == true {
responses.FailWithMessage("请勿重复投票", c)
return
}
// 开始事务
tx := global.Db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
utils.LogJsonErr("投票失败", r)
responses.FailWithMessage("投票失败", c)
return
}
}()
// 增加投票数
err = videoDao.Inc(tx, videoId, "vote_num", 1)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 增加投票记录
nowDay, _ := time.Parse("2006-01-02", time.Now().Format("2006-01-02"))
votedAt := model.LocalTime(nowDay)
videoVoteDay := &model.VideoVoteDay{
VideoId: videoId,
UserId: userId,
VotedAt: &votedAt,
}
videoVoteDayDao := dao.VideoVoteDayDao{}
videoVoteDay, err = videoVoteDayDao.AddVideoVoteDay(tx, videoVoteDay)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 增加数据表-投票数量
dataDao := dao.DataDao{}
err = dataDao.Inc(tx, 1, "vote_num", 1)
if err != nil {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
// 新增投票缓存
result := userService.AddUserVoteDayCache(userId, videoId, 2)
if result == false {
tx.Rollback()
responses.FailWithMessage("投票失败", c)
return
}
tx.Commit()
responses.Ok(c)
}