基于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);
}
相关推荐
Theodore_10222 小时前
4 设计模式原则之接口隔离原则
java·开发语言·设计模式·java-ee·接口隔离原则·javaee
冰帝海岸3 小时前
01-spring security认证笔记
java·笔记·spring
世间万物皆对象3 小时前
Spring Boot核心概念:日志管理
java·spring boot·单元测试
没书读了4 小时前
ssm框架-spring-spring声明式事务
java·数据库·spring
小二·4 小时前
java基础面试题笔记(基础篇)
java·笔记·python
开心工作室_kaic4 小时前
ssm161基于web的资源共享平台的共享与开发+jsp(论文+源码)_kaic
java·开发语言·前端
懒洋洋大魔王4 小时前
RocketMQ的使⽤
java·rocketmq·java-rocketmq
武子康4 小时前
Java-06 深入浅出 MyBatis - 一对一模型 SqlMapConfig 与 Mapper 详细讲解测试
java·开发语言·数据仓库·sql·mybatis·springboot·springcloud
转世成为计算机大神5 小时前
易考八股文之Java中的设计模式?
java·开发语言·设计模式
qq_327342735 小时前
Java实现离线身份证号码OCR识别
java·开发语言