首次提交

2021-12-01 16:19:14 +08:00
parent 45eaa5af4a
commit 5b998bd30c
17 changed files with 1253 additions and 0 deletions
--- a/src/main/java/com/zdjizhi/syncfile/utils/HttpUtil.java
+++ b/src/main/java/com/zdjizhi/syncfile/utils/HttpUtil.java
@@ -0,0 +1,103 @@
+package com.zdjizhi.syncfile.utils;
+
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.log.Log;
+import cn.hutool.log.LogFactory;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.zdjizhi.syncfile.entity.PostFileResponse;
+import com.zdjizhi.syncfile.monitor.MonitorProperties;
+import org.apache.commons.io.IOUtils;
+import org.apache.http.client.config.RequestConfig;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.client.methods.HttpPost;
+import org.apache.http.entity.InputStreamEntity;
+import org.apache.http.impl.client.CloseableHttpClient;
+import org.apache.http.util.EntityUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.io.InputStream;
+
+@Component
+public class HttpUtil {
+    private static Log log = LogFactory.get();
+
+    @Autowired
+    private CloseableHttpClient httpClient;
+    @Autowired
+    private RequestConfig requestConfig;
+    @Autowired
+    MonitorProperties monitorProperties;
+
+    public InputStream httpGetInputStream(String url) {
+        InputStream result = null;
+        CloseableHttpResponse response = null;
+        try {
+            HttpGet httpGet = new HttpGet(url);
+            httpGet.setConfig(requestConfig);
+            response = httpClient.execute(httpGet);
+            if (response.getStatusLine().getStatusCode() == 200) {
+                result = IOUtils.toBufferedInputStream(response.getEntity().getContent());
+                log.info("get file success. current url: {}", url);
+                monitorProperties.addDownloadFileSuccessCount();
+                monitorProperties.addDownloadFileSize(Integer.parseInt(response.getFirstHeader("Content-Length").getValue()));
+            }else if (response.getStatusLine().getStatusCode() == 500){
+                log.error("get file error. Hos service error, please check hos. current url: {}", url);
+                monitorProperties.addHosError();
+            } else {
+                log.error("get file error. current url: {}, code: {}, msg: {}", url, response.getStatusLine().getStatusCode(), EntityUtils.toString(response.getEntity(), "UTF-8"));
+                monitorProperties.addFileSyncError();
+            }
+        } catch (Exception e) {
+            log.error("get file error. current url: {}, error: {}", url, e.toString());
+            monitorProperties.addFileSyncError();
+        } finally {
+            IoUtil.close(response);
+        }
+        return result;
+    }
+
+    public boolean httpPostInputStream(String url, InputStream data) {
+        boolean isSuccess = false;
+        CloseableHttpResponse response = null;
+        try {
+            HttpPost httpPost = new HttpPost(url);
+            httpPost.setConfig(requestConfig);
+            httpPost.setEntity(new InputStreamEntity(data));
+            response = httpClient.execute(httpPost);
+            if (response.getStatusLine().getStatusCode() == 200) {
+                String responseEntity = EntityUtils.toString(response.getEntity(), "UTF-8");
+                JSONObject jsonObj = (JSONObject) JSON.parse(responseEntity);
+                if(jsonObj!=null){
+                    if (responseEntity.contains("\"code\": 200")) {
+                        PostFileResponse postFileResponse = JSON.toJavaObject(jsonObj, PostFileResponse.class);
+                        isSuccess = true;
+                        log.info("post file success. current url: {}, msg: {}", url, responseEntity);
+                        monitorProperties.addPostFileSuccessCount();
+                        monitorProperties.addPostFileSize(postFileResponse.getData().getFileSize());
+                    } else {
+                        log.error("post file error. current url: {}, msg: {}", url,responseEntity);
+                        monitorProperties.addFileSyncError();
+                    }
+                }else {
+                    log.error("post file error, response body error. current url: {}", url);
+                    monitorProperties.addOssError();
+                }
+            } else if(response.getStatusLine().getStatusCode() == 500){
+                log.error("post file error. Oss service error.current url: {}, code: 500, msg: {}", url, response.getStatusLine().getStatusCode(), EntityUtils.toString(response.getEntity(), "UTF-8"));
+                monitorProperties.addOssError();
+            }else {
+                log.error("post file error. current url: {}, code: {}, msg: {}", url, response.getStatusLine().getStatusCode(), EntityUtils.toString(response.getEntity(), "UTF-8"));
+                monitorProperties.addFileSyncError();
+            }
+        } catch (Exception e) {
+            log.error("post file error. current url: " + url, e);
+            monitorProperties.addFileSyncError();
+        } finally {
+            IoUtil.close(response);
+        }
+        return isSuccess;
+    }
+}