Changeset 661

Timestamp:

05/21/07 13:56:13 (2 years ago)

Author:

yuanying

Message:

HttpClientConnection? supports Follow Redirects.

Files:

• org.jalcedo.client/trunk/org.jalcedo.client.rest/src/org/jalcedo/client/rest/connect/HttpClientConnection.java (modified) (4 diffs)

org.jalcedo.client/trunk/org.jalcedo.client.rest/src/org/jalcedo/client/rest/connect/HttpClientConnection.java

@@ -14 +14 @@
-import java.io.OutputStream;
-import java.util.ArrayList;
+import java.util.HashMap;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-
+import org.apache.commons.httpclient.Header;
-import org.apache.commons.httpclient.HttpClient;
-import org.apache.commons.httpclient.HttpException;
-import org.apache.commons.httpclient.HttpMethod;
+import org.apache.commons.httpclient.HttpStatus;
-import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager;
-import org.apache.commons.httpclient.NameValuePair;
+import org.apache.commons.httpclient.URI;
+import org.apache.commons.httpclient.URIException;
-import org.apache.commons.httpclient.methods.DeleteMethod;
-import org.apache.commons.httpclient.methods.EntityEnclosingMethod;
@@ -42 +47 @@
-
-    protected final Log logger = LogFactory.getLog(this.getClass());
+    
+    private boolean followRedirects = true;
+
+    /**
+     * 
+     * @return
+     */
+    public synchronized boolean isFollowRedirects() {
+        return followRedirects;
+    }
+
+    /**
+     * 
+     * @param followRedirects
+     */
+    public synchronized void setFollowRedirects(boolean followRedirects) {
+        this.followRedirects = followRedirects;
+    }
-
-    /*
@@ -70 +93 @@
-        try {
-            GetMethod method = new GetMethod(url);
+            if (this.isFollowRedirects()) {
+                method.setFollowRedirects(true);
+            }
-            return doMethod(queryParams, method);
-        } catch (IOException e) {
@@ -90 +116 @@
-        this.execute(method);
-        this.checkHttpResponse(method);
+        if (this.isFollowRedirects() && this.needRedirect(method)) {
+            URI redirectUri = this.extractRedirectURI(method);
+            return this.get(redirectUri.toString(), new HashMap<String, String>());
+        }
-        return method.getResponseBodyAsString();
+    }
+    
+    /**
+     * 
+     * @param method
+     * @return
+     */
+    private URI extractRedirectURI(HttpMethod method) {
+        URI redirectUri = null;
+        Header header = method.getRequestHeader("location");
+        try {
+            redirectUri = new URI(header.getValue(), true);
+            if (!redirectUri.isRelativeURI()) {
+                redirectUri = new URI(method.getURI(), redirectUri);
+            }
+        } catch (URIException e) {
+            // TODO Auto-generated catch block
+            throw new RuntimeException(e);
+        } catch (NullPointerException e) {
+            // TODO Auto-generated catch block
+            throw new RuntimeException(e);
+        }
+        return redirectUri;
+    }
+
+    /**
+     * this http method needs redirect.
+     * @param method
+     * @return
+     */
+    private boolean needRedirect(HttpMethod method) {
+        switch (method.getStatusCode()) {
+        case HttpStatus.SC_MOVED_PERMANENTLY:
+        case HttpStatus.SC_MOVED_TEMPORARILY:
+        case HttpStatus.SC_SEE_OTHER:
+        case HttpStatus.SC_TEMPORARY_REDIRECT:
+            return true;
+        default:
+            return false;
+        }
-    }
-