java从URL下载文件、图片到本地-蒲公英云

java从URL下载文件、图片到本地

傷城~ 2023-02-19 06:27 116阅读 0赞

有时候我们需要从网络上下载一个文件、图片、视频等到本地，不想依赖jar，就需要自己实现

代码：

import java.io.*;
import java.net.URL;
import java.net.URLConnection;
/**
 * @author : Vick C
 * @version : v1.0
 * @date : 2020/6/19
 */
public class FileUtil {
    /**
     * download file from internet
     * @param fileUrl  file url
     * @param fileName file name
     * @param savePath save path
     * @return success return the file path, fail return null
     */
    public static String downLoadFromInternet(String fileUrl, String fileName, String savePath) {
        try {
            URLConnection conn = new URL(fileUrl).openConnection();
            // 设置超时间为10秒
            conn.setConnectTimeout(10 * 1000);
            // 防止屏蔽程序抓取而返回403错误
            conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
            // 得到输入流
            InputStream inputStream = conn.getInputStream();
            // 获取字节数组
            byte[] byteArray = readInputStream(inputStream);
            assert byteArray != null;
            // 文件保存位置
            File saveDir = new File(savePath);
            if (!saveDir.exists()) {
                saveDir.mkdir();
            }
            File file = new File(saveDir + File.separator + fileName);
            FileOutputStream fos = new FileOutputStream(file);
            fos.write(byteArray);
            fos.close();
            inputStream.close();
            return saveDir + File.separator + fileName;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
    /**
     * 从输入流中获取字节数组
     * @param inputStream input stream
     * @return byte array
     */
    public static byte[] readInputStream(InputStream inputStream) {
        try {
            byte[] buffer = new byte[1024];
            int len = 0;
            ByteArrayOutputStream bos = new ByteArrayOutputStream();
            while ((len = inputStream.read(buffer)) != -1) {
                bos.write(buffer, 0, len);
            }
            bos.close();
            return bos.toByteArray();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
}