Adding batch request content files

Author: deepak2016

src/main/java/com/microsoft/graph/content/MSBatchRequestContent.java

@@ -0,0 +1,107 @@
+package com.microsoft.graph.content;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.http.Header;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpRequest;
+import org.apache.http.client.methods.HttpEntityEnclosingRequestBase;
+import org.apache.http.util.EntityUtils;
+import org.json.simple.JSONObject;
+import org.json.simple.JSONValue;
+
+public class MSBatchRequestContent {
+	private List<MSBatchRequestStep> batchRequestStepsArray;
+	private final int maxNumberOfRequests = 20;
+	
+	public MSBatchRequestContent(List<MSBatchRequestStep> batchRequestStepsArray) {
+		batchRequestStepsArray = new ArrayList<>();
+		if(batchRequestStepsArray.size() <= maxNumberOfRequests) {
+			for(MSBatchRequestStep requestStep: batchRequestStepsArray)
+				addBatchRequestStep(requestStep);
+		}
+	}
+	
+	public MSBatchRequestContent() {
+		batchRequestStepsArray = new ArrayList<>();
+	}
+	
+	public void addBatchRequestStep(MSBatchRequestStep batchRequestStep) {
+		if(batchRequestStep.getRequestId().compareTo("") == 0)
+			return;
+		if(batchRequestStepsArray.size() == maxNumberOfRequests)
+			return;
+		for(MSBatchRequestStep requestStep: batchRequestStepsArray) {
+			if(batchRequestStep.getRequestId().compareTo(requestStep.getRequestId()) == 0)
+				return;
+		}
+		batchRequestStepsArray.add(batchRequestStep);
+	}
+	
+	public void removeBatchRequesStepWithId(String requestId) {
+		for (int i = batchRequestStepsArray.size()-1; i >= 0; i--)
+	    {
+	        MSBatchRequestStep requestStep = batchRequestStepsArray.get(i);
+	        for (int j = requestStep.getArrayOfDependsOnIds().size() - 1; j >= 0; j--)
+	        {
+	            String dependsOnId = requestStep.getArrayOfDependsOnIds().get(j);
+	            if(dependsOnId.compareTo(requestId) == 0)
+	            {
+	                requestStep.getArrayOfDependsOnIds().remove(j);
+	            }
+	        }
+	        if(requestId.compareTo(requestStep.getRequestId()) == 0)
+	            batchRequestStepsArray.remove(i);
+	    }
+	}
+	
+	public String getBatchRequestContent() {
+		Map<String, List<Map<String, String>>> batchRequestContentMap = new HashMap();
+		List<Map<String, String>> batchContentArray = new ArrayList();
+		for(MSBatchRequestStep requestStep : batchRequestStepsArray) {
+			batchContentArray.add(getBatchRequestMapFromRequestStep(requestStep));
+		}
+		batchRequestContentMap.put("requests", batchContentArray);
+		return JSONValue.toJSONString(batchRequestContentMap);
+	}
+	
+	private Map<String, String> getBatchRequestMapFromRequestStep(MSBatchRequestStep batchRequestStep){
+		Map<String, String> contentmap = new HashMap();
+		contentmap.put("id", batchRequestStep.getRequestId());
+		contentmap.put("url", batchRequestStep.getRequest().getRequestLine().getUri());
+		contentmap.put("method", batchRequestStep.getRequest().getRequestLine().getMethod());
+		Header[] headers = batchRequestStep.getRequest().getAllHeaders();
+		if(headers != null) {
+			JSONObject obj = new JSONObject();
+			for(Header header: headers) {
+				obj.put(header.getName(), header.getValue());
+			}
+			contentmap.put("headers", obj.toJSONString());
+		}
+		HttpEntity entity = null;
+		HttpRequest request = batchRequestStep.getRequest();
+		if(request instanceof HttpEntityEnclosingRequestBase) {
+			HttpEntityEnclosingRequestBase httprequest = (HttpEntityEnclosingRequestBase)request;
+			entity = httprequest.getEntity();
+		}
+		if(entity != null) {
+			try {
+				String body = EntityUtils.toString(entity);
+				contentmap.put("body", body);
+			}
+			catch(Exception e) {
+			}
+		}
+		
+		List<String> arrayOfDependsOnIds = batchRequestStep.getArrayOfDependsOnIds();
+		if(arrayOfDependsOnIds != null) {
+			contentmap.put("dependsOn", JSONValue.toJSONString(arrayOfDependsOnIds));
+		}
+		
+		return contentmap;
+	}
+
+}

src/main/java/com/microsoft/graph/content/MSBatchRequestStep.java

@@ -0,0 +1,29 @@
+package com.microsoft.graph.content;
+
+import java.util.List;
+
+import org.apache.http.HttpRequest;
+
+public class MSBatchRequestStep {
+	private String requestId;
+	private HttpRequest request;
+	private List<String> arrayOfDependsOnIds;
+	
+	public MSBatchRequestStep(String requestId, HttpRequest request, List<String> arrayOfDependsOnIds) {
+		this.requestId = requestId;
+		this.request = request;
+		this.arrayOfDependsOnIds = arrayOfDependsOnIds;
+	}
+	
+	public String getRequestId() {
+		return requestId;
+	}
+	
+	public HttpRequest getRequest() {
+		return request;
+	}
+	
+	public List<String> getArrayOfDependsOnIds(){
+		return arrayOfDependsOnIds;
+	}
+}

src/main/java/com/microsoft/graph/content/MSBatchResponseContent.java

@@ -0,0 +1,31 @@
+package com.microsoft.graph.content;
+
+import org.json.simple.JSONObject;
+import org.json.simple.parser.JSONParser;
+import org.json.simple.parser.ParseException;
+
+public class MSBatchResponseContent {
+
+	private JSONObject batchResponseObj;
+	
+	public MSBatchResponseContent(String batchResponseData ) {
+		JSONParser parser = new JSONParser();
+		try {
+			batchResponseObj = (JSONObject) parser.parse(batchResponseData);
+		}
+		catch(ParseException e) {
+		}
+	}
+	
+	public String getResponseById(String requestId) {
+		 if(batchResponseObj.get(requestId) != null)
+			 return batchResponseObj.get(requestId).toString();
+		 return null;
+	}
+	
+	public String getResponses() {
+		if(batchResponseObj != null)
+			return batchResponseObj.toJSONString();
+		return null;
+	}
+}