SpringBoot实现文件上传跟删除文件

介绍

文件上传是后台必不可少的技能，提供给前台图片上传最为常见
今天我们来学习一下图片上传

开始代码

图片上传比较简单，首先们写一个方法提供给前台调用

创建公共返回类BaseResult

@Data
public class BaseResult {
    private boolean success;
    private String msg;
    private Object data;
}

创建ImgsController

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import sun.tools.jstat.Token;
import zhongmeiutilsapi.td0f7.cn.base.BaseResult;
import zhongmeiutilsapi.td0f7.cn.entrty.Imgs;
import zhongmeiutilsapi.td0f7.cn.entrty.User;
import zhongmeiutilsapi.td0f7.cn.mapper.ImgsMapper;
import zhongmeiutilsapi.td0f7.cn.mapper.UserMapper;
import zhongmeiutilsapi.td0f7.cn.token.TokenUtil;
import zhongmeiutilsapi.td0f7.cn.token.UserLoginToken;
import zhongmeiutilsapi.td0f7.cn.utils.FileUtils;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

@RestController
@RequestMapping("Imgs")
public class ImgsController {

    @PostMapping//文件参数必须用MultipartFile接收
    public BaseResult upload(@RequestParam(value = "file") MultipartFile file) {
        BaseResult data = new BaseResult();
        try {
		//文件保存的路径
            String path = "/www/wwwroot/app.td0f7.cn/images";

            String result = FileUtils.upload(file, path, new File(file.getOriginalFilename()).getPath());
		//通过工具类保存文件
            data.setSuccess(true);
            data.setMsg("成功上传！");
            data.setData(result);//result是保存文件后返回的路径
        } catch (Exception e) {
            e.printStackTrace();
            data.setMsg(e.getMessage());
		//如果发生异常，则代表文件保存出错了Success默认是false，所以不用修改
        }
        return data;
    }

    @PutMapping
    public BaseResult delete(@RequestParam(value = "id") int id) {
        BaseResult result = new BaseResult();
        Imgs imgs = mapper.findById(id);
        result.setSuccess(FileUtils.deleteFile(imgs.getPath()));
        if (result.isSuccess()) mapper.delete(id);//删除工具类返回一个bool得值，直接给Success字段
        result.setMsg(result.isSuccess() ? "删除成功！" : "删除失败！");//如果Success=true删除成功，反之删除失败
        return result;
    }
}

创建文件上传工具类

import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.util.UUID;

public class FileUtils {
    public static String upload(MultipartFile file, String path, String fileName) throws Exception {
        // 生成新的文件名
        String realPath = path + "/" + UUID.randomUUID().toString().replace("-", "") + fileName.substring(fileName.lastIndexOf("."));
        File dest = new File(realPath);
        // 判断文件父目录是否存在
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdir();
        }
        // 保存文件
        file.transferTo(dest);
        return realPath;
    }

    public static boolean deleteFile(String path) {
        File file = new File(path);
        if (file.exists()) {
            return file.delete();
        }
        return false;
    }
}

这样图片上传就完成啦