update offset and count

Sumate Kongpui
1 parent 72eef90c
Showing 1 changed file with 49 additions and 6 deletions Show diff stats
src/main/java/sourcecode/MockUp/MockUp.java
@@ -8,6 +8,7 @@ import java.nio.charset.Charset;
 import java.nio.charset.StandardCharsets;
 import java.sql.Date;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Iterator;
  
@@ -40,8 +41,8 @@ public class MockUp
 	private final static String PUT = "PUT";
 	private final static String DELETE = "DELETE";
 	private static int port = 6300;
-	private static String DBNAME = "coop_mockup";
-	private static String DBHOST = "10.1.2.144:27017";
+	private static String DBNAME = "sdf";
+	private static String DBHOST = "10.1.2.146:27017";
 	private static String DBUSERNAME = "";
 	private static String DBPASSWORD = "";
  
@@ -103,6 +104,18 @@ public class MockUp
  
 	        		}
 	        	}
+	        	
+	        	int offset = 0;
+	        	
+	        	if(!getKey("offset", url).equals(""))
+	        	{
+	        		try{
+	        			offset = Integer.parseInt(getKey("offset", url));
+	        		}catch(Exception e) 
+	        		{
+	        			
+	        		}
+	        	}
  
  
 	        	String[] fields = new String[0];
@@ -253,7 +266,10 @@ public class MockUp
 	        	}
  
 	        	//limit
-	        	responsJSON = limitData(responsJSON,limit);   	        	
+	        	if(method.equals(GET) && responsJSON.has("resultCode") && responsJSON.get("resultCode").equals("20000")) {
+	        		responsJSON = offsetData(responsJSON,offset);
+	        		responsJSON = limitData(responsJSON,limit);
+	        	}
  
 	        	//field
 	        	responsJSON = FieldsData(responsJSON,fields); 
@@ -550,8 +566,10 @@ public class MockUp
         		//resultData not array 
         		size = 1;
         	}
-        	responsJSON.remove("rowCount");
-        	responsJSON.put("rowCount", size);
+//        	responsJSON.remove("rowCount");
+        	if(!responsJSON.has("rowCount")) {
+            	responsJSON.put("rowCount", size);
+        	}
         }
  
     	return responsJSON;
@@ -690,5 +708,30 @@ public class MockUp
 		return value;
 	}
  
-    
+    private static JSONObject offsetData(JSONObject responsJSON, int offset) {
+    	
+    	try {
+    		System.out.println("Offset : " + offset);
+    		JSONArray JSONArrayData = responsJSON.getJSONArray("resultData");
+    		
+    		int n = JSONArrayData.length();
+    		
+    		ArrayList<Object> offsetResultData = new ArrayList<Object>();	
+
+    		for (int i = 0; i < n; i++)
+    		{
+    			if (i >= offset)
+    				offsetResultData.add(JSONArrayData.get(i));
+    		}
+    		
+    		responsJSON.remove("resultData");
+    		responsJSON.put("resultData",offsetResultData);		    
+    		
+    	} catch (Exception e) {
+			// TODO: handle exception
+		}
+    	
+    	
+    	return responsJSON;
+    }
 }
...	...	@@ -8,6 +8,7 @@ import java.nio.charset.Charset;
8	8	import java.nio.charset.StandardCharsets;
9	9	import java.sql.Date;
10	10	import java.util.ArrayList;
	11	+import java.util.Arrays;
11	12	import java.util.HashMap;
12	13	import java.util.Iterator;
13	14
...	...	@@ -40,8 +41,8 @@ public class MockUp
40	41	private final static String PUT = "PUT";
41	42	private final static String DELETE = "DELETE";
42	43	private static int port = 6300;
43		- private static String DBNAME = "coop_mockup";
44		- private static String DBHOST = "10.1.2.144:27017";
	44	+ private static String DBNAME = "sdf";
	45	+ private static String DBHOST = "10.1.2.146:27017";
45	46	private static String DBUSERNAME = "";
46	47	private static String DBPASSWORD = "";
47	48
...	...	@@ -103,6 +104,18 @@ public class MockUp
103	104
104	105	}
105	106	}
	107	+
	108	+ int offset = 0;
	109	+
	110	+ if(!getKey("offset", url).equals(""))
	111	+ {
	112	+ try{
	113	+ offset = Integer.parseInt(getKey("offset", url));
	114	+ }catch(Exception e)
	115	+ {
	116	+
	117	+ }
	118	+ }
106	119
107	120
108	121	String[] fields = new String[0];
...	...	@@ -253,7 +266,10 @@ public class MockUp
253	266	}
254	267
255	268	//limit
256		- responsJSON = limitData(responsJSON,limit);
	269	+ if(method.equals(GET) && responsJSON.has("resultCode") && responsJSON.get("resultCode").equals("20000")) {
	270	+ responsJSON = offsetData(responsJSON,offset);
	271	+ responsJSON = limitData(responsJSON,limit);
	272	+ }
257	273
258	274	//field
259	275	responsJSON = FieldsData(responsJSON,fields);
...	...	@@ -550,8 +566,10 @@ public class MockUp
550	566	//resultData not array
551	567	size = 1;
552	568	}
553		- responsJSON.remove("rowCount");
554		- responsJSON.put("rowCount", size);
	569	+// responsJSON.remove("rowCount");
	570	+ if(!responsJSON.has("rowCount")) {
	571	+ responsJSON.put("rowCount", size);
	572	+ }
555	573	}
556	574
557	575	return responsJSON;
...	...	@@ -690,5 +708,30 @@ public class MockUp
690	708	return value;
691	709	}
692	710
693		-
	711	+ private static JSONObject offsetData(JSONObject responsJSON, int offset) {
	712	+
	713	+ try {
	714	+ System.out.println("Offset : " + offset);
	715	+ JSONArray JSONArrayData = responsJSON.getJSONArray("resultData");
	716	+
	717	+ int n = JSONArrayData.length();
	718	+
	719	+ ArrayList<Object> offsetResultData = new ArrayList<Object>();
	720	+
	721	+ for (int i = 0; i < n; i++)
	722	+ {
	723	+ if (i >= offset)
	724	+ offsetResultData.add(JSONArrayData.get(i));
	725	+ }
	726	+
	727	+ responsJSON.remove("resultData");
	728	+ responsJSON.put("resultData",offsetResultData);
	729	+
	730	+ } catch (Exception e) {
	731	+ // TODO: handle exception
	732	+ }
	733	+
	734	+
	735	+ return responsJSON;
	736	+ }
694	737	}
...	...