feat:聊天信息支持 文件、图片、语音发送,增加一键启停脚本
This commit is contained in:
@@ -2,13 +2,19 @@
|
||||
package controller
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"github.com/echochat/backend/app/file/service"
|
||||
"github.com/echochat/backend/pkg/middleware"
|
||||
"github.com/echochat/backend/pkg/utils"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
const maxUploadSize = 10 << 20 // 10 MB
|
||||
const (
|
||||
maxUploadSize = 50 << 20 // 50 MB(通用上传)
|
||||
maxImageUploadSize = 20 << 20 // 20 MB(图片上传)
|
||||
maxVoiceUploadSize = 5 << 20 // 5 MB(语音上传)
|
||||
)
|
||||
|
||||
// FileController 文件上传控制器
|
||||
type FileController struct {
|
||||
@@ -20,9 +26,9 @@ func NewFileController(fileService *service.FileService) *FileController {
|
||||
return &FileController{fileService: fileService}
|
||||
}
|
||||
|
||||
// Upload 处理文件上传请求
|
||||
// Upload 处理通用文件上传请求
|
||||
// POST /api/v1/upload
|
||||
// 支持 multipart/form-data,字段名为 "file"
|
||||
// 支持 multipart/form-data,字段名为 "file",最大 50MB
|
||||
func (ctl *FileController) Upload(c *gin.Context) {
|
||||
ctx := c.Request.Context()
|
||||
_, ok := middleware.GetCurrentUserID(c)
|
||||
@@ -38,7 +44,7 @@ func (ctl *FileController) Upload(c *gin.Context) {
|
||||
}
|
||||
|
||||
if fileHeader.Size > maxUploadSize {
|
||||
utils.ResponseBadRequest(c, "文件大小不能超过 10MB")
|
||||
utils.ResponseBadRequest(c, "文件大小不能超过 50MB")
|
||||
return
|
||||
}
|
||||
|
||||
@@ -51,14 +57,94 @@ func (ctl *FileController) Upload(c *gin.Context) {
|
||||
utils.ResponseOK(c, result)
|
||||
}
|
||||
|
||||
// UploadImage 处理图片上传请求,自动生成缩略图
|
||||
// POST /api/v1/upload/image
|
||||
// 支持 multipart/form-data,字段名为 "file",最大 20MB
|
||||
// 返回原图 URL + 缩略图 URL + 宽高信息
|
||||
func (ctl *FileController) UploadImage(c *gin.Context) {
|
||||
ctx := c.Request.Context()
|
||||
_, ok := middleware.GetCurrentUserID(c)
|
||||
if !ok {
|
||||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||||
return
|
||||
}
|
||||
|
||||
fileHeader, err := c.FormFile("file")
|
||||
if err != nil {
|
||||
utils.ResponseBadRequest(c, "请选择要上传的图片")
|
||||
return
|
||||
}
|
||||
|
||||
if fileHeader.Size > maxImageUploadSize {
|
||||
utils.ResponseBadRequest(c, "图片大小不能超过 20MB")
|
||||
return
|
||||
}
|
||||
|
||||
result, err := ctl.fileService.UploadImage(ctx, fileHeader)
|
||||
if err != nil {
|
||||
ctl.handleError(c, err, "图片上传失败")
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, result)
|
||||
}
|
||||
|
||||
// UploadVoice 处理语音上传请求,校验格式和时长
|
||||
// POST /api/v1/upload/voice
|
||||
// 支持 multipart/form-data,字段名为 "file",另需 "duration" 字段(秒)
|
||||
// 最大 5MB,时长不超过 60 秒
|
||||
func (ctl *FileController) UploadVoice(c *gin.Context) {
|
||||
ctx := c.Request.Context()
|
||||
_, ok := middleware.GetCurrentUserID(c)
|
||||
if !ok {
|
||||
utils.ResponseUnauthorized(c, "无法获取当前用户信息")
|
||||
return
|
||||
}
|
||||
|
||||
fileHeader, err := c.FormFile("file")
|
||||
if err != nil {
|
||||
utils.ResponseBadRequest(c, "请选择要上传的语音文件")
|
||||
return
|
||||
}
|
||||
|
||||
if fileHeader.Size > maxVoiceUploadSize {
|
||||
utils.ResponseBadRequest(c, "语音文件大小不能超过 5MB")
|
||||
return
|
||||
}
|
||||
|
||||
durationStr := c.PostForm("duration")
|
||||
duration := 0
|
||||
if durationStr != "" {
|
||||
duration, err = strconv.Atoi(durationStr)
|
||||
if err != nil || duration < 0 {
|
||||
utils.ResponseBadRequest(c, "语音时长参数无效")
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
result, err := ctl.fileService.UploadVoice(ctx, fileHeader, duration)
|
||||
if err != nil {
|
||||
ctl.handleError(c, err, "语音上传失败")
|
||||
return
|
||||
}
|
||||
|
||||
utils.ResponseOK(c, result)
|
||||
}
|
||||
|
||||
// handleError 统一业务错误映射
|
||||
// 已知业务错误 → 返回 Service 层定义的具体提示
|
||||
// 未知错误 → 返回 fallbackMsg(未传则默认"服务器内部错误")
|
||||
func (ctl *FileController) handleError(c *gin.Context, err error, fallbackMsg ...string) {
|
||||
switch err {
|
||||
case service.ErrFileOpen:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrUploadFailed:
|
||||
case service.ErrInvalidImage:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrInvalidVoice:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrVoiceTooLong:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrFileTooLarge:
|
||||
utils.ResponseBadRequest(c, err.Error())
|
||||
case service.ErrUploadFailed, service.ErrThumbnailGen:
|
||||
utils.ResponseError(c, err.Error())
|
||||
default:
|
||||
msg := "服务器内部错误"
|
||||
|
||||
Reference in New Issue
Block a user