Forráskód Böngészése

更新 WebRequest 的 upload 接口, 添加 token 参数

reghao 2 éve
szülő
commit
c0f931c836

+ 1 - 1
jdk/src/main/java/cn/reghao/jutil/jdk/http/WebClient.java

@@ -68,7 +68,7 @@ public class WebClient implements WebRequest {
     }
 
     @Override
-    public WebResponse upload(String url, UploadParam uploadParam) {
+    public WebResponse upload(String url, UploadParam uploadParam, String token) {
         return null;
     }
 

+ 1 - 1
jdk/src/main/java/cn/reghao/jutil/jdk/http/WebRequest.java

@@ -14,6 +14,6 @@ public interface WebRequest {
     WebResponse get(String url);
     WebResponse postFormData(String url, Map<String, String> formData);
     WebResponse postJson(String url, String json);
-    WebResponse upload(String url, UploadParam uploadParam);
+    WebResponse upload(String url, UploadParam uploadParam, String token);
     void download(String url, String referer, String filePath) throws IOException;
 }

+ 4 - 33
tool/src/main/java/cn/reghao/jutil/tool/http/DefaultWebRequest.java

@@ -108,7 +108,7 @@ public class DefaultWebRequest extends BaseWebRequest implements WebRequest {
     }
 
     @Override
-    public WebResponse upload(String url, UploadParam uploadParam) {
+    public WebResponse upload(String url, UploadParam uploadParam, String token) {
         MultipartEntityBuilder builder = MultipartEntityBuilder.create();
         builder.setMode(HttpMultipartMode.BROWSER_COMPATIBLE);
         builder.setCharset(StandardCharsets.UTF_8);
@@ -129,6 +129,9 @@ public class DefaultWebRequest extends BaseWebRequest implements WebRequest {
         }
 
         HttpPost post = new HttpPost(url);
+        if (token != null) {
+            post.addHeader("Authorization", "Bearer " + token);
+        }
         if (headers != null) {
             headers.forEach(post::addHeader);
         }
@@ -142,38 +145,6 @@ public class DefaultWebRequest extends BaseWebRequest implements WebRequest {
         }
     }
 
-    public WebResponse upload1(String url, UploadParam uploadParam, long userId) {
-        MultipartEntityBuilder builder = MultipartEntityBuilder.create();
-        builder.setMode(HttpMultipartMode.BROWSER_COMPATIBLE);
-        builder.setCharset(StandardCharsets.UTF_8);
-
-        ContentBody contentBody;
-        if (uploadParam.getFile() != null) {
-            contentBody = new FileBody(uploadParam.getFile());
-        } else if (uploadParam.getBytes() != null) {
-            contentBody = new ByteArrayBody(uploadParam.getBytes(), uploadParam.getMimeType());
-        } else {
-            return new WebResponse(600, "not data in UploadParam");
-        }
-
-        builder.addPart("file", contentBody);
-        Map<String, String> map = uploadParam.getTextParams();
-        if (map != null) {
-            map.forEach(builder::addTextBody);
-        }
-
-        HttpPost post = new HttpPost(url);
-        post.addHeader("x-user-id", ""+userId);
-        post.setEntity(builder.build());
-        try (CloseableHttpResponse response = client.execute(post)) {
-            int statusCode = response.getStatusLine().getStatusCode();
-            String body = EntityUtils.toString(response.getEntity(), StandardCharsets.UTF_8);;
-            return new WebResponse(statusCode, body);
-        } catch (Exception e) {
-            return new WebResponse(600, e.getMessage());
-        }
-    }
-
     private WebResponse execRequest(HttpRequestBase request) {
         if (headers != null) {
             headers.forEach(request::addHeader);