基于OKHttp的大文件下载

1.需求:

根据Url下载大文件,本地创建文件,从输入流读取并写入文件,要求能满足大文件的下载,不能出现OOM

2.引入依赖
xml 复制代码
<!-- OK HTTP -->
<dependency>
    <groupId>com.squareup.okhttp3</groupId>
    <artifactId>okhttp</artifactId>
    <version>4.12.0</version>
</dependency>
<!-- Hutool工具包 -->
<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-all</artifactId>
    <version>5.7.16</version>
</dependency>
3.下载文件代码
java 复制代码
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.net.URLDecoder;
import java.nio.charset.StandardCharsets;


/**
 * 下载文件并保存
 * @param url 下载地址
 * @param folder 本地保存文件夹路径,比如:E:\test\down
 * @return 文件(下载失败返回null)
 */
public File downloadFile(String url, String folder){
    OkHttpClient client = new OkHttpClient();
    Request request = new Request.Builder().url(url).build();

    try(Response response = client.newCall(request).execute()){
        if(response.isSuccessful()){
            if(Objects.isNull(response.body())){
                logger.info("下载失败,响应体为空");
                return null;
            }
            //从Url提取文件名
            String fileName = getFileNameFromUrl(url);
     
            //拼接完整文件路径
            String filePath = folder + File.separator + fileName;
            //创建本地文件
            File file = new File(filePath);
            FileUtil.touch(file);
            try(
                InputStream in = response.body().byteStream();//获取响应输入流
                FileOutputStream out = new FileOutputStream(file)//创建文件输出流
            ){
                //写入本地文件
                int len;
                byte[] buffer = new byte[4096];
                while((len = in.read(buffer)) != -1) { //从输入流中读取数据到缓冲区
                    out.write(buffer, 0, len); //将缓冲区的数据写入输出流
                }
                out.flush(); //刷新输出流缓冲区
            }
            return file;
        }else{
            logger.info("响应失败:httpStatus={}", response.code());
            return null;
        }
    }catch (Exception e){
        logger.info("文件下载失败,地址:{},错误:{}", url, e.getMessage(), e);
        return null;
    }
}

/**
 * 从Url提取文件名
 * @param url 文件路径
 * @return 文件名
 */
private static String getFileNameFromUrl(String url){
    String uri = URLDecoder.decode(url.trim(), StandardCharsets.UTF_8);
    int index = uri.lastIndexOf("/");
    return  index==-1 ? uri : uri.substring(index+1);
}
相关推荐
ajsbxi3 分钟前
苍穹外卖学习记录
java·笔记·后端·学习·nginx·spring·servlet
StayInLove22 分钟前
G1垃圾回收器日志详解
java·开发语言
对许26 分钟前
SLF4J: Failed to load class “org.slf4j.impl.StaticLoggerBinder“
java·log4j
无尽的大道30 分钟前
Java字符串深度解析:String的实现、常量池与性能优化
java·开发语言·性能优化
小鑫记得努力39 分钟前
Java类和对象(下篇)
java
binishuaio43 分钟前
Java 第11天 (git版本控制器基础用法)
java·开发语言·git
zz.YE1 小时前
【Java SE】StringBuffer
java·开发语言
老友@1 小时前
aspose如何获取PPT放映页“切换”的“持续时间”值
java·powerpoint·aspose
wrx繁星点点1 小时前
状态模式(State Pattern)详解
java·开发语言·ui·设计模式·状态模式
Upaaui1 小时前
Aop+自定义注解实现数据字典映射
java