Skip to content

Commit

Permalink
Drop ResponseBase, remove one hierarchy level
Browse files Browse the repository at this point in the history
  • Loading branch information
slandelle committed Nov 19, 2015
1 parent b23004e commit 74b9881
Show file tree
Hide file tree
Showing 6 changed files with 173 additions and 318 deletions.
10 changes: 0 additions & 10 deletions client/src/main/java/org/asynchttpclient/HttpResponseStatus.java
Expand Up @@ -17,7 +17,6 @@
package org.asynchttpclient; package org.asynchttpclient;


import java.net.SocketAddress; import java.net.SocketAddress;
import java.util.List;


import org.asynchttpclient.uri.Uri; import org.asynchttpclient.uri.Uri;


Expand All @@ -43,15 +42,6 @@ public final Uri getUri() {
return uri; return uri;
} }


/**
* Prepare a {@link Response}
*
* @param headers {@link HttpResponseHeaders}
* @param bodyParts list of {@link HttpResponseBodyPart}
* @return a {@link Response}
*/
public abstract Response prepareResponse(HttpResponseHeaders headers, List<HttpResponseBodyPart> bodyParts);

/** /**
* Return the response status code * Return the response status code
* *
Expand Down
3 changes: 2 additions & 1 deletion client/src/main/java/org/asynchttpclient/Response.java
Expand Up @@ -17,6 +17,7 @@
package org.asynchttpclient; package org.asynchttpclient;


import org.asynchttpclient.cookie.Cookie; import org.asynchttpclient.cookie.Cookie;
import org.asynchttpclient.netty.NettyResponse;
import org.asynchttpclient.uri.Uri; import org.asynchttpclient.uri.Uri;


import io.netty.handler.codec.http.HttpHeaders; import io.netty.handler.codec.http.HttpHeaders;
Expand Down Expand Up @@ -199,7 +200,7 @@ public ResponseBuilder accumulate(HttpResponseBodyPart bodyPart) {
* @return a {@link Response} instance * @return a {@link Response} instance
*/ */
public Response build() { public Response build() {
return status == null ? null : status.prepareResponse(headers, bodyParts); return status == null ? null : new NettyResponse(status, headers, bodyParts);
} }


/** /**
Expand Down
144 changes: 0 additions & 144 deletions client/src/main/java/org/asynchttpclient/ResponseBase.java

This file was deleted.

138 changes: 132 additions & 6 deletions client/src/main/java/org/asynchttpclient/netty/NettyResponse.java
Expand Up @@ -13,41 +13,53 @@
*/ */
package org.asynchttpclient.netty; package org.asynchttpclient.netty;


import static io.netty.handler.codec.http.HttpHeaders.Names.*;
import static org.asynchttpclient.util.HttpUtils.*;
import static org.asynchttpclient.util.MiscUtils.isNonEmpty; import static org.asynchttpclient.util.MiscUtils.isNonEmpty;
import io.netty.handler.codec.http.HttpHeaders; import io.netty.handler.codec.http.HttpHeaders;


import java.io.ByteArrayInputStream; import java.io.ByteArrayInputStream;
import java.io.InputStream; import java.io.InputStream;
import java.net.SocketAddress;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.nio.charset.Charset; import java.nio.charset.Charset;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map;


import org.asynchttpclient.HttpResponseBodyPart; import org.asynchttpclient.HttpResponseBodyPart;
import org.asynchttpclient.HttpResponseHeaders; import org.asynchttpclient.HttpResponseHeaders;
import org.asynchttpclient.HttpResponseStatus; import org.asynchttpclient.HttpResponseStatus;
import org.asynchttpclient.ResponseBase; import org.asynchttpclient.Response;
import org.asynchttpclient.cookie.Cookie; import org.asynchttpclient.cookie.Cookie;
import org.asynchttpclient.cookie.CookieDecoder; import org.asynchttpclient.cookie.CookieDecoder;
import org.asynchttpclient.uri.Uri;


/** /**
* Wrapper around the {@link org.asynchttpclient.Response} API. * Wrapper around the {@link org.asynchttpclient.Response} API.
*/ */
public class NettyResponse extends ResponseBase { public class NettyResponse implements Response {

private final List<HttpResponseBodyPart> bodyParts;
private final HttpResponseHeaders headers;
private final HttpResponseStatus status;
private List<Cookie> cookies;


public NettyResponse(HttpResponseStatus status,// public NettyResponse(HttpResponseStatus status,//
HttpResponseHeaders headers,// HttpResponseHeaders headers,//
List<HttpResponseBodyPart> bodyParts) { List<HttpResponseBodyPart> bodyParts) {
super(status, headers, bodyParts); this.bodyParts = bodyParts;
this.headers = headers;
this.status = status;
} }


protected List<Cookie> buildCookies() { private List<Cookie> buildCookies() {


List<String> setCookieHeaders = headers.getHeaders().getAll(HttpHeaders.Names.SET_COOKIE2); List<String> setCookieHeaders = headers.getHeaders().getAll(SET_COOKIE2);


if (!isNonEmpty(setCookieHeaders)) { if (!isNonEmpty(setCookieHeaders)) {
setCookieHeaders = headers.getHeaders().getAll(HttpHeaders.Names.SET_COOKIE); setCookieHeaders = headers.getHeaders().getAll(SET_COOKIE);
} }


if (isNonEmpty(setCookieHeaders)) { if (isNonEmpty(setCookieHeaders)) {
Expand All @@ -63,6 +75,94 @@ protected List<Cookie> buildCookies() {
return Collections.emptyList(); return Collections.emptyList();
} }


@Override
public final int getStatusCode() {
return status.getStatusCode();
}

@Override
public final String getStatusText() {
return status.getStatusText();
}

@Override
public final Uri getUri() {
return status.getUri();
}

@Override
public SocketAddress getRemoteAddress() {
return status.getRemoteAddress();
}

@Override
public SocketAddress getLocalAddress() {
return status.getLocalAddress();
}

@Override
public final String getContentType() {
return headers != null ? getHeader(CONTENT_TYPE) : null;
}

@Override
public final String getHeader(String name) {
return headers != null ? getHeaders().get(name) : null;
}

@Override
public final List<String> getHeaders(String name) {
return headers != null ? getHeaders().getAll(name) : Collections.<String> emptyList();
}

@Override
public final HttpHeaders getHeaders() {
return headers != null ? headers.getHeaders() : HttpHeaders.EMPTY_HEADERS;
}

@Override
public final boolean isRedirected() {
switch (status.getStatusCode()) {
case 301:
case 302:
case 303:
case 307:
case 308:
return true;
default:
return false;
}
}

@Override
public List<Cookie> getCookies() {

if (headers == null) {
return Collections.emptyList();
}

if (cookies == null) {
cookies = buildCookies();
}
return cookies;

}

@Override
public boolean hasResponseStatus() {
return status != null;
}

@Override
public boolean hasResponseHeaders() {
return headers != null && !headers.getHeaders().isEmpty();
}

@Override
public boolean hasResponseBody() {
return isNonEmpty(bodyParts);
}

@Override @Override
public byte[] getResponseBodyAsBytes() { public byte[] getResponseBodyAsBytes() {
return getResponseBodyAsByteBuffer().array(); return getResponseBodyAsByteBuffer().array();
Expand All @@ -87,6 +187,17 @@ public String getResponseBody() {
return getResponseBody(null); return getResponseBody(null);
} }


private Charset computeCharset(Charset charset) {

if (charset == null) {
String contentType = getContentType();
if (contentType != null)
charset = parseCharset(contentType); // parseCharset can return
// null
}
return charset != null ? charset : DEFAULT_CHARSET;
}

@Override @Override
public String getResponseBody(Charset charset) { public String getResponseBody(Charset charset) {
return new String(getResponseBodyAsBytes(), computeCharset(charset)); return new String(getResponseBodyAsBytes(), computeCharset(charset));
Expand All @@ -96,4 +207,19 @@ public String getResponseBody(Charset charset) {
public InputStream getResponseBodyAsStream() { public InputStream getResponseBodyAsStream() {
return new ByteArrayInputStream(getResponseBodyAsBytes()); return new ByteArrayInputStream(getResponseBodyAsBytes());
} }

@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append(getClass().getSimpleName()).append(" {\n")//
.append("\tstatusCode=").append(getStatusCode()).append("\n")//
.append("\theaders=\n");

for (Map.Entry<String, String> header : getHeaders()) {
sb.append("\t\t").append(header.getKey()).append(": ").append(header.getValue()).append("\n");
}
sb.append("\tbody=\n").append(getResponseBody()).append("\n")//
.append("}").toString();
return sb.toString();
}
} }

0 comments on commit 74b9881

Please sign in to comment.