doc work, new example for expanding shortened urls

2013-08-11 01:27:27 +03:00
parent 1f5cf7151e
commit 1157dc6b58
6 changed files with 335 additions and 217 deletions
@@ -0,0 +1,102 @@
+package org.baeldung.live;
+
+import static org.hamcrest.Matchers.equalTo;
+import static org.junit.Assert.assertThat;
+
+import java.io.IOException;
+import java.io.InputStream;
+
+import org.apache.http.Header;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpHeaders;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.methods.HttpGet;
+import org.apache.http.impl.client.DefaultHttpClient;
+import org.apache.http.params.BasicHttpParams;
+import org.apache.http.params.HttpParams;
+import org.apache.http.util.EntityUtils;
+import org.junit.Before;
+import org.junit.Test;
+
+import com.google.common.base.Preconditions;
+
+public class HttpLiveServiceTemp {
+
+    private DefaultHttpClient client;
+
+    // fixtures
+
+    @Before
+    public final void before() {
+        final HttpParams httpParameters = new BasicHttpParams();
+        httpParameters.setParameter("http.protocol.handle-redirects", false);
+
+        client = new DefaultHttpClient(httpParameters);
+    }
+
+    // tests
+
+    @Test
+    public final void givenShortenedOnce_whenUrlIsUnshortened_thenCorrectResult() throws IOException {
+        final String expectedResult = "http://www.baeldung.com/rest-versioning";
+        final String actualResult = expandSingleLevel("http://bit.ly/13jEoS1");
+        assertThat(actualResult, equalTo(expectedResult));
+    }
+
+    @Test
+    public final void givenShortenedMultiple_whenUrlIsUnshortened_thenCorrectResult() throws IOException {
+        final String expectedResult = "http://www.baeldung.com/rest-versioning";
+        final String actualResult = expand("http://t.co/e4rDDbnzmk");
+        assertThat(actualResult, equalTo(expectedResult));
+    }
+
+    // API
+
+    final String expand(final String urlArg) throws IOException {
+        String originalUrl = urlArg;
+        String newUrl = expandSingleLevel(originalUrl);
+        while (!originalUrl.equals(newUrl)) {
+            originalUrl = newUrl;
+            newUrl = expandSingleLevel(originalUrl);
+        }
+
+        return newUrl;
+    }
+
+    final String expandSingleLevel(final String url) throws IOException {
+        HttpGet request = null;
+        HttpEntity httpEntity = null;
+        InputStream entityContentStream = null;
+
+        try {
+            request = new HttpGet(url);
+            final HttpResponse httpResponse = client.execute(request);
+
+            httpEntity = httpResponse.getEntity();
+            entityContentStream = httpEntity.getContent();
+
+            final int statusCode = httpResponse.getStatusLine().getStatusCode();
+            if (statusCode != 301 && statusCode != 302) {
+                return url;
+            }
+            final Header[] headers = httpResponse.getHeaders(HttpHeaders.LOCATION);
+            Preconditions.checkState(headers.length == 1);
+            final String newUrl = headers[0].getValue();
+
+            return newUrl;
+        } catch (final IllegalArgumentException uriEx) {
+            return url;
+        } finally {
+            if (request != null) {
+                request.releaseConnection();
+            }
+            if (entityContentStream != null) {
+                entityContentStream.close();
+            }
+            if (httpEntity != null) {
+                EntityUtils.consume(httpEntity);
+            }
+        }
+    }
+
+}