FileUtil 工具类

package com.jiayou.peis.report.biz.utils;

import lombok.SneakyThrows;
import org.apache.commons.io.IOUtils;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.util.Base64;

public class FileUtil {
    /**
     * 根据URL下载文件(pdf/图片)并转换为Base64
     * @param fileUrl
     * @return
     */
    public static String transUrlToBase64(String fileUrl){
        try {
            URL url = new URL(fileUrl);
            InputStream inputStream = url.openStream();
            ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
            IOUtils.copy(inputStream,outputStream);
            byte[] byteArray = outputStream.toByteArray();
            return Base64.getEncoder().encodeToString(byteArray);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 下载文件
     *
     * @param filePath 文件路径
     * @param response 响应体
     */
    @SneakyThrows
    public static void downloadFile(String filePath, HttpServletResponse response) {
        File file = new File(filePath);
        //输入流,读取文件内容
        FileInputStream fileInputStream = new FileInputStream(file.getPath());
        //输出流,将文件写回浏览器
        ServletOutputStream servletOutputStream = response.getOutputStream();
        //响应格式(图片)
        response.addHeader("Content-Disposition", "attachment;filename=" + new String(file.getName().getBytes()));
        response.addHeader("Content-Length", String.valueOf(file.length()));
        response.setContentType("application/octet-stream");
        int len = 0;
        byte[] bytes = new byte[1024];
        while ((len = fileInputStream.read(bytes)) != -1) {
            servletOutputStream.write(bytes, 0, len);
            servletOutputStream.flush();
        }
        fileInputStream.close();
        servletOutputStream.close();
    }

    /**
     * 下载远程文件并保存到本地
     *
     * @param remoteFilePath-远程文件路径
     * @param localFilePath-本地文件路径(带文件名)
     */
    public static void downloadFile1(String remoteFilePath, String localFilePath) {
        URL urlfile = null;
        HttpURLConnection httpUrl = null;
        BufferedInputStream bis = null;
        BufferedOutputStream bos = null;
        File f = new File(localFilePath);
        try {
            urlfile = new URL(remoteFilePath);
            httpUrl = (HttpURLConnection) urlfile.openConnection();
            httpUrl.connect();
            bis = new BufferedInputStream(httpUrl.getInputStream());
            bos = new BufferedOutputStream(new FileOutputStream(f));
            int len = 2048;
            byte[] b = new byte[len];
            while ((len = bis.read(b)) != -1) {
                bos.write(b, 0, len);
            }
            bos.flush();
            bis.close();
            httpUrl.disconnect();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                bis.close();
                bos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 下载远程文件并保存到本地
     *
     * @param remoteFilePath-远程文件路径
     * @param localFilePath-本地文件路径(带文件名)
     */
    public static void downloadFile2(String remoteFilePath, String localFilePath) {
        URL website = null;
        ReadableByteChannel rbc = null;
        FileOutputStream fos = null;
        try {
            website = new URL(remoteFilePath);
            rbc = Channels.newChannel(website.openStream());
            fos = new FileOutputStream(localFilePath);//本地要存储的文件地址 例如:test.txt
            fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE);
        } catch (Exception e) {
            e.printStackTrace();
        }finally{
            if(fos!=null){
                try {
                    fos.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(rbc!=null){
                try {
                    rbc.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

private static void test() throws FileNotFoundException {
        String path="http://10.32.1.123/report/download/userReport/2/2023/07/14/729600323349381120/604527●伍仁华●职业健康体检报告.pdf";
        String pdfName = path.substring(path.lastIndexOf("/")+1);

        String tempDir = "C:\\Users\\Administrator\\Desktop\\新建文件夹\\path\\zip\\";//System.getProperty("java.io.tmpdir") + "temp" + File.separator;
        File tempDirFile = new File(tempDir);
        if (!tempDirFile.exists()) tempDirFile.mkdirs();
        // 压缩包名称
        String zipName = "压缩包导出" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + ".zip";
        String homeFolderPath = tempDir + zipName;
        ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(homeFolderPath));

        String remoteFilePath="http://10.32.1.123/report/download/userReport/2/2023/07/14/729600323349381120/604527%E2%97%8F%E4%BC%8D%E4%BB%81%E5%8D%8E%E2%97%8F%E8%81%8C%E4%B8%9A%E5%81%A5%E5%BA%B7%E4%BD%93%E6%A3%80%E6%8A%A5%E5%91%8A.pdf";//StringZdyUtils.zwEncode2(path);
        List<String> pathList=new ArrayList<>();

        downloadFile1(remoteFilePath,tempDirFile.getAbsolutePath()+pdfName,zipOut);
    }

    /**
     * 下载远程文件并保存到本地
     *
     * @param remoteFilePath-远程文件路径
     * @param genPath-本地文件路径(带文件名)
     */
    public static void downloadFile1(String remoteFilePath, String genPath,ZipOutputStream zipOut) {
        URL urlfile = null;
        HttpURLConnection httpUrl = null;
        BufferedInputStream bis = null;
        File f = new File(genPath);
        try {
            urlfile = new URL(remoteFilePath);
            httpUrl = (HttpURLConnection) urlfile.openConnection();
            httpUrl.connect();
            bis = new BufferedInputStream(httpUrl.getInputStream());
            ZipEntry zipEntry = new ZipEntry(f.getName());

            // 将ZIP条目添加到ZIP文件
            zipOut.putNextEntry(zipEntry);

            // 从输入流读取数据并写入ZIP文件
            byte[] bytes = new byte[1024];
            int length;
            while ((length = bis.read(bytes)) >= 0) {
                zipOut.write(bytes, 0, length);
            }
            httpUrl.disconnect();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                zipOut.close();
                bis.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
    }



    private static void localDw() {
        // 两个文件的路径
        String file1Path = "C:\\Users\\Administrator\\Desktop\\新建文件夹\\path/file1.txt";
        String file2Path = "C:\\Users\\Administrator\\Desktop\\新建文件夹\\path/file2.txt";

        // ZIP文件的输出路径
        String zipOutputPath = "C:\\Users\\Administrator\\Desktop\\新建文件夹\\path/output.zip";

        try (FileOutputStream fos = new FileOutputStream(zipOutputPath);
             ZipOutputStream zipOut = new ZipOutputStream(fos);
             FileInputStream fis1 = new FileInputStream(file1Path);
             FileInputStream fis2 = new FileInputStream(file2Path)) {

            // 添加第一个文件到ZIP文件
            addToZipFile(file1Path, fis1, zipOut);

            // 添加第二个文件到ZIP文件
            addToZipFile(file2Path, fis2, zipOut);

            System.out.println("文件成功打包成ZIP文件!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static void addToZipFile(String filePath, FileInputStream fis, ZipOutputStream zipOut)
            throws IOException {
        // 创建ZIP条目
        ZipEntry zipEntry = new ZipEntry(new File(filePath).getName());

        // 将ZIP条目添加到ZIP文件
        zipOut.putNextEntry(zipEntry);

        // 从输入流读取数据并写入ZIP文件
        byte[] bytes = new byte[1024];
        int length;
        while ((length = fis.read(bytes)) >= 0) {
            zipOut.write(bytes, 0, length);
        }

        // 关闭当前ZIP条目
        zipOut.closeEntry();

        // 关闭输入流
        fis.close();
    }


    /**
     * 小试牛刀
     * @param args
     */
    public static void main(String[] args) {
        /*远程文件路径*/
        String remoteFilePath1 = "http://192.168.150.190:9000/reportfile/20230727/TJH24289_1.pdf";
        //String remoteFilePath2 = "https://pic.3gbizhi.com/2019/1112/20191112013312648.jpg";
        /*本地文件路径(带文件名)*/
        String localFilePath1 ="C:\\Users\\Administrator\\Desktop\\biji\\广州塔.pdf";
        //String localFilePath2 ="E:\\LeStoreDownload\\update\\大桥.jpg";
        downloadFile1(remoteFilePath1,localFilePath1);
        //downloadFile2(remoteFilePath2,localFilePath2);
    }



}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值