515 lines
11 KiB
Go
515 lines
11 KiB
Go
package controller
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
"io"
|
||
"math/rand"
|
||
"mime/multipart"
|
||
"strconv"
|
||
"time"
|
||
"vote-admin-video-api/api/dao"
|
||
"vote-admin-video-api/api/dto"
|
||
"vote-admin-video-api/api/model"
|
||
"vote-admin-video-api/api/requests"
|
||
"vote-admin-video-api/api/responses"
|
||
"vote-admin-video-api/extend/aliyun"
|
||
"vote-admin-video-api/global"
|
||
"vote-admin-video-api/utils"
|
||
)
|
||
|
||
type Video struct{}
|
||
|
||
// GetVideoPage 获取视频列表-分页
|
||
func (r *Video) GetVideoPage(c *gin.Context) {
|
||
VideoRequest := requests.VideoRequest{}
|
||
req := VideoRequest.GetVideoPage
|
||
if err := c.ShouldBindJSON(&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
|
||
}
|
||
|
||
if req.Order != nil {
|
||
if req.Order.VoteNum != "" {
|
||
if req.Order.VoteNum != "desc" && req.Order.VoteNum != "asc" {
|
||
responses.FailWithMessage("排序字段错误", c)
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
// 获取数据
|
||
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)
|
||
|
||
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.GetVideoById(videoId)
|
||
if err != nil {
|
||
responses.FailWithMessage("视频错误", c)
|
||
return
|
||
}
|
||
|
||
// 获取作者数据
|
||
VideoAuthorDao := dao.VideoAuthorDao{}
|
||
maps := make(map[string]interface{})
|
||
maps["video_id"] = videoId
|
||
VideoAuthors, err := VideoAuthorDao.GetVideoAuthorPreloadList(maps)
|
||
if err != nil {
|
||
responses.FailWithMessage("视频错误", c)
|
||
return
|
||
}
|
||
|
||
// 获取排名
|
||
rank, _ := VideoDao.GetVideoRank(video.VideoId)
|
||
|
||
// 处理返回值
|
||
g := dto.GetVideoDto(video)
|
||
|
||
// 加载数据-作者
|
||
g.LoadVideoAuthor(VideoAuthors)
|
||
|
||
// 加载数据-作者排名
|
||
g.LoadRank(rank)
|
||
|
||
responses.OkWithData(g, c)
|
||
}
|
||
|
||
// PutVideo 修改视频详情
|
||
func (r *Video) PutVideo(c *gin.Context) {
|
||
VideoRequest := requests.VideoRequest{}
|
||
req := VideoRequest.PutVideo
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
// 参数验证
|
||
if err := global.Validate.Struct(req); err != nil {
|
||
responses.FailWithMessage(utils.Translate(err), c)
|
||
return
|
||
}
|
||
|
||
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
|
||
}
|
||
|
||
if req.VoteNum != nil {
|
||
if *req.VoteNum < 0 {
|
||
responses.FailWithMessage("票数需大于0", c)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 获取订单数据
|
||
videoDao := dao.VideoDao{}
|
||
video, err := videoDao.GetVideoById(videoId)
|
||
if err != nil {
|
||
responses.FailWithMessage("视频异常", c)
|
||
return
|
||
}
|
||
|
||
// 开始事务
|
||
tx := global.Db.Begin()
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
tx.Rollback()
|
||
fmt.Println(r)
|
||
}
|
||
}()
|
||
|
||
videoData := make(map[string]interface{})
|
||
|
||
// 视频标题
|
||
if video.VideoTitle != req.VideoTitle {
|
||
videoData["video_title"] = req.VideoTitle
|
||
}
|
||
|
||
// 视频状态
|
||
if video.VideoStatus != req.VideoStatus {
|
||
videoData["video_status"] = req.VideoStatus
|
||
}
|
||
|
||
if req.VoteNum != nil {
|
||
if video.VoteNum != *req.VoteNum {
|
||
videoData["vote_num"] = *req.VoteNum
|
||
}
|
||
}
|
||
|
||
// 视频编号
|
||
if video.VideoNo != req.VideoNo {
|
||
videoData["video_no"] = req.VideoNo
|
||
}
|
||
|
||
// 视频内容
|
||
if video.VideoContent != req.VideoContent {
|
||
videoData["video_content"] = req.VideoContent
|
||
}
|
||
|
||
// 修改
|
||
if len(videoData) > 0 {
|
||
err = videoDao.EditVideoById(tx, videoId, videoData)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 删除作者
|
||
VideoAuthorDao := dao.VideoAuthorDao{}
|
||
maps := make(map[string]interface{})
|
||
maps["video_id"] = videoId
|
||
err = VideoAuthorDao.DeleteVideoAuthor(tx, maps)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
baseHospitalDao := dao.BaseHospitalDao{}
|
||
// 新增作者
|
||
for _, author := range req.VideoAuthor {
|
||
if author.HospitalId == "" {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("请选择作者所属医院", c)
|
||
return
|
||
}
|
||
|
||
if author.AuthorName == "" {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("请输入作者名称", c)
|
||
return
|
||
}
|
||
|
||
// 检测医院
|
||
hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId)
|
||
if err != nil || baseHospital == nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("医院错误", c)
|
||
return
|
||
}
|
||
|
||
videoAuthor := &model.VideoAuthor{
|
||
VideoId: videoId,
|
||
AuthorName: author.AuthorName,
|
||
AuthorAvatar: author.AuthorAvatar,
|
||
HospitalId: hospitalId,
|
||
}
|
||
videoAuthor, err = VideoAuthorDao.AddVideoAuthor(tx, videoAuthor)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
}
|
||
|
||
tx.Commit()
|
||
responses.Ok(c)
|
||
}
|
||
|
||
// AddVideo 新增视频详情
|
||
func (r *Video) AddVideo(c *gin.Context) {
|
||
videoRequest := requests.VideoRequest{}
|
||
req := videoRequest.AddVideo
|
||
if err := c.ShouldBindJSON(&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.VoteNum != nil {
|
||
if *req.VoteNum < 0 {
|
||
responses.FailWithMessage("票数需大于0", c)
|
||
return
|
||
}
|
||
}
|
||
|
||
// 开始事务
|
||
tx := global.Db.Begin()
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("操作失败", c)
|
||
return
|
||
}
|
||
}()
|
||
|
||
videoDao := dao.VideoDao{}
|
||
|
||
video := &model.Video{
|
||
VideoTitle: req.VideoTitle,
|
||
VideoStatus: req.VideoStatus,
|
||
VideoNo: req.VideoNo,
|
||
VideoContent: req.VideoContent,
|
||
}
|
||
|
||
if req.VoteNum != nil {
|
||
video.VoteNum = *req.VoteNum
|
||
}
|
||
|
||
// 获取总数量
|
||
maps := make(map[string]interface{})
|
||
total, err := videoDao.GetVideoCount(maps)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
// 生成视频编号
|
||
video.VideoNumber = fmt.Sprintf("%03d", total+1)
|
||
|
||
video, err = videoDao.AddVideo(tx, video)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
// 新增作者
|
||
baseHospitalDao := dao.BaseHospitalDao{}
|
||
videoAuthorDao := dao.VideoAuthorDao{}
|
||
for _, author := range req.VideoAuthor {
|
||
if author.HospitalId == "" {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("请选择作者所属医院", c)
|
||
return
|
||
}
|
||
|
||
if author.AuthorName == "" {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("请输入作者名称", c)
|
||
return
|
||
}
|
||
|
||
// 检测医院
|
||
hospitalId, err := strconv.ParseInt(author.HospitalId, 10, 64)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
baseHospital, err := baseHospitalDao.GetBaseHospitalById(hospitalId)
|
||
if err != nil || baseHospital == nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("医院错误", c)
|
||
return
|
||
}
|
||
|
||
videoAuthor := &model.VideoAuthor{
|
||
VideoId: video.VideoId,
|
||
AuthorName: author.AuthorName,
|
||
AuthorAvatar: author.AuthorAvatar,
|
||
HospitalId: hospitalId,
|
||
}
|
||
videoAuthor, err = videoAuthorDao.AddVideoAuthor(tx, videoAuthor)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
}
|
||
|
||
tx.Commit()
|
||
responses.Ok(c)
|
||
}
|
||
|
||
// PutVideoStatus 操作视频状态
|
||
func (r *Video) PutVideoStatus(c *gin.Context) {
|
||
videoRequest := requests.VideoRequest{}
|
||
req := videoRequest.PutVideoStatus
|
||
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
|
||
}
|
||
|
||
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.GetVideoById(videoId)
|
||
if err != nil {
|
||
responses.FailWithMessage("视频异常", c)
|
||
return
|
||
}
|
||
|
||
// 开始事务
|
||
tx := global.Db.Begin()
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
tx.Rollback()
|
||
}
|
||
}()
|
||
|
||
videoData := make(map[string]interface{})
|
||
if req.VideoStatus != video.VideoStatus {
|
||
videoData["video_status"] = req.VideoStatus
|
||
}
|
||
|
||
if len(videoData) > 0 {
|
||
err = videoDao.EditVideoById(tx, videoId, videoData)
|
||
if err != nil {
|
||
tx.Rollback()
|
||
responses.FailWithMessage("修改失败", c)
|
||
return
|
||
}
|
||
}
|
||
|
||
tx.Commit()
|
||
responses.Ok(c)
|
||
}
|
||
|
||
// UploadAuthorAvatar 上传作者头像
|
||
func (r *Video) UploadAuthorAvatar(c *gin.Context) {
|
||
file, err := c.FormFile("file")
|
||
if err != nil {
|
||
responses.FailWithMessage("文件错误", c)
|
||
return
|
||
}
|
||
|
||
f, err := file.Open()
|
||
if err != nil {
|
||
responses.FailWithMessage("文件错误", c)
|
||
return
|
||
}
|
||
defer func(f multipart.File) {
|
||
err := f.Close()
|
||
if err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
}(f)
|
||
|
||
// 读取文件内容到字节切片
|
||
fileBytes, err := io.ReadAll(f)
|
||
if err != nil {
|
||
responses.FailWithMessage("文件错误", c)
|
||
return
|
||
}
|
||
|
||
// 验证文件类型（只允许图片）
|
||
fileType := "jpg"
|
||
if file.Filename != "" {
|
||
fileType = utils.GetExtension(file.Filename)
|
||
if fileType == "" {
|
||
fileType = "jpg"
|
||
}
|
||
}
|
||
|
||
// 验证是否为图片格式
|
||
allowedTypes := map[string]bool{
|
||
"jpg": true,
|
||
"jpeg": true,
|
||
"png": true,
|
||
"gif": true,
|
||
"webp": true,
|
||
}
|
||
if !allowedTypes[fileType] {
|
||
responses.FailWithMessage("只支持图片格式（jpg、jpeg、png、gif、webp）", c)
|
||
return
|
||
}
|
||
|
||
// 生成OSS路径
|
||
now := time.Now()
|
||
dateTimeString := now.Format("20060102150405")
|
||
rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
ossPath := "static/author_avatar/" + fmt.Sprintf("%d", rand.Intn(9000)+1000) + dateTimeString + "." + fileType
|
||
|
||
// 上传到OSS
|
||
_, err = aliyun.PutObjectByte(ossPath, fileBytes)
|
||
if err != nil {
|
||
responses.FailWithMessage(err.Error(), c)
|
||
return
|
||
}
|
||
|
||
// 返回上传后的URL
|
||
result := make(map[string]interface{})
|
||
result["url"] = utils.AddOssDomain("/" + ossPath)
|
||
result["path"] = "/" + ossPath
|
||
|
||
responses.OkWithData(result, c)
|
||
}
|