Skip to content

Added toBuilder methods to Request #1673

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ public <T> ListenableFuture<T> executeRequest(Request request, AsyncHandler<T> h
try {
List<Cookie> cookies = config.getCookieStore().get(request.getUri());
if (!cookies.isEmpty()) {
RequestBuilder requestBuilder = new RequestBuilder(request);
RequestBuilder requestBuilder = request.toBuilder();
for (Cookie cookie : cookies) {
requestBuilder.addOrReplaceCookie(cookie);
}
Expand Down Expand Up @@ -264,7 +264,7 @@ private <T> FilterContext<T> preProcessRequest(FilterContext<T> fc) throws Filte
}

if (request.getRangeOffset() != 0) {
RequestBuilder builder = new RequestBuilder(request);
RequestBuilder builder = request.toBuilder();
builder.setHeader("Range", "bytes=" + request.getRangeOffset() + "-");
request = builder.build();
}
Expand Down
8 changes: 8 additions & 0 deletions client/src/main/java/org/asynchttpclient/Request.java
Original file line number Diff line number Diff line change
Expand Up @@ -180,4 +180,12 @@ public interface Request {
* @return the NameResolver to be used to resolve hostnams's IP
*/
NameResolver<InetAddress> getNameResolver();

/**
* @return a new request builder using this request as a prototype
*/
@SuppressWarnings("deprecation")
default RequestBuilder toBuilder() {
return new RequestBuilder(this);
}
}
5 changes: 5 additions & 0 deletions client/src/main/java/org/asynchttpclient/RequestBuilder.java
Original file line number Diff line number Diff line change
Expand Up @@ -39,10 +39,15 @@ public RequestBuilder(String method, boolean disableUrlEncoding, boolean validat
super(method, disableUrlEncoding, validateHeaders);
}

/**
* @deprecated Use request.toBuilder() instead
*/
@Deprecated
public RequestBuilder(Request prototype) {
super(prototype);
}

@Deprecated
public RequestBuilder(Request prototype, boolean disableUrlEncoding, boolean validateHeaders) {
super(prototype, disableUrlEncoding, validateHeaders);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ public Request adjustRequestRange(Request request) {
byteTransferred.set(resumableListener.length());
}

RequestBuilder builder = new RequestBuilder(request);
RequestBuilder builder = request.toBuilder();
if (request.getHeaders().get(RANGE) == null && byteTransferred.get() != 0) {
builder.setHeader(RANGE, "bytes=" + byteTransferred.get() + "-");
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ public boolean exitAfterHandlingConnect(Channel channel,

future.setReuseChannel(true);
future.setConnectAllowed(false);
Request targetRequest = new RequestBuilder(future.getTargetRequest()).build();
Request targetRequest = future.getTargetRequest().toBuilder().build();
if (whenHandshaked == null) {
requestSender.drainChannelAndExecuteNextRequest(channel, future, targetRequest);
} else {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,7 @@ public boolean exitAfterHandling407(Channel channel,
throw new IllegalStateException("Invalid Authentication scheme " + proxyRealm.getScheme());
}

RequestBuilder nextRequestBuilder = new RequestBuilder(future.getCurrentRequest()).setHeaders(requestHeaders);
RequestBuilder nextRequestBuilder = future.getCurrentRequest().toBuilder().setHeaders(requestHeaders);
if (future.getCurrentRequest().getUri().isSecured()) {
nextRequestBuilder.setMethod(CONNECT);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ public boolean exitAfterHandling401(final Channel channel,
throw new IllegalStateException("Invalid Authentication scheme " + realm.getScheme());
}

final Request nextRequest = new RequestBuilder(future.getCurrentRequest()).setHeaders(requestHeaders).build();
final Request nextRequest = future.getCurrentRequest().toBuilder().setHeaders(requestHeaders).build();

LOGGER.debug("Sending authentication to {}", request.getUri());
if (future.isKeepAlive()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ public void testEncodesQueryParameters() {
public void testChaining() {
Request request = get("http://foo.com").addQueryParam("x", "value").build();

Request request2 = new RequestBuilder(request).build();
Request request2 = request.toBuilder().build();

assertEquals(request2.getUri(), request.getUri());
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ public void replayResponseFilterTest() throws Exception {
ResponseFilter responseFilter = new ResponseFilter() {
public <T> FilterContext<T> filter(FilterContext<T> ctx) {
if (replay.getAndSet(false)) {
Request request = new RequestBuilder(ctx.getRequest()).addHeader("X-Replay", "true").build();
Request request = ctx.getRequest().toBuilder().addHeader("X-Replay", "true").build();
return new FilterContext.FilterContextBuilder<T>().asyncHandler(ctx.getAsyncHandler()).request(request).replayRequest(true).build();
}
return ctx;
Expand All @@ -123,7 +123,7 @@ public void replayStatusCodeResponseFilterTest() throws Exception {
ResponseFilter responseFilter = new ResponseFilter() {
public <T> FilterContext<T> filter(FilterContext<T> ctx) {
if (ctx.getResponseStatus() != null && ctx.getResponseStatus().getStatusCode() == 200 && replay.getAndSet(false)) {
Request request = new RequestBuilder(ctx.getRequest()).addHeader("X-Replay", "true").build();
Request request = ctx.getRequest().toBuilder().addHeader("X-Replay", "true").build();
return new FilterContext.FilterContextBuilder<T>().asyncHandler(ctx.getAsyncHandler()).request(request).replayRequest(true).build();
}
return ctx;
Expand All @@ -145,7 +145,7 @@ public void replayHeaderResponseFilterTest() throws Exception {
ResponseFilter responseFilter = new ResponseFilter() {
public <T> FilterContext<T> filter(FilterContext<T> ctx) {
if (ctx.getResponseHeaders() != null && ctx.getResponseHeaders().get("Ping").equals("Pong") && replay.getAndSet(false)) {
Request request = new RequestBuilder(ctx.getRequest()).addHeader("Ping", "Pong").build();
Request request = ctx.getRequest().toBuilder().addHeader("Ping", "Pong").build();
return new FilterContext.FilterContextBuilder<T>().asyncHandler(ctx.getAsyncHandler()).request(request).replayRequest(true).build();
}
return ctx;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -275,7 +275,7 @@ public Future<Response> options(BodyConsumer bodyConsumer, ThrowableHandler thro
}

private RequestBuilder rebuildRequest(Request rb) {
return new RequestBuilder(rb);
return rb.toBuilder();
}

private Future<Response> execute(RequestBuilder rb, BodyConsumer bodyConsumer, ThrowableHandler throwableHandler) throws IOException {
Expand Down Expand Up @@ -422,7 +422,7 @@ public Builder() {
}

private Builder(SimpleAsyncHttpClient client) {
this.requestBuilder = new RequestBuilder(client.requestBuilder.build());
this.requestBuilder = client.requestBuilder.build().toBuilder();
this.defaultThrowableHandler = client.defaultThrowableHandler;
this.errorDocumentBehaviour = client.errorDocumentBehaviour;
this.enableResumableDownload = client.resumeEnabled;
Expand Down