feat:聊天信息支持 文件、图片、语音发送,增加一键启停脚本

This commit is contained in:
bujinyuan
2026-04-20 11:53:27 +08:00
parent 1a5aac9998
commit 0c1540bdee
116 changed files with 20277 additions and 144 deletions

View File

@@ -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 := "服务器内部错误"

View File

@@ -12,5 +12,7 @@ func RegisterRoutes(r *gin.Engine, ctrl *controller.FileController, jwtAuth gin.
authed.Use(jwtAuth)
{
authed.POST("/upload", ctrl.Upload)
authed.POST("/upload/image", ctrl.UploadImage)
authed.POST("/upload/voice", ctrl.UploadVoice)
}
}

View File

@@ -3,14 +3,20 @@
package service
import (
"bytes"
"context"
"errors"
"fmt"
"image"
_ "image/gif"
_ "image/jpeg"
_ "image/png"
"mime/multipart"
"path/filepath"
"strings"
"time"
"github.com/disintegration/imaging"
"github.com/echochat/backend/config"
"github.com/echochat/backend/pkg/logs"
"github.com/google/uuid"
@@ -19,8 +25,13 @@ import (
)
var (
ErrFileOpen = errors.New("打开上传文件失败")
ErrUploadFailed = errors.New("文件上传失败")
ErrFileOpen = errors.New("打开上传文件失败")
ErrUploadFailed = errors.New("文件上传失败")
ErrFileTooLarge = errors.New("文件大小超出限制")
ErrInvalidImage = errors.New("无效的图片文件")
ErrInvalidVoice = errors.New("不支持的语音格式,仅支持 mp3/wav/aac/m4a")
ErrVoiceTooLong = errors.New("语音时长不能超过 60 秒")
ErrThumbnailGen = errors.New("缩略图生成失败")
)
// FileService 文件上传服务
@@ -37,13 +48,41 @@ func NewFileService(client *minio.Client, cfg *config.MinioConfig) *FileService
}
}
// UploadResult 文件上传结果
// UploadResult 通用文件上传结果
type UploadResult struct {
URL string `json:"url"`
FileName string `json:"file_name"`
Size int64 `json:"size"`
}
// ImageUploadResult 图片上传结果(含缩略图和尺寸)
type ImageUploadResult struct {
URL string `json:"url"`
ThumbnailURL string `json:"thumbnail_url"`
FileName string `json:"file_name"`
Size int64 `json:"size"`
Width int `json:"width"`
Height int `json:"height"`
MimeType string `json:"mime_type"`
}
// VoiceUploadResult 语音上传结果(含时长)
type VoiceUploadResult struct {
URL string `json:"url"`
FileName string `json:"file_name"`
Size int64 `json:"size"`
Duration int `json:"duration"`
MimeType string `json:"mime_type"`
}
// allowedVoiceExts 允许的语音文件扩展名
var allowedVoiceExts = map[string]bool{
".mp3": true,
".wav": true,
".aac": true,
".m4a": true,
}
// Upload 上传文件到 MinIO返回文件访问 URL
// 文件按日期目录组织uploads/2026/03/04/{uuid}.{ext}
func (s *FileService) Upload(ctx context.Context, fileHeader *multipart.FileHeader) (*UploadResult, error) {
@@ -76,11 +115,7 @@ func (s *FileService) Upload(ctx context.Context, fileHeader *multipart.FileHead
return nil, ErrUploadFailed
}
scheme := "http"
if s.minioCfg.UseSSL {
scheme = "https"
}
url := fmt.Sprintf("%s://%s/%s/%s", scheme, s.minioCfg.Endpoint, s.minioCfg.Bucket, objectName)
url := s.buildURL(objectName)
return &UploadResult{
URL: url,
@@ -88,3 +123,142 @@ func (s *FileService) Upload(ctx context.Context, fileHeader *multipart.FileHead
Size: fileHeader.Size,
}, nil
}
// UploadImage 上传图片并生成缩略图
// 原图和缩略图均存储到 MinIO缩略图宽度 200px 等比缩放JPEG quality 80
func (s *FileService) UploadImage(ctx context.Context, fileHeader *multipart.FileHeader) (*ImageUploadResult, error) {
funcName := "service.file_service.UploadImage"
logs.Info(ctx, funcName, "上传图片",
zap.String("file_name", fileHeader.Filename),
zap.Int64("size", fileHeader.Size))
file, err := fileHeader.Open()
if err != nil {
logs.Error(ctx, funcName, "打开图片文件失败", zap.Error(err))
return nil, ErrFileOpen
}
defer file.Close()
img, _, err := image.Decode(file)
if err != nil {
logs.Error(ctx, funcName, "解码图片失败", zap.Error(err))
return nil, ErrInvalidImage
}
bounds := img.Bounds()
width := bounds.Dx()
height := bounds.Dy()
now := time.Now()
fileID := uuid.New().String()
dateDir := now.Format("2006/01/02")
// 上传原图(回到文件开头)
if _, err = file.Seek(0, 0); err != nil {
logs.Error(ctx, funcName, "重置文件指针失败", zap.Error(err))
return nil, ErrFileOpen
}
ext := strings.ToLower(filepath.Ext(fileHeader.Filename))
objectName := fmt.Sprintf("uploads/%s/%s%s", dateDir, fileID, ext)
contentType := fileHeader.Header.Get("Content-Type")
if contentType == "" {
contentType = "image/jpeg"
}
_, err = s.minioClient.PutObject(ctx, s.minioCfg.Bucket, objectName, file, fileHeader.Size, minio.PutObjectOptions{
ContentType: contentType,
})
if err != nil {
logs.Error(ctx, funcName, "上传原图到 MinIO 失败", zap.Error(err))
return nil, ErrUploadFailed
}
// 生成缩略图:宽度 200px等比缩放
thumb := imaging.Resize(img, 200, 0, imaging.Lanczos)
var thumbBuf bytes.Buffer
if err = imaging.Encode(&thumbBuf, thumb, imaging.JPEG, imaging.JPEGQuality(80)); err != nil {
logs.Error(ctx, funcName, "编码缩略图失败", zap.Error(err))
return nil, ErrThumbnailGen
}
thumbObjectName := fmt.Sprintf("uploads/%s/%s_thumb.jpg", dateDir, fileID)
_, err = s.minioClient.PutObject(ctx, s.minioCfg.Bucket, thumbObjectName, &thumbBuf, int64(thumbBuf.Len()), minio.PutObjectOptions{
ContentType: "image/jpeg",
})
if err != nil {
logs.Error(ctx, funcName, "上传缩略图到 MinIO 失败", zap.Error(err))
return nil, ErrThumbnailGen
}
return &ImageUploadResult{
URL: s.buildURL(objectName),
ThumbnailURL: s.buildURL(thumbObjectName),
FileName: fileHeader.Filename,
Size: fileHeader.Size,
Width: width,
Height: height,
MimeType: contentType,
}, nil
}
// UploadVoice 上传语音文件,校验格式和前端传入的时长
// duration 由前端录音回调提供(秒),后端仅做范围校验
func (s *FileService) UploadVoice(ctx context.Context, fileHeader *multipart.FileHeader, duration int) (*VoiceUploadResult, error) {
funcName := "service.file_service.UploadVoice"
logs.Info(ctx, funcName, "上传语音",
zap.String("file_name", fileHeader.Filename),
zap.Int64("size", fileHeader.Size),
zap.Int("duration", duration))
ext := strings.ToLower(filepath.Ext(fileHeader.Filename))
if !allowedVoiceExts[ext] {
return nil, ErrInvalidVoice
}
if duration > 60 {
return nil, ErrVoiceTooLong
}
file, err := fileHeader.Open()
if err != nil {
logs.Error(ctx, funcName, "打开语音文件失败", zap.Error(err))
return nil, ErrFileOpen
}
defer file.Close()
now := time.Now()
objectName := fmt.Sprintf("uploads/%s/%s%s", now.Format("2006/01/02"), uuid.New().String(), ext)
contentType := fileHeader.Header.Get("Content-Type")
if contentType == "" {
contentType = "audio/mpeg"
}
_, err = s.minioClient.PutObject(ctx, s.minioCfg.Bucket, objectName, file, fileHeader.Size, minio.PutObjectOptions{
ContentType: contentType,
})
if err != nil {
logs.Error(ctx, funcName, "上传语音到 MinIO 失败", zap.Error(err))
return nil, ErrUploadFailed
}
return &VoiceUploadResult{
URL: s.buildURL(objectName),
FileName: fileHeader.Filename,
Size: fileHeader.Size,
Duration: duration,
MimeType: contentType,
}, nil
}
// buildURL 根据 MinIO 配置构建文件访问 URL
func (s *FileService) buildURL(objectName string) string {
scheme := "http"
if s.minioCfg.UseSSL {
scheme = "https"
}
return fmt.Sprintf("%s://%s/%s/%s", scheme, s.minioCfg.Endpoint, s.minioCfg.Bucket, objectName)
}