Java实现文件上传下载工具类

不念不忘少年蓝@ 2022-06-07 03:56 370阅读 0赞

本篇文章在Eclipse环境下采用Java语言实现文件上传下载工具类。上传时，为避免文件名在服务器中重复，采用“服务器时间（定义到毫秒）+文件名+文件后缀“的方式作为服务器上的文件名；下载过程中利用 spring mvc ResponseEntity 做文件下载，返回的是字节流，下载成功后可自定义文件的保存路径。具体源码如下所示：

package com.utils;
    
    import java.io.File;
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.UnsupportedEncodingException;
    import java.text.SimpleDateFormat;
    import java.util.Date;
    
    import org.apache.commons.io.FileUtils;
    import org.springframework.http.HttpHeaders;
    import org.springframework.http.HttpStatus;
    import org.springframework.http.MediaType;
    import org.springframework.http.ResponseEntity;
    
    /** * 文件上传下载工具类 * */
    public class FileHelper { 
    
        /** * 根据路径确定目录，没有目录，则创建目录 * * @param path */
        private static void createDir(String path) {
            File fileDir = new File(path);
            if (!fileDir.exists() && !fileDir.isDirectory()) {
       // 判断/download目录是否存在
                fileDir.mkdir();// 创建目录
            }
        }
    
        /** * 将文件名解析成文件的上传路径 * * @param fileName * @return 上传到服务器的文件名 */
        public static String transPath(String fileName, String path) {
            createDir(path);
            Date date = new Date();
            SimpleDateFormat dateformat = new SimpleDateFormat("yyyyMMddhhmmssSSS");// 定义到毫秒
            String nowStr = dateformat.format(date);
            String filenameStr = fileName.substring(0, fileName.lastIndexOf("."));// 去掉后缀的文件名
            String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);// 后缀
            if (fileName.trim() != "") {
       // 如果名称不为"",说明该文件存在，否则说明该文件不存在
                path += "\\" + filenameStr + nowStr + "." + suffix;// 定义上传路径
            }
            return path;
        }
    
        /** * 提醒文件下载 * * @param fileName * @param path * @return */
        public static ResponseEntity<byte[]> downloadFile(String fileName, String path) {
            try {
                fileName = new String(fileName.getBytes("GB2312"), "ISO_8859_1");// 避免文件名中文不显示
            } catch (UnsupportedEncodingException e1) {
                e1.printStackTrace();
            }
            File file = new File(path);
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            headers.setContentDispositionFormData("attachment", fileName);
    
            ResponseEntity<byte[]> byteArr = null;
            try {
                byteArr = new ResponseEntity<byte[]>(FileUtils.readFileToByteArray(file), headers, HttpStatus.OK);
            } catch (IOException e) {
                e.printStackTrace();
            }
            return byteArr;
        }
    
        /** * 将输入流中的数据写入字节数组 * * @param in * @return */
        public static byte[] inputStream2ByteArray(InputStream in, boolean isClose) {
            byte[] byteArray = null;
            try {
                int total = in.available();
                byteArray = new byte[total];
                in.read(byteArray);
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                if (isClose) {
                    try {
                        in.close();
                    } catch (Exception e2) {
                        System.out.println("关闭流失败");
                    }
                }
            }
            return byteArray;
        }
    }