通过URL下载文件解决301问题

通过URL下载文件

    @ApiOperation("文件下载")
    @ApiImplicitParam(name = "downFile")
    @RequestMapping("/downFile")
    public void downFile(HttpServletRequest request) {
        String url = "http://www.blog.pythonlibrary.org/wp-content/uploads/2012/06/wxDbViewer.zip";
        String fileUrl = "/Users/lee/Downloads/download";
        try {
            downLoadFileFromUrl(url,fileUrl);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 从网络URL中下载文件
     * @Author LJ
     * @Date 2020/10/16
     * @Time 10:32
     * @param urlStr
     * @param savePath
     * @return void
     */
    public static void downLoadFileFromUrl(String urlStr, String savePath) throws Exception {
        URL url = new URL(urlStr);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        //设置超时时间为3秒
        conn.setConnectTimeout(3 * 1000);
        //防止屏蔽程序抓取而返回403错误
        conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");

        String newUrl = urlStr;
        Map<String, List<String>> map = conn.getHeaderFields();
        //遍历所有的响应头字段
        for (String key : map.keySet()) {
            System.out.println(key + "--->" + map.get(key).get(0));
            if ("Location".equals(key)) {
                //获取新地址
                newUrl = map.get(key).get(0);
                System.out.println("newUrl--->" + newUrl);
                break;
            }
        }
        //重新实例化url对象
        url = new URL(newUrl);
        //重新打开和URL之间的连接
        conn = (HttpURLConnection) url.openConnection();

        //得到输入流
        InputStream inputStream = conn.getInputStream();
        //获取自己的数组
        byte[] getData = readInputStream(inputStream);

        //文件保存位置
        File saveDir = new File(savePath);
        if (!saveDir.exists()) {
            saveDir.mkdirs();
        }

        //拿到URL里的文件名
        String fileName = urlStr.substring(urlStr.lastIndexOf("/") + 1, urlStr.length());
        File file = new File(saveDir + File.separator + fileName);
        FileOutputStream fos = new FileOutputStream(file);
        fos.write(getData);

        if (fos != null) {
            fos.close();
        }
        if (inputStream != null) {
            inputStream.close();
        }
        System.out.println(url + " download success");
    }

    /**
     * 从输入流中获取字节数组
     * @Author LJ
     * @Date 2020/10/16
     * @Time 10:31
     * @param inputStream
     * @return byte[]
     */
    public static byte[] readInputStream(InputStream inputStream) throws IOException {
        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();
    }