java下载文件并保存

1. HttpClient

private static final String FILE_PATH_SEP = "/";

private static String getRemoteFileSuffix(String url) {
     Assert.hasText(url, "URL is null");
     return url.substring(url.lastIndexOf("."));
}

/**
     * 文件下载工具
     * 1.创建目录
     * 2.获取URL文件后缀，拼接完整路径
     * 3.通过outputStream下保存
     *
     * @param fileName 需要保存的文件名 不带后缀
     * @param filePath 需要保存的文件路径
     * @param url      需要下载的文件地址
     * @author Lawrence
     * @date 2021/4/15 16:19
     */
    public static String download(@NonNull String url, String filePath, String fileName) throws IOException {
        //如果不带分隔符则加上
        if (!filePath.endsWith(FILE_PATH_SEP)) {
            filePath += FILE_PATH_SEP;
        }
        String suffix = getRemoteFileSuffix(url);
        fileName = filePath + fileName + suffix;
        File path = new File(filePath);
        if (!path.exists()) {
            if (!path.mkdirs()) {
                throw new IOException(filePath + " create failed");
            }
        }
        CloseableHttpResponse httpResponse = null;
        try (CloseableHttpClient client = HttpClients.createDefault();) {
            HttpGet get = new HttpGet(url);
            get.setConfig(REQUEST_CONFIG);
            httpResponse = client.execute(get);
            //这里可以根据返回头的content-type获取文件类型，我直接用url后缀代替了
//System.out.println(httpResponse.getFirstHeader(HttpHeaders.CONTENT_TYPE).getValue());
            HttpEntity httpEntity = httpResponse.getEntity();
            if (httpEntity != null) {
                try (FileOutputStream fos = new FileOutputStream(new File(fileName))) {
                    httpEntity.writeTo(fos);
                    EntityUtils.consume(httpEntity);
                }
            }
        } catch (Exception e) {
            log.error(e.getMessage());
        } finally {
            if (null != httpResponse) {
                httpResponse.close();
            }
        }
        return fileName;
    }

2. okhttp

public static void download(String url, Headers headers, String fileName) {
        OkHttpClient client = new OkHttpClient().newBuilder().connectTimeout(3, TimeUnit.SECONDS).readTimeout(3, TimeUnit.SECONDS).build();
        Request.Builder builder = new Request.Builder().get();
        if (null != headers) {
            builder.headers(headers);
        }
        Request request = builder.url(url).build();
        try (Response response = client.newCall(request).execute();
             ResponseBody responseBody = response.body()) {
            if (response.isSuccessful()) {
                IAssert.hasReply(responseBody);
                try (InputStream is = responseBody.byteStream();
                     FileOutputStream fos = new FileOutputStream(new File(fileName))) {
                    int len;
                    while ((len = is.read()) != -1) {
                        fos.write(len);
                    }
                    fos.flush();
                }
            } else {
                log.error("request: {} failed", url);
                String returned = "";
                if (responseBody != null) {
                    returned = new String(responseBody.bytes());
                    log.error("returned :{}", returned);
                }
                throw new GlpErrorException(url, returned);
            }
        } catch (IOException ioe) {
            log.error(ioe.getMessage());
            throw new GlpErrorException(url, ioe.getMessage());
        }
    }

okhttp有个问题，貌似跑完之后还有个线程还在跑，还没找到具体问题。