Add retry support.

Author: androidquery

src/com/androidquery/callback/AbstractAjaxCallback.java

@@ -150,6 +150,7 @@ public abstract class AbstractAjaxCallback<T, K> implements Runnable{
 	private HttpUriRequest request;
 
 	private boolean uiCallback = true;
+	private int retry = 0;
 
 	@SuppressWarnings("unchecked")
 	private K self(){
@@ -288,6 +289,11 @@ public K timeout(int timeout){
 		return self();
 	}
 
+	public K retry(int retry){
+		this.retry = retry;
+		return self();
+	}
+	
 	/**
 	 * Set the transformer that transform raw data to desired type.
 	 * If not set, default transformer will be used.
@@ -1079,7 +1085,7 @@ private void networkWork(){
 
 		try{
 
-			network();
+			network(retry + 1);
 
 			if(ah != null && ah.expired(this, status) && !reauth){
 				AQUtility.debug("reauth needed", status.getMessage());	
@@ -1212,6 +1218,31 @@ private static Map<String, Object> extractParams(Uri uri){
 		return params;
 	}
 
+	//added retry logic
+	private void network(int attempts) throws IOException{
+		
+		if(attempts <= 1){
+			network();
+			return;
+		}
+				
+		for(int i = 0; i < attempts; i++){
+		
+			try{
+				network();
+				return;
+			}catch(IOException e){
+				if(i == attempts - 1){
+					throw e;
+				}
+			}
+			
+			
+			
+		}
+		
+		
+	}
 
 	private void network() throws IOException{
 

tests/src/com/androidquery/test/AQueryAsyncTest.java

@@ -1673,4 +1673,42 @@ public void testAjaxActiveCount() {
         assertNotNull(jo.opt("responseData"));
 
     }
+	
+	public void testRetryFailed() {
+		
+		String url = "http://www.androidquery.com/p/retry?wait=5000";
+	
+		AjaxCallback<JSONObject> cb = new AjaxCallback<JSONObject>();
+		cb.weakHandler(this, "jsonCb").timeout(1000);
+		
+		aq.ajax(url, JSONObject.class, cb);
+        
+        waitAsync();
+       
+        JSONObject jo = (JSONObject) result;
+        
+        assertNull(jo);
+        assertEquals(-101, status.getCode());
+        
+        
+	}
+	
+	public void testRetryOk() {
+		
+		String url = "http://www.androidquery.com/p/retry?wait=3000";
+	
+		AjaxCallback<JSONObject> cb = new AjaxCallback<JSONObject>();
+		cb.retry(1).weakHandler(this, "jsonCb").timeout(1000);
+		
+		aq.ajax(url, JSONObject.class, cb);
+        
+        waitAsync();
+       
+        JSONObject jo = (JSONObject) result;
+        
+        assertNotNull(jo);
+        assertEquals(200, status.getCode());
+        
+        
+	}
 }