上传用户头像

Author: summerblue

app/http/controllers/api/v1/users_controller.go

@@ -4,6 +4,8 @@ import (
 	"gohub/app/models/user"
 	"gohub/app/requests"
 	"gohub/pkg/auth"
+	"gohub/pkg/config"
+	"gohub/pkg/file"
 	"gohub/pkg/response"
 
 	"github.com/gin-gonic/gin"
@@ -110,3 +112,23 @@ func (ctrl *UsersController) UpdatePassword(c *gin.Context) {
 		response.Success(c)
 	}
 }
+
+func (ctrl *UsersController) UpdateAvatar(c *gin.Context) {
+
+	request := requests.UserUpdateAvatarRequest{}
+	if ok := requests.Validate(c, &request, requests.UserUpdateAvatar); !ok {
+		return
+	}
+
+	avatar, err := file.SaveUploadAvatar(c, request.Avatar)
+	if err != nil {
+		response.Abort500(c, "上传头像失败，请稍后尝试~")
+		return
+	}
+
+	currentUser := auth.CurrentUser(c)
+	currentUser.Avatar = config.GetString("app.url") + avatar
+	currentUser.Save()
+
+	response.Data(c, currentUser)
+}

app/requests/requests.go

@@ -46,3 +46,14 @@ func validate(data interface{}, rules govalidator.MapData, messages govalidator.
 	// 开始验证
 	return govalidator.New(opts).ValidateStruct()
 }
+
+func validateFile(c *gin.Context, data interface{}, rules govalidator.MapData, messages govalidator.MapData) map[string][]string {
+	opts := govalidator.Options{
+		Request:       c.Request,
+		Rules:         rules,
+		Messages:      messages,
+		TagIdentifier: "valid",
+	}
+	// 调用 govalidator 的 Validate 方法来验证文件
+	return govalidator.New(opts).Validate()
+}

app/requests/user_request.go

@@ -3,6 +3,7 @@ package requests
 import (
 	"gohub/app/requests/validators"
 	"gohub/pkg/auth"
+	"mime/multipart"
 
 	"github.com/gin-gonic/gin"
 	"github.com/thedevsaddam/govalidator"
@@ -155,3 +156,27 @@ func UserUpdatePassword(data interface{}, c *gin.Context) map[string][]string {
 
 	return errs
 }
+
+type UserUpdateAvatarRequest struct {
+	Avatar *multipart.FileHeader `valid:"avatar" form:"avatar"`
+}
+
+func UserUpdateAvatar(data interface{}, c *gin.Context) map[string][]string {
+
+	rules := govalidator.MapData{
+		// size 的单位为 bytes
+		// - 1024 bytes 为 1kb
+		// - 1048576 bytes 为 1mb
+		// - 20971520 bytes 为 20mb
+		"file:avatar": []string{"ext:png,jpg,jpeg", "size:20971520", "required"},
+	}
+	messages := govalidator.MapData{
+		"file:avatar": []string{
+			"ext:ext头像只能上传 png, jpg, jpeg 任意一种的图片",
+			"size:头像文件最大不能超过 20MB",
+			"required:必须上传图片",
+		},
+	}
+
+	return validateFile(c, data, rules, messages)
+}

pkg/file/file.go

@@ -2,10 +2,17 @@
 package file
 
 import (
+	"fmt"
+	"gohub/pkg/app"
+	"gohub/pkg/auth"
+	"gohub/pkg/helpers"
 	"io/ioutil"
+	"mime/multipart"
 	"os"
 	"path/filepath"
 	"strings"
+
+	"github.com/gin-gonic/gin"
 )
 
 // Put 将数据存入文件
@@ -27,3 +34,25 @@ func Exists(fileToCheck string) bool {
 func FileNameWithoutExtension(fileName string) string {
 	return strings.TrimSuffix(fileName, filepath.Ext(fileName))
 }
+func SaveUploadAvatar(c *gin.Context, file *multipart.FileHeader) (string, error) {
+
+	var avatar string
+	// 确保目录存在，不存在创建
+	publicPath := "public"
+	dirName := fmt.Sprintf("/uploads/avatars/%s/%s/", app.TimenowInTimezone().Format("2006/01/02"), auth.CurrentUID(c))
+	os.MkdirAll(publicPath+dirName, 0755)
+
+	// 保存文件
+	fileName := randomNameFromUploadFile(file)
+	// public/uploads/avatars/2021/12/22/1/nFDacgaWKpWWOmOt.png
+	avatarPath := publicPath + dirName + fileName
+	if err := c.SaveUploadedFile(file, avatarPath); err != nil {
+		return avatar, err
+	}
+
+	return avatarPath, nil
+}
+
+func randomNameFromUploadFile(file *multipart.FileHeader) string {
+	return helpers.RandomString(16) + filepath.Ext(file.Filename)
+}

public/uploads/.gitignore

@@ -0,0 +1,2 @@
+*
+!.gitignore

routes/api.go

@@ -63,6 +63,7 @@ func RegisterAPIRoutes(r *gin.Engine) {
 			usersGroup.PUT("/email", middlewares.AuthJWT(), uc.UpdateEmail)
 			usersGroup.PUT("/phone", middlewares.AuthJWT(), uc.UpdatePhone)
 			usersGroup.PUT("/password", middlewares.AuthJWT(), uc.UpdatePassword)
+			usersGroup.PUT("/avatar", middlewares.AuthJWT(), uc.UpdateAvatar)
 		}
 
 		cgc := new(controllers.CategoriesController)