:sparkles: 图片下载功能完善

.idea/misc.xml

@@ -1,4 +1,3 @@
-<?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
   <component name="ExternalStorageConfigurationManager" enabled="true" />
   <component name="MavenProjectsManager">

download-center-common/pom.xml

@@ -16,6 +16,7 @@
     <maven.compiler.target>8</maven.compiler.target>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     <hutool.version>5.5.2</hutool.version>
+    <commons-io.version>1.3.2</commons-io.version>
   </properties>
 
   <dependencies>
@@ -31,6 +32,11 @@
       <artifactId>hutool-all</artifactId>
       <version>${hutool.version}</version>
     </dependency>
+    <dependency>
+      <groupId>org.apache.commons</groupId>
+      <artifactId>commons-io</artifactId>
+      <version>${commons-io.version}</version>
+    </dependency>
   </dependencies>
 
 </project>

download-center-common/src/main/java/com/shanghaichengdi/downloadcentercommon/util/SSLUtil.java

@@ -0,0 +1,100 @@
+package com.shanghaichengdi.downloadcentercommon.util;
+
+import java.io.OutputStreamWriter;
+import java.net.URL;
+import java.net.URLConnection;
+import java.security.cert.CertificateException;
+import java.security.cert.X509Certificate;
+import javax.net.ssl.HostnameVerifier;
+import javax.net.ssl.HttpsURLConnection;
+import javax.net.ssl.SSLContext;
+import javax.net.ssl.SSLSession;
+import javax.net.ssl.TrustManager;
+import javax.net.ssl.X509TrustManager;
+import org.apache.commons.io.IOUtils;
+
+/**
+ * lag
+ */
+public class SSLUtil {
+    private static void trustAllHttpsCertificates() throws Exception {
+        TrustManager[] trustAllCerts = new TrustManager[1];
+        TrustManager tm = new miTM();
+        trustAllCerts[0] = tm;
+        SSLContext sc = SSLContext.getInstance("SSL");
+        sc.init(null, trustAllCerts, null);
+        HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory());
+    }
+
+    static class miTM implements TrustManager, X509TrustManager {
+        public X509Certificate[] getAcceptedIssuers() {
+            return null;
+        }
+
+        public boolean isServerTrusted(X509Certificate[] certs) {
+            return true;
+        }
+
+        public boolean isClientTrusted(X509Certificate[] certs) {
+            return true;
+        }
+
+        public void checkServerTrusted(X509Certificate[] certs, String authType)
+                throws CertificateException {
+            return;
+        }
+
+        public void checkClientTrusted(X509Certificate[] certs, String authType)
+                throws CertificateException {
+            return;
+        }
+    }
+
+    /**
+     * 忽略HTTPS请求的SSL证书，必须在openConnection之前调用
+     *
+     * @throws Exception
+     */
+    public static void ignoreSsl() throws Exception {
+        HostnameVerifier hv = new HostnameVerifier() {
+            public boolean verify(String urlHostName, SSLSession session) {
+                System.out.println("Warning: URL Host: " + urlHostName
+                        + " vs. " + session.getPeerHost());
+                return true;
+            }
+        };
+        trustAllHttpsCertificates();
+        HttpsURLConnection.setDefaultHostnameVerifier(hv);
+    }
+
+    public static String getRequest(String url, int timeOut) throws Exception {
+        URL u = new URL(url);
+        if ("https".equalsIgnoreCase(u.getProtocol())) {
+            SSLUtil.ignoreSsl();
+        }
+        URLConnection conn = u.openConnection();
+        conn.setConnectTimeout(timeOut);
+        conn.setReadTimeout(timeOut);
+        return IOUtils.toString(conn.getInputStream());
+    }
+
+    public static String postRequest(String urlAddress, String args, int timeOut)
+            throws Exception {
+        URL url = new URL(urlAddress);
+        if ("https".equalsIgnoreCase(url.getProtocol())) {
+            SSLUtil.ignoreSsl();
+        }
+        URLConnection u = url.openConnection();
+        u.setDoInput(true);
+        u.setDoOutput(true);
+        u.setConnectTimeout(timeOut);
+        u.setReadTimeout(timeOut);
+        OutputStreamWriter osw = new OutputStreamWriter(u.getOutputStream(),
+                "UTF-8");
+        osw.write(args);
+        osw.flush();
+        osw.close();
+        u.getOutputStream();
+        return IOUtils.toString(u.getInputStream());
+    }
+}

download-center-server/src/main/java/com/shanghaichengdi/downloadcenterserver/controller/PicDownloadController.java

@@ -1,10 +1,19 @@
 package com.shanghaichengdi.downloadcenterserver.controller;
 
 import com.alibaba.fastjson.JSONObject;
+import com.shanghaichengdi.downloadcentercommon.util.ImageDownload;
+import com.shanghaichengdi.downloadcentercommon.util.SSLUtil;
 import com.shanghaichengdi.downloadcenterserver.domain.PicDownload;
 import com.shanghaichengdi.downloadcenterserver.faced.PicDownloadFaced;
 import com.shanghaigeography.entity.ResultMsg;
 import com.shanghaigeography.eum.ResultState;
+import java.io.File;
+import java.net.URL;
+import java.net.URLDecoder;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.web.bind.annotation.CrossOrigin;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -32,4 +41,73 @@ public class PicDownloadController {
     }
     return new ResultMsg<String>().setMsg("下载成功").setCode(ResultState.SUCCESS).toJSON();
   }
+
+  @Deprecated
+  @PostMapping("/first/pictures")
+  public String downloadFirstPictures(@RequestBody String data) {
+    PicDownload picDownload = JSONObject.parseObject(data, PicDownload.class);
+    int resultNum = this.downloadPics(picDownload.getImageUrls(), picDownload.getPassTime());
+    if (resultNum != picDownload.getImageUrls().size()) {
+      return new ResultMsg<String>()
+          .setMsg("下载失败")
+          .setCode(ResultState.ERROR)
+          .setData(String.valueOf(resultNum))
+          .toJSON();
+    }
+    return new ResultMsg<String>()
+        .setMsg("下载成功")
+        .setCode(ResultState.SUCCESS)
+        .setData(String.valueOf(resultNum))
+        .toJSON();
+  }
+
+  @Deprecated
+  private int downloadPics(List<String> imageUrls, String passTime) {
+    String time = passTime.split("\\+")[0];
+    Date times;
+    try {
+      SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+      times =
+          formatter.parse(
+              time.split("\\.")[0].split("T")[0] + " " + time.split("\\.")[0].split("T")[1]);
+    } catch (ParseException e) {
+      e.printStackTrace();
+      return -1;
+    }
+    SimpleDateFormat formatter = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss");
+    SimpleDateFormat formatter1 = new SimpleDateFormat("yyyyMMdd_HHmmss");
+    String dir = formatter.format(times);
+    String newTime = formatter1.format(times) + time.split("\\.")[1];
+    String year = dir.split("\\_")[0];
+    String month = dir.split("\\_")[1];
+    String day = dir.split("\\_")[2];
+    String hour = dir.split("\\_")[3];
+    String dirs = year + "\\" + year + month + day + "\\" + hour;
+
+    int i = 0;
+    File photoDir = new File("\\\\K6XL82KY31YIWFC\\photo\\" + dirs);
+    photoDir.mkdirs();
+    for (String imageUrl : imageUrls) {
+      boolean isSuccess = false;
+      URL url = null;
+      try {
+        url = new URL(URLDecoder.decode(imageUrl, "utf-8"));
+        if ("https".equalsIgnoreCase(url.getProtocol())) {
+          SSLUtil.ignoreSsl();
+        }
+        String photoFile =
+            "\\\\K6XL82KY31YIWFC\\photo\\" + dirs + "\\" + "P" + i + "_" + newTime + ".jpg";
+        System.out.println(url);
+        isSuccess = ImageDownload.download(imageUrl, photoFile);
+        if (!isSuccess) {
+          return i;
+        }
+        i++;
+      } catch (Exception e) {
+        e.printStackTrace();
+        return i;
+      }
+    }
+    return i;
+  }
 }

download-center-server/src/main/java/com/shanghaichengdi/downloadcenterserver/faced/PicDownloadFaced.java

@@ -4,15 +4,16 @@ import com.shanghaichengdi.downloadcentercommon.util.ImageDownload;
 import java.io.File;
 import java.io.IOException;
 import java.net.URL;
-import java.net.URLDecoder;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
 import java.util.Date;
 import java.util.List;
+import lombok.extern.slf4j.Slf4j;
 
 /**
  * @author gengaoliu
  */
+@Slf4j
 public class PicDownloadFaced extends Thread {
   private List<String> imageUrls;
   private String passTime;
@@ -24,20 +25,19 @@ public class PicDownloadFaced extends Thread {
 
   @Override
   public void run() {
-    String time = passTime.split("\\+")[0];
+    String time = passTime;
     Date times = null;
     try {
       SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
       times =
-          formatter.parse(
-              time.split("\\.")[0].split("T")[0] + " " + time.split("\\.")[0].split("T")[1]);
+          formatter.parse(time);
     } catch (ParseException e) {
       e.printStackTrace();
     }
     SimpleDateFormat formatter = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss");
     SimpleDateFormat formatter1 = new SimpleDateFormat("yyyyMMdd_HHmmss");
     String dir = formatter.format(times);
-    String newTime = formatter1.format(times) + time.split("\\.")[1];
+    String newTime = formatter1.format(times);
     String year = dir.split("\\_")[0];
     String month = dir.split("\\_")[1];
     String day = dir.split("\\_")[2];
@@ -49,34 +49,19 @@ public class PicDownloadFaced extends Thread {
     for (String imageUrl : imageUrls) {
       boolean isSuccess = false;
       URL url = null;
-      URL url2 = null;
       try {
-        url = new URL(URLDecoder.decode("https://192.168.143.21" + imageUrl, "utf-8"));
-        url2 =
-            new URL(
-                URLDecoder.decode(
-                    "https://192.168.143.21" + imageUrl.replace("/origin", ""), "utf-8"));
         String photoFile = "/Users/gengaoliu/" + dirs + "/" + "P" + i + "_" + newTime + ".jpg";
-        String photoFile2 = "/Users/gengaoliu/" + dirs + "/" + "P" + i + "_" + newTime + "_1.jpg";
         while (!isSuccess) {
           try {
-//            isSuccess = ImageDownload.download("https://192.168.143.21" + imageUrl, photoFile);
             isSuccess = ImageDownload.download(imageUrl, photoFile);
           } catch (Exception e) {
-            e.printStackTrace();
-          }
-        }
-        isSuccess = false;
-        while (!isSuccess) {
-          try {
-//            isSuccess = ImageDownload.download("https://192.168.143.21" + imageUrl, photoFile2);
-            isSuccess = ImageDownload.download(imageUrl, photoFile2);
-          } catch (Exception e) {
+            log.error(e.getMessage());
             e.printStackTrace();
           }
         }
         i++;
-      } catch (IOException e) {
+      } catch (Exception e) {
+        log.error(e.getMessage());
         e.printStackTrace();
       }
     }