SpringMvc上传文件工具类

  在日常开发系统中,经常会有上传文件的要求,包括上传图片、txt文档、压缩文件等,下面是一个简单的本地上传文件封装的方法,仅供参考:

/**
 * @Author: chenyiwu
 * @Describtion: 文件上传工具类
 * @Create Time:2018/7/9
 */
public class FilesUtils {
    private static final Logger logger = LoggerFactory.getLogger(FilesUtils.class);

    //@Value("${uploadDir}")
    private String uploadDir = "D://images";

    /**
     * 功能描述:单文件上传工具类
     *
     * @auther: Martin、chen
     * @date: 2018/7/9 10:39
     * @param: file
     * @return: com.base.common.message.ResponseMessage
     */
    //@RequestMapping(value = "/uploadImage", method = RequestMethod.POST)
    public ResponseMessage uploadImage(@RequestParam(value = "file") MultipartFile file) throws RuntimeException {

        if (file.isEmpty()) {
            return new ResponseMessage("上传文件失败");
            // return JsonUtil.getFailJsonObject("文件不能为空");
        }
        // 获取文件名
        String fileName = file.getOriginalFilename();
        logger.info("上传的文件名为:" + fileName);
        // 获取文件的后缀名
        String suffixName = fileName.substring(fileName.lastIndexOf("."));
        logger.info("上传的后缀名为:" + suffixName);
        // 文件上传后的路径
        String filePath = uploadDir;
        // 解决中文问题,liunx下中文路径,图片显示问题
        // fileName = UUID.randomUUID() + suffixName;
        File dest = new File(filePath + fileName);
        // 检测是否存在目录
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        try {
            file.transferTo(dest);
            logger.info("上传成功后的文件路径未:" + filePath + fileName);
            return new ResponseMessage(fileName);
            //return JsonUtil.getSuccessJsonObject(fileName);
        } catch (IllegalStateException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return new ResponseMessage("文件上传失败");
        //return JsonUtil.getFailJsonObject("文件上传失败");
    }

    /**
     * 功能描述:文件下载
     *
     * @param imageName
     * @param request
     * @param response
     * @auther: Martin、chen
     * @date: 2018/7/9 10:40
     * @return: java.lang.String
     */
    //文件下载相关代码
    //@RequestMapping(value = "/downloadImage", method = RequestMethod.GET)
    public String downloadImage(String imageName, HttpServletRequest request, HttpServletResponse response) {

        //String fileName = "123.JPG";
        logger.debug("the imageName is : " + imageName);
        String fileUrl = uploadDir + imageName;
        if (fileUrl != null) {
            //当前是从该工程的WEB-INF//File//下获取文件(该目录可以在下面一行代码配置)然后下载到C:\\users\\downloads即本机的默认下载的目录
           /* String realPath = request.getServletContext().getRealPath(
                    "//WEB-INF//");*/
            /*File file = new File(realPath, fileName);*/
            File file = new File(fileUrl);
            if (file.exists()) {
                response.setContentType("application/force-download");// 设置强制下载不打开
                response.addHeader("Content-Disposition",
                        "attachment;fileName=" + imageName);// 设置文件名
                byte[] buffer = new byte[1024];
                FileInputStream fis = null;
                BufferedInputStream bis = null;
                try {
                    fis = new FileInputStream(file);
                    bis = new BufferedInputStream(fis);
                    OutputStream os = response.getOutputStream();
                    int i = bis.read(buffer);
                    while (i != -1) {
                        os.write(buffer, 0, i);
                        i = bis.read(buffer);
                    }
                    System.out.println("success");
                } catch (Exception e) {
                    e.printStackTrace();
                } finally {
                    if (bis != null) {
                        try {
                            bis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                    if (fis != null) {
                        try {
                            fis.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
        }
        return null;
    }

    /**
     * 功能描述:多文件上传
     *
     * @param files
     * @auther: Martin、chen
     * @date: 2018/7/9 10:44
     * @return: com.base.common.message.ResponseMessage
     */
    public ResponseMessage uploadFiles(@RequestParam(value = "file") MultipartFile[] files) {

        StringBuffer result = new StringBuffer();
        try {
            for (int i = 0; i < files.length; i++) {
                if (files[i] != null) {
                    //调用上传方法
                    String fileName = executeUpload(files[i]);
                    result.append(fileName + ";");
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
            return new ResponseMessage("文件上传失败");
        }
        return new ResponseMessage(result.toString());
    }

    /**
     * 功能描述: 多文件上传,抽取单文件代码
     *
     * @param file
     * @auther: Martin、chen
     * @date: 2018/7/9 10:42
     * @return: java.lang.String
     */
    private String executeUpload(MultipartFile file) throws Exception {

        //文件后缀名
        String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
        //上传文件名
        String fileName = UUID.randomUUID() + suffix;
        //服务端保存的文件对象
        File serverFile = new File(uploadDir + fileName);
        // 检测是否存在目录
        if (!serverFile.getParentFile().exists()) {
            serverFile.getParentFile().mkdirs();
        }
        //将上传的文件写入到服务器端文件内
        file.transferTo(serverFile);
        return fileName;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_756589808/article/details/82706829