用户头像上传限制只能为图片格式

RuoYi

2022-05-22 da0309eb09fe1c99f49b1666712a8da24d67e09a

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysProfileController.java

@@ -1,6 +1,5 @@
package com.ruoyi.web.controller.system;



import java.io.IOException;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.PostMapping;

@@ -21,6 +20,7 @@
import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

import com.ruoyi.common.utils.file.FileUploadUtils;

import com.ruoyi.common.utils.file.MimeTypeUtils;

import com.ruoyi.framework.web.service.TokenService;

import com.ruoyi.system.service.ISysUserService;



@@ -72,7 +72,7 @@
                && UserConstants.NOT_UNIQUE.equals(userService.checkEmailUnique(user)))

        {

            return AjaxResult.error("修改用户'" + user.getUserName() + "'失败，邮箱账号已存在");

        }       
        }

        user.setUserId(sysUser.getUserId());

        user.setPassword(null);

        if (userService.updateUserProfile(user) > 0)

@@ -121,12 +121,12 @@
     */

    @Log(title = "用户头像", businessType = BusinessType.UPDATE)

    @PostMapping("/avatar")

    public AjaxResult avatar(@RequestParam("avatarfile") MultipartFile file) throws IOException

    public AjaxResult avatar(@RequestParam("avatarfile") MultipartFile file) throws Exception

    {

        if (!file.isEmpty())

        {

            LoginUser loginUser = getLoginUser();

            String avatar = FileUploadUtils.upload(RuoYiConfig.getAvatarPath(), file);

            String avatar = FileUploadUtils.upload(RuoYiConfig.getAvatarPath(), file, MimeTypeUtils.IMAGE_EXTENSION);

            if (userService.updateUserAvatar(loginUser.getUsername(), avatar))

            {

                AjaxResult ajax = AjaxResult.success();