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

傷城~ 2023-02-19 06:27 3阅读 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;
        }
    }