文件上传控制器

/**
 * @author: xuzhi
 * @date: 2020/11/5 14:02
 */
@Data
public class UploadFileVO {
    
    
    private String src;

    private String title;
}

/**
 * @author: xuzhi
 * @date: 2020/11/5 13:57
 */
@Data
public class JsonResult<T> {
    
    
    /**
     * 错误码
     */
    private Integer code;

    /**
     * 提示信息
     */
    private String msg;

    /**
     * 返回的具体内容
     */
    private T data;


    public JsonResult(Integer code, String msg, T data) {
    
    
        this.code = code;
        this.msg = msg;
        this.data = data;
    }

    public JsonResult() {
    
    
    }


    public JsonResult fail() {
    
    
        return new JsonResult(1, "操作失败", null);
    }

    public JsonResult fail(String msg) {
    
    
        return new JsonResult(1, msg, null);
    }

    public JsonResult ok() {
    
    
        return new JsonResult(0, "操作成功", null);
    }


    public JsonResult ok(T data) {
    
    
        return new JsonResult(0, "操作成功", data);
    }
}

import com.ssm.blog.dto.JsonResult;
import com.ssm.blog.dto.UploadFileVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.util.Calendar;

/**
 * @author: xuzhi
 * @date: 2020/11/7 9:40
 */
@Slf4j
@Controller
@RequestMapping("/admin/upload")
public class UploadFileController {
    
    
    /**
     * 文件保存目录
     */
    public final String rootPath="G:/Github/ForestBlog/uploads";
    public final String allowSuffix=".bmp.jpg.jpeg.png.gif.pdf.doc.zip.rar.gz";

    /**
     * 上传文件
     *
     * @param file
     * @return
     * @throws
     */
    @RequestMapping(value = "/img", method = RequestMethod.POST)
    @ResponseBody
    public JsonResult uploadFile(@RequestParam("file") MultipartFile file) {
    
    

        //1.文件后缀过滤，只允许部分后缀
        //文件的完整名称,如spring.jpeg
        String filename = file.getOriginalFilename();
        //文件名,如spring
        String name = filename.substring(0, filename.indexOf("."));
        //文件后缀,如.jpeg
        String suffix = filename.substring(filename.lastIndexOf("."));

        if (allowSuffix.indexOf(suffix) == -1) {
    
    
            return new JsonResult().fail("不允许上传该后缀的文件！");
        }


        //2.创建文件目录
        //创建年月文件夹
        Calendar date = Calendar.getInstance();
        File dateDirs = new File(date.get(Calendar.YEAR)
                + File.separator + (date.get(Calendar.MONTH) + 1));

        //目标文件
        File descFile = new File(rootPath + File.separator + dateDirs + File.separator + filename);
        int i = 1;
        //若文件存在重命名
        String newFilename = filename;
        while (descFile.exists()) {
    
    
            newFilename = name + "(" + i + ")" + suffix;
            String parentPath = descFile.getParent();
            descFile = new File(parentPath + File.separator + newFilename);
            i++;
        }
        //判断目标文件所在的目录是否存在
        if (!descFile.getParentFile().exists()) {
    
    
            //如果目标文件所在的目录不存在，则创建父目录
            descFile.getParentFile().mkdirs();
        }

        //3.存储文件
        //将内存中的数据写入磁盘
        try {
    
    
            file.transferTo(descFile);
        } catch (Exception e) {
    
    
            e.printStackTrace();
            log.error("上传失败，cause:{}", e);
        }
        //完整的url
        String fileUrl = "/uploads/" + dateDirs + "/" + newFilename;

        //4.返回URL
        UploadFileVO uploadFileVO = new UploadFileVO();
        uploadFileVO.setTitle(filename);
        uploadFileVO.setSrc(fileUrl);
        return new JsonResult().ok(uploadFileVO);
    }
}