这个是可以直接拿去用的
DownLoadFromUrlController
package com.richfit.richfit.controller;
import com.richfit.richfit.DownLoadFromUrlUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.io.File;
/**
* @ClassName DownLoadFromUrlController
* @Description:
* @Author BruthLi
* @Date 2020/2/9
* @Version V1.0
**/
@Api(value = "从网络Url中下载文件")
@RestController
@RequestMapping(value = "/sys/downUrl")
@Slf4j
public class DownLoadFromUrlController {
@ApiOperation(value = "从网络Url中下载文件", notes = "从网络Url中下载文件")
@RequestMapping(value = "/downFromOUTUrl", method = RequestMethod.POST)
public File sendEmail(@ApiParam(name = "urlStr",value = "下载网络地址",required = true) @RequestParam(value = "urlStr",required = true)String urlStr,
@ApiParam(name = "fileName",value = "下载内容放到例如 text.txt文件名称",required = true) @RequestParam(value = "fileName",required = true)String fileName,
@ApiParam(name = "savePath",value = "保存路径",required = true)@RequestParam(value = "savePath",required = true)String savePath)throws Exception{
return DownLoadFromUrlUtil.downLoadFromUrl(urlStr,fileName,savePath);
}
}
DownLoadFromUrlUtil
package com.richfit.richfit;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
/**
* @ClassName DownLoadFromUrlUtil
* @Description: 从网络Url中下载文件
* @Author BruthLi
* @Date 2020/2/9
* @Version V1.0
**/
public class DownLoadFromUrlUtil {
/**
* 从网络Url中下载文件
* @param urlStr
* @param fileName
* @param savePath
* @throws IOException
*/
public static File downLoadFromUrl(String urlStr, String fileName,String savePath) throws IOException {
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)");
// 得到输入流
InputStream inputStream = conn.getInputStream();
// 获取自己数组
byte[] getData = readInputStream(inputStream);
// 文件保存位置
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(getData);
if (fos != null) {
fos.close();
}
if (inputStream != null) {
inputStream.close();
}
System.out.println("info:" + url + " download success");
return file;
}
/**
* 从输入流中获取字节数组
* @param inputStream
* @return
* @throws IOException
*/
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();
}
}