Springboot怎么上传文件或头像

2023-07-06 04:10

短信预约 -IT技能 免费直播动态提醒

本篇内容主要讲解“Springboot怎么上传文件或头像”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“Springboot怎么上传文件或头像”吧!

1. 在配置文件中指定外部环境, 注入到代码中

头像上传路径, 若不存在, 会根据该路径创建指定路径文件夹

upload:  path: D:\\upload\headImgs

创建类 FileUtils 并读取配置文件中的值

@Component@ConfigurationProperties(prefix = "upload")@Datapublic class FileUtils {    private String path;    public File getPath() {        // 构建上传文件的存放 "文件夹" 路径        String fileDirPath = new String(path);        File fileDir = new File(fileDirPath);        if (!fileDir.exists()) {            // 递归生成文件夹            fileDir.mkdirs();        }        return fileDir;    }    public boolean del(String filename) {        File file = new File(path + File.separator + filename);        return file.delete();    }    public boolean del(String path, String filename) {        return new File(path + File.separator + filename).delete();    }}

2. 设置上传文件的限制配置

spring:  servlet:    multipart:      max-request-size: 10MB # 上传文件的最大值      max-file-size: 5MB # 单个文件上传的最大值

3. 设置外部路径映射到url

创建config类
注意: 映射路径时, 最后面一定要加 / (File.separator)

@Configurationpublic class WebMvcConfig extends WebMvcConfigurationSupport {    @Autowired    private FileUtils fileUtils;    @Override    public void addResourceHandlers(ResourceHandlerRegistry registry) {        // 加入头像文件夹映射 可通过 localhost:7188/headimage/....   访问到指定位置的图片        registry.addResourceHandler("/static    private String getCurrentuserName() {       // ...        return username;    }        public User getCurrentuser() {       // ...        return user;    }    @Override    public R modify(MultipartFile headImg, String email) {        // 校验图片格式        if (!imageTypeRight(headImg)) return R.fail("图片格式不正确");        // 获取上传文件后的路径        String path = uploadFile(headImg);        User currentuser = getCurrentuser();        // 删除之前的头像(如果是默认头像不删除)        String image = currentuser.getImage();        if (!image.equals("static/default.png")) {            if (!fileUtils.del(image.substring(path.indexOf("/") + 1))) {                log.info("修改头像时, 原来的头像删除失败");            } else {                log.info("修改头像时, 原来的头像删除成功");            }        }        // 修改数据库中头像的路径信息 和 邮箱        update(Wrappers.<User>lambdaUpdate()                .set(User::getEmail, email)                .set(User::getImage, path)                .eq(User::getUsername, currentuser.getUsername()));        // 该路径为图片相对路径 可放在url中的服务后面 进行访问        // 比如: http://localhost:9000/cloudos-opt/headimage/01c8806dc26d45539b53c22c766cd250.jpg        // http://localhost:9000/cloudos-opt/static/default.png        return R.success(null, path);    }        private boolean imageTypeRight(MultipartFile file) {        // 首先校验图片格式        List<String> imageType = Lists.newArrayList("jpg", "jpeg", "png", "bmp", "gif");        // 获取文件名，带后缀        String originalFilename = file.getOriginalFilename();        // 获取文件的后缀格式        String fileSuffix = originalFilename.substring(originalFilename.lastIndexOf(".") + 1).toLowerCase();  //不带 .        if (!imageType.contains(fileSuffix)) return false;        return true;    }         public String uploadFile(MultipartFile file) {        String originalFilename = file.getOriginalFilename();        String fileSuffix = originalFilename.substring(originalFilename.lastIndexOf(".") + 1).toLowerCase();        // 只有当满足图片格式时才进来，重新赋图片名，防止出现名称重复的情况        String newFileName = UUID.randomUUID().toString().replaceAll("-", "") + "." + fileSuffix;        // 该方法返回的为当前项目的工作目录，即在哪个地方启动的java线程        File fileTransfer = new File(fileUtils.getPath(), newFileName);        try {            file.transferTo(fileTransfer);            log.info("头像上传: " + fileTransfer.getPath());        } catch (IOException e) {            e.printStackTrace();        }        // 将图片相对路径返回给前端        return "headimage/" + newFileName;    }}