|
@@ -0,0 +1,233 @@
|
|
|
+package org.jeecg.common.util;
|
|
|
+
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.apache.commons.io.IOUtils;
|
|
|
+
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.io.*;
|
|
|
+import java.net.HttpURLConnection;
|
|
|
+import java.net.URL;
|
|
|
+
|
|
|
+@Slf4j
|
|
|
+public class HttpDownUtils {
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从Minio下载文件
|
|
|
+ *
|
|
|
+ * @param fileName
|
|
|
+ * @param savePath
|
|
|
+ */
|
|
|
+ public static void downLoadFromMinio(String fileId, String fileName, String savePath) throws Exception {
|
|
|
+
|
|
|
+ InputStream inputStream = null;
|
|
|
+ FileOutputStream fos = null;
|
|
|
+ try {
|
|
|
+ //得到输入流
|
|
|
+ inputStream = MinioUtil.getMinioFile(fileId);
|
|
|
+ //获取自己数组
|
|
|
+ byte[] getData = readInputStream(inputStream);
|
|
|
+
|
|
|
+ //文件保存位置
|
|
|
+ File saveDir = new File(savePath);
|
|
|
+ if (!saveDir.exists()) {
|
|
|
+ saveDir.mkdirs();
|
|
|
+ }
|
|
|
+ File file = new File(saveDir + File.separator + fileName);
|
|
|
+ fos = new FileOutputStream(file);
|
|
|
+ fos.write(getData);
|
|
|
+ } finally {
|
|
|
+ if (fos != null) {
|
|
|
+ fos.close();
|
|
|
+ }
|
|
|
+ if (inputStream != null) {
|
|
|
+ inputStream.close();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从网络Url中下载文件
|
|
|
+ *
|
|
|
+ * @param urlStr
|
|
|
+ * @param fileName
|
|
|
+ * @param savePath
|
|
|
+ * @throws IOException
|
|
|
+ */
|
|
|
+ public static void downLoadFromUrl(String urlStr, String fileName, String savePath) throws Exception {
|
|
|
+ URL url = new URL(urlStr);
|
|
|
+ HttpURLConnection conn = (HttpURLConnection) url.openConnection();
|
|
|
+ //设置超时间为3秒
|
|
|
+ conn.setConnectTimeout(10 * 1000);
|
|
|
+ conn.setReadTimeout(30 * 1000);
|
|
|
+ //防止屏蔽程序抓取而返回403错误
|
|
|
+ conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
|
|
|
+ InputStream inputStream = null;
|
|
|
+ FileOutputStream fos = null;
|
|
|
+ try {
|
|
|
+ //得到输入流
|
|
|
+ inputStream = conn.getInputStream();
|
|
|
+ //获取自己数组
|
|
|
+ byte[] getData = readInputStream(inputStream);
|
|
|
+
|
|
|
+ //文件保存位置
|
|
|
+ File saveDir = new File(savePath);
|
|
|
+ if (!saveDir.exists()) {
|
|
|
+ saveDir.mkdirs();
|
|
|
+ }
|
|
|
+ File file = new File(saveDir + File.separator + fileName);
|
|
|
+ fos = new FileOutputStream(file);
|
|
|
+ fos.write(getData);
|
|
|
+ } finally {
|
|
|
+ if (fos != null) {
|
|
|
+ fos.close();
|
|
|
+ }
|
|
|
+ if (inputStream != null) {
|
|
|
+ inputStream.close();
|
|
|
+ }
|
|
|
+ if (conn != null) {
|
|
|
+ conn.disconnect();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从网络Url中下载文件,返回流
|
|
|
+ *
|
|
|
+ * @param urlStr
|
|
|
+ * @param fileName
|
|
|
+ * @throws IOException
|
|
|
+ */
|
|
|
+ public static void downLoadFromUrl(String urlStr, String fileName, HttpServletResponse response) {
|
|
|
+ OutputStream outputStream = null;
|
|
|
+ InputStream inputStream = null;
|
|
|
+ try {
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
+ response.addHeader("Content-Disposition", "attachment;fileName=" +
|
|
|
+ new String(fileName.getBytes("UTF-8"), "iso-8859-1"));
|
|
|
+ URL url = new URL(urlStr);
|
|
|
+ HttpURLConnection conn = (HttpURLConnection) url.openConnection();
|
|
|
+ //设置超时间为3秒
|
|
|
+ conn.setConnectTimeout(10 * 1000);
|
|
|
+ conn.setReadTimeout(30 * 1000);
|
|
|
+ //防止屏蔽程序抓取而返回403错误
|
|
|
+ conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
|
|
|
+
|
|
|
+ outputStream = response.getOutputStream();
|
|
|
+ //得到输入流
|
|
|
+ inputStream = conn.getInputStream();
|
|
|
+ //获取自己数组
|
|
|
+ byte[] getData = readInputStream(inputStream);
|
|
|
+ outputStream.write(getData);
|
|
|
+ response.flushBuffer();
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("{}下载失败:{}", fileName, e.getMessage());
|
|
|
+ e.printStackTrace();
|
|
|
+ } finally {
|
|
|
+ if (inputStream != null) {
|
|
|
+ try {
|
|
|
+ inputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (outputStream != null) {
|
|
|
+ try {
|
|
|
+ outputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从minio下载文件,返回流
|
|
|
+ *
|
|
|
+ * @param fileName
|
|
|
+ */
|
|
|
+ public static void downLoadFromMinio(String fileId, String fileName, HttpServletResponse response) {
|
|
|
+ OutputStream outputStream = null;
|
|
|
+ InputStream inputStream = null;
|
|
|
+ try {
|
|
|
+ response.setContentType("application/octet-stream");
|
|
|
+ response.addHeader("Content-Disposition", "attachment;fileName=" +
|
|
|
+ new String(fileName.getBytes("UTF-8"), "iso-8859-1"));
|
|
|
+ inputStream = MinioUtil.getMinioFile(fileId);
|
|
|
+ //获取自己数组
|
|
|
+ byte[] getData = readInputStream(inputStream);
|
|
|
+ outputStream = response.getOutputStream();
|
|
|
+ outputStream.write(getData);
|
|
|
+ response.flushBuffer();
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("{}下载失败:{}", fileName, e.getMessage());
|
|
|
+ e.printStackTrace();
|
|
|
+ } finally {
|
|
|
+ if (inputStream != null) {
|
|
|
+ try {
|
|
|
+ inputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (outputStream != null) {
|
|
|
+ try {
|
|
|
+ outputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从输入流中获取字节数组
|
|
|
+ *
|
|
|
+ * @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();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 从网络Url中读取文件内容
|
|
|
+ *
|
|
|
+ * @param urlStr
|
|
|
+ * @throws IOException
|
|
|
+ */
|
|
|
+ public static String readFromUrl(String urlStr) throws Exception {
|
|
|
+ InputStream inputStream = null;
|
|
|
+ try {
|
|
|
+ URL url = new URL(urlStr);
|
|
|
+ HttpURLConnection conn = (HttpURLConnection) url.openConnection();
|
|
|
+ //设置超时间为3秒
|
|
|
+ conn.setConnectTimeout(10 * 1000);
|
|
|
+ conn.setReadTimeout(30 * 1000);
|
|
|
+ //防止屏蔽程序抓取而返回403错误
|
|
|
+ conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
|
|
|
+
|
|
|
+ //得到输入流
|
|
|
+ inputStream = conn.getInputStream();
|
|
|
+ return IOUtils.toString(inputStream, "utf-8");
|
|
|
+ } finally {
|
|
|
+ if (inputStream != null) {
|
|
|
+ try {
|
|
|
+ inputStream.close();
|
|
|
+ } catch (IOException e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|