SpringBoot+Shiro图片上传

---

1、准备工作

1.1、工具类

@Component
public class FileUploadUtils {

    @Autowired
    CommonUtils commonUtils;

    /**
     * 给用户上传头像并自定义文件名
     *
     * @param file
     * @param customFileName
     * @return
     */
    public String uploadAvatarFile(MultipartFile file,String customFileName) {
        //window下
//        String path = "D:/学习/码云/arcsoft_Administration/arcsoft_Administration/src/main/resources/static/"+customFileName;
        //linux服务器下
        String path = "/usr/java/arcsoftAdministrationFolder/"+customFileName;
        File filePath = new File(path);
        System.out.println("文件的保存路径：" + path);
        if (!filePath.exists() && !filePath.isDirectory()) {
            System.out.println("目录不存在，创建目录:" + filePath);
            filePath.mkdirs(); // mkdir()不会创建目录,找不到相应路径时返回false;而mkdirs()当目录不存在时则会创建相应目录
        }
        //获取原始文件名称(包含格式)
        String originalFileName = file.getOriginalFilename();
        //获取文件类型，以最后一个`.`为标识,(substring)字符串包括 start 处的字符或者start 处到 stop-1 处的所有字符
        String type = originalFileName.substring(originalFileName.lastIndexOf(".") + 1);
        //给文件起个名字,为用户id
        String fileName = commonUtils.getUUID() + "." + type; // 新文件名，这里可以根据需要改名
        //在指定路径下创建一个文件
        File targetFile = new File(path, fileName); // 未使用outputStream.write()的时候,是一个File对象,保存在内存中,硬盘中看不到,但是可以使用这个对象
        try {
            // 使用springmvc的transferTo方法上传文件
            file.transferTo(targetFile);
            //返回路径自定义名字+uuid
            return customFileName+fileName;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

    public boolean customDeleteFile(String fileName){
        //window下文件路径
//        String path = "D:/学习/码云/arcsoft_Administration/arcsoft_Administration/src/main/resources/static/";
        //linux服务器下文件路径
        String path = "/usr/java/arcsoftAdministrationFolder/";

        try {
            File file = new File(path+fileName);
            if (file.exists()){//文件是否存在
                file.delete();//删除文件
            }
//            System.out.println("=======================");
//            System.out.println(path+fileName);
//            System.out.println("=======================");
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }
}

/**
 * 自定义工具类
 */
@Component
public class CommonUtils {
    /**
     * 获得一个UUID
     * @return String UUID
     */
    public  String getUUID(){
        String uuid = UUID.randomUUID().toString();
        //去掉“-”符号
        return uuid.replaceAll("-", "");
    }

}

1.2、自定义WebMvcConfigurer

@Configuration
@EnableWebMvc
public class MvcConfig implements WebMvcConfigurer {
    @Override
    public void addResourceHandlers(ResourceHandlerRegistry registry) {
        //window代码
//        registry.addResourceHandler("/file/**").addResourceLocations("file:D:/学习/码云/arcsoft_Administration/arcsoft_Administration/src/main/resources/static/");
        //服务器代码（前者为）虚拟路径，后者为真实路径
        registry.addResourceHandler("/file/**").addResourceLocations("file:/usr/java/arcsoftAdministrationFolder/");
    }
}

---

2、逻辑层

2.1、Service层实现类

@Service
public class FileUploadServiceImpl implements fileUploadService {
    @Autowired
    FileUploadUtils fileUploadUtils;

    @Autowired
    CommonUtils commonUtils;

    @Autowired
    UserMapper userMapper;

    @Override
    public JsonData updateUserProfile(MultipartFile file, User newUser) {
        //添加图片文件
        String headImage = fileUploadUtils.uploadAvatarFile(file, "headImage/");
        //删除老图片文件
        fileUploadUtils.customDeleteFile(newUser.getHeadImage());
        //刷新权限
        ShiroUtils.setUser(newUser);
        //更新用户的头像地址
        newUser.setHeadImage(headImage);
        //更新用户信息
        userMapper.updateUserInfoById(newUser);
        return JsonData.buildSuccess();
    }
}

2.2、Controller层

@RestController
@RequestMapping("authc")
@CrossOrigin//跨域
@Transactional(rollbackFor = Exception.class)//事物回滚
public class FileUploadController {

    @Autowired
    fileUploadService fileUploadService;

    @Autowired
    UserService userService;

    @PostMapping("/uploadHead")
    public JsonData uploadHead(@RequestParam("file") MultipartFile file) {
        User user = (User) SecurityUtils.getSubject().getPrincipal();
        //通过用户账号重新获取用户的全部信息，包括权限
        User allUserInfoByUsername = userService.findAllUserInfoByUsername(user.getUsername());
        try {
            return fileUploadService.updateUserProfile(file,allUserInfoByUsername);
        } catch (Exception e) {
            e.printStackTrace();
            return JsonData.buildError("头像上传失败");
        }
    }
}

3、其他

3.1、Shiro的清除当前缓存并赋予新值

public class ShiroUtils {
    /**
     *
     * @return
     */
    public static Subject getSubjct() {
        return SecurityUtils.getSubject();
    }

    /**
     * 获取用户信息
     * @return
     */
    public static User getUser() {
        return (User) getSubjct().getPrincipal();
    }

    /**
     * 切换身份，登录后，动态更改subject的用户属性
     *
     * @param newUserInfo
     */
    public static void setUser(User newUserInfo) {
        Subject subject = SecurityUtils.getSubject();
        PrincipalCollection principalCollection = subject.getPrincipals();
        String realmName = principalCollection.getRealmNames().iterator().next();
        PrincipalCollection newPrincipalCollection =
                new SimplePrincipalCollection(newUserInfo, realmName);
        subject.runAs(newPrincipalCollection);
    }
}