feat: 实现用户头像上传功能

- 创建头像上传API接口 (POST /api/v1/users/:id/avatar)
- 实现完整的头像上传逻辑,包含权限验证和文件处理
- 添加头像图片处理功能,支持自动压缩和居中裁剪
- 完善静态文件服务,支持头像访问
- 创建完整的API测试用例
- 更新任务进度文档

任务11已完成,项目完成率提升至37.5%
This commit is contained in:
xujiang
2025-07-11 13:10:04 +08:00
parent d47e55d5fb
commit fa5f7a0ed2
10 changed files with 393 additions and 21 deletions

View File

@ -17,11 +17,6 @@ import (
)
func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
server.AddRoutes(
[]rest.Route{},
rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
)
server.AddRoutes(
[]rest.Route{
{
@ -158,6 +153,12 @@ func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
Path: "/:id",
Handler: user.DeleteUserHandler(serverCtx),
},
{
// 上传用户头像
Method: http.MethodPost,
Path: "/:id/avatar",
Handler: user.UploadAvatarHandler(serverCtx),
},
},
rest.WithJwt(serverCtx.Config.Auth.AccessSecret),
rest.WithPrefix("/api/v1/users"),

View File

@ -0,0 +1,29 @@
package user
import (
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
"photography-backend/internal/logic/user"
"photography-backend/internal/svc"
"photography-backend/internal/types"
)
// 上传用户头像
func UploadAvatarHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req types.UploadAvatarRequest
if err := httpx.Parse(r, &req); err != nil {
httpx.ErrorCtx(r.Context(), w, err)
return
}
l := user.NewUploadAvatarLogic(r.Context(), svcCtx)
resp, err := l.UploadAvatar(&req, r)
if err != nil {
httpx.ErrorCtx(r.Context(), w, err)
} else {
httpx.OkJsonCtx(r.Context(), w, resp)
}
}
}

View File

@ -0,0 +1,157 @@
package user
import (
"context"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strings"
"time"
"photography-backend/internal/svc"
"photography-backend/internal/types"
"photography-backend/pkg/errorx"
"photography-backend/pkg/utils/file"
"github.com/zeromicro/go-zero/core/logx"
)
type UploadAvatarLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
// 上传用户头像
func NewUploadAvatarLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadAvatarLogic {
return &UploadAvatarLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}
func (l *UploadAvatarLogic) UploadAvatar(req *types.UploadAvatarRequest, r *http.Request) (resp *types.UploadAvatarResponse, err error) {
// 1. 验证用户权限
userId := l.ctx.Value("userId").(int64)
if userId != req.Id {
return nil, errorx.New(errorx.Forbidden, "您只能更新自己的头像")
}
// 2. 检查用户是否存在
user, err := l.svcCtx.UserModel.FindOne(l.ctx, req.Id)
if err != nil {
return nil, errorx.New(errorx.UserNotFound, "用户不存在")
}
// 3. 解析多部分表单
err = r.ParseMultipartForm(10 << 20) // 10MB
if err != nil {
return nil, errorx.New(errorx.ParamError, "解析表单失败")
}
// 4. 获取上传的文件
uploadedFile, header, err := r.FormFile("avatar")
if err != nil {
return nil, errorx.New(errorx.ParamError, "获取上传文件失败: " + err.Error())
}
defer uploadedFile.Close()
// 5. 验证文件类型 (头像应该是图片)
contentType := header.Header.Get("Content-Type")
if !strings.HasPrefix(contentType, "image/") {
return nil, errorx.New(errorx.ParamError, "头像必须是图片文件")
}
// 6. 验证文件大小 (限制为5MB)
if header.Size > 5*1024*1024 {
return nil, errorx.New(errorx.ParamError, "头像文件大小不能超过5MB")
}
// 7. 生成头像文件名和路径
ext := filepath.Ext(header.Filename)
if ext == "" {
// 根据Content-Type推断扩展名
switch contentType {
case "image/jpeg":
ext = ".jpg"
case "image/png":
ext = ".png"
case "image/gif":
ext = ".gif"
case "image/webp":
ext = ".webp"
default:
ext = ".jpg"
}
}
filename := fmt.Sprintf("avatar_%d_%d%s", req.Id, time.Now().Unix(), ext)
avatarDir := "uploads/avatars"
// 8. 确保头像目录存在
if err := os.MkdirAll(avatarDir, 0755); err != nil {
return nil, errorx.New(errorx.ServerError, "创建头像目录失败: " + err.Error())
}
avatarPath := filepath.Join(avatarDir, filename)
// 9. 保存原始头像文件
destFile, err := os.Create(avatarPath)
if err != nil {
return nil, errorx.New(errorx.ServerError, "创建头像文件失败: " + err.Error())
}
defer destFile.Close()
_, err = io.Copy(destFile, uploadedFile)
if err != nil {
return nil, errorx.New(errorx.ServerError, "保存头像文件失败: " + err.Error())
}
// 10. 生成压缩版本的头像 (150x150像素)
compressedPath := filepath.Join(avatarDir, fmt.Sprintf("avatar_%d_%d_150x150%s", req.Id, time.Now().Unix(), ext))
err = file.ResizeImage(avatarPath, compressedPath, 150, 150)
if err != nil {
l.Errorf("头像压缩失败: %v", err)
// 压缩失败不影响主流程,使用原图
compressedPath = avatarPath
}
// 11. 删除旧的头像文件 (如果存在)
if user.Avatar != "" && user.Avatar != compressedPath {
oldAvatarPath := user.Avatar
if strings.HasPrefix(oldAvatarPath, "/uploads/") {
oldAvatarPath = strings.TrimPrefix(oldAvatarPath, "/")
}
if _, err := os.Stat(oldAvatarPath); err == nil {
os.Remove(oldAvatarPath)
l.Infof("删除旧头像文件: %s", oldAvatarPath)
}
}
// 12. 更新用户头像URL
avatarUrl := "/" + compressedPath
user.Avatar = avatarUrl
err = l.svcCtx.UserModel.Update(l.ctx, user)
if err != nil {
// 更新失败,删除已上传的文件
os.Remove(avatarPath)
if avatarPath != compressedPath {
os.Remove(compressedPath)
}
return nil, errorx.New(errorx.ServerError, "更新用户头像失败: " + err.Error())
}
return &types.UploadAvatarResponse{
BaseResponse: types.BaseResponse{
Code: 200,
Message: "头像上传成功",
},
Data: types.UploadAvatarData{
AvatarUrl: avatarUrl,
},
}, nil
}

View File

@ -215,6 +215,19 @@ type UpdateUserResponse struct {
Data User `json:"data"`
}
type UploadAvatarData struct {
AvatarUrl string `json:"avatar_url"`
}
type UploadAvatarRequest struct {
Id int64 `path:"id"`
}
type UploadAvatarResponse struct {
BaseResponse
Data UploadAvatarData `json:"data"`
}
type UploadPhotoRequest struct {
Title string `json:"title" validate:"required"`
Description string `json:"description,optional"`