[master] - overrided POST method

Author: dzaiats

src/main/java/http/connections/ConnectionBuilder.java

@@ -97,6 +97,62 @@ public CloseableHttpResponse sendPOST(Map<Object, Object> map, String endpoint,
         return httpResponse;
     }
 
+    public CloseableHttpResponse sendPOST(Map<Object, Object> map, Map<String, String> headers, String endpoint, String token, boolean withMediaFile) throws IOException {
+        HttpClientBuilder httpClientBuilder = getHttpClientBuilder();
+
+        HttpClient httpClient = httpClientBuilder.build();
+
+        System.out.println("\n\n------------------------------------------------------------");
+        System.out.println("POST request to: " + url + endpoint);
+        HttpPost request = new HttpPost(url + endpoint);
+
+        for (Map.Entry<String, String> entry : headers.entrySet()) {
+            request.addHeader(entry.getKey(), entry.getValue());
+        }
+        String fileName = null;
+        File file = null;
+        if (withMediaFile) {
+            String boundary = "-------------" + System.currentTimeMillis();
+            fileName = (String) map.get("file");
+
+            try {
+                file = Helper.createFile(fileName);
+
+            } catch (AWTException e) {
+                e.printStackTrace();
+            }
+
+            MultipartEntityBuilder entity = MultipartEntityBuilder.create()
+                    .setMode(HttpMultipartMode.BROWSER_COMPATIBLE)
+                    .setBoundary(boundary);
+
+            if (fileName != null) {
+                entity.addPart("file", new ByteArrayBody(Files.readAllBytes(file.toPath()), fileName));
+            }
+
+            for (Map.Entry entry: map.entrySet()){
+                entity.addPart(String.valueOf(entry.getKey()), new StringBody((String) entry.getValue()));
+            }
+
+
+            request.setEntity(entity.build());
+            request.setHeader("Content-type", "multipart/form-data; boundary=" + boundary);
+        } else {
+            StringEntity params = new StringEntity(JSONObject.toJSONString(map));
+            System.out.println("with data: " + JSONObject.toJSONString(map));
+            request.addHeader("Content-Type", "application/json;charset=UTF-8");
+            request.setEntity(params);
+        }
+
+        CloseableHttpResponse httpResponse = (CloseableHttpResponse) httpClient.execute(request);
+
+        if (file != null) {
+            file.delete();
+        }
+
+        return httpResponse;
+    }
+
     private HttpClientBuilder getHttpClientBuilder() {
         SSLContext sslContext = null;
         try {
@@ -189,6 +245,22 @@ protected Map<Integer, String> getPOST(Map map, String endpoint, String token, b
         return mapResult;
     }
 
+    protected Map<Integer, String> getPOST(Map map, Map headers, String endpoint, String token, boolean withMediaFile) throws IOException {
+        Map<Integer, String> mapResult = new HashMap<Integer, String>();
+
+        CloseableHttpResponse response = sendPOST(map, headers, endpoint, token, withMediaFile);
+
+        StringBuffer body = getBodyResponse(response);
+
+        mapResult.put(response.getStatusLine().getStatusCode(), body.toString());
+
+        System.out.println("\n\nResponse:");
+        System.out.println(response.getStatusLine());
+        System.out.println(body.toString());
+
+        return mapResult;
+    }
+
     protected Map<Integer, String> getGET(String endpoint, String token) throws IOException {
         Map<Integer, String> mapResult = new HashMap<Integer, String>();
 

src/main/java/http/connections/ConnectionFactory.java

@@ -22,6 +22,10 @@ public Map<Integer, String> sendPost(Map map, String endpoint, String token, boo
         return new ConnectionBuilder(url).getPOST(map, endpoint, token, withMediaFile);
     }
 
+    public Map<Integer, String> sendPost(Map map, Map headers, String endpoint, String token, boolean withMediaFile) throws IOException {
+        return new ConnectionBuilder(url).getPOST(map, headers, endpoint, token, withMediaFile);
+    }
+
     public Map<Integer, String> sendGet(String endpoint, String token) throws IOException {
         return new ConnectionBuilder(url).getGET(endpoint, token);
     }