io.vertx.ext.web.client.impl.HttpRequestImpl Maven / Gradle / Ivy
/*
* Copyright (c) 2011-2013 The original author or authors
* ------------------------------------------------------
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Apache License v2.0 which accompanies this distribution.
*
* The Eclipse Public License is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* The Apache License v2.0 is available at
* http://www.opensource.org/licenses/apache2.0.php
*
* You may elect to redistribute this code under either of these licenses.
*/
package io.vertx.ext.web.client.impl;
import io.netty.handler.codec.http.QueryStringDecoder;
import io.vertx.core.AsyncResult;
import io.vertx.core.Handler;
import io.vertx.core.MultiMap;
import io.vertx.core.buffer.Buffer;
import io.vertx.core.http.HttpHeaders;
import io.vertx.core.http.HttpMethod;
import io.vertx.core.http.RequestOptions;
import io.vertx.core.json.JsonObject;
import io.vertx.core.net.SocketAddress;
import io.vertx.core.streams.ReadStream;
import io.vertx.ext.auth.authentication.Credentials;
import io.vertx.ext.web.client.HttpRequest;
import io.vertx.ext.web.client.HttpResponse;
import io.vertx.ext.web.client.WebClientOptions;
import io.vertx.ext.web.client.predicate.ResponsePredicate;
import io.vertx.ext.web.codec.BodyCodec;
import io.vertx.ext.web.multipart.MultipartForm;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
/**
* @author Julien Viet
*/
public class HttpRequestImpl implements HttpRequest {
final WebClientInternal client;
final WebClientOptions options;
SocketAddress serverAddress;
MultiMap params;
HttpMethod method;
String protocol;
private Integer port;
private String host;
String virtualHost;
String uri;
MultiMap headers;
long timeout = -1;
BodyCodec codec;
boolean followRedirects;
Boolean ssl;
boolean multipartMixed = true;
public List expectations;
HttpRequestImpl(WebClientInternal client, HttpMethod method, SocketAddress serverAddress, Boolean ssl, Integer port, String host, String uri, BodyCodec
codec, WebClientOptions options) {
this(client, method, serverAddress, null, ssl, port, host, uri, codec, options);
}
HttpRequestImpl(WebClientInternal client, HttpMethod method, SocketAddress serverAddress, String protocol, Boolean ssl, Integer port, String host, String
uri, BodyCodec codec, WebClientOptions options) {
this.client = client;
this.method = method;
this.protocol = protocol;
this.codec = codec;
this.port = port;
this.host = host;
this.uri = uri;
this.ssl = ssl;
this.serverAddress = serverAddress;
this.followRedirects = options.isFollowRedirects();
this.options = options;
if (options.isUserAgentEnabled()) {
headers = HttpHeaders.set(HttpHeaders.USER_AGENT, options.getUserAgent());
} else {
headers = HttpHeaders.headers();
}
}
private HttpRequestImpl(HttpRequestImpl other) {
this.client = other.client;
this.serverAddress = other.serverAddress;
this.options = other.options;
this.method = other.method;
this.protocol = other.protocol;
this.port = other.port;
this.host = other.host;
this.timeout = other.timeout;
this.uri = other.uri;
this.headers = other.headers != null ? HttpHeaders.headers().addAll(other.headers) : HttpHeaders.headers();
this.params = other.params != null ? MultiMap.caseInsensitiveMultiMap().addAll(other.params) : null;
this.codec = other.codec;
this.followRedirects = other.followRedirects;
this.ssl = other.ssl;
this.multipartMixed = other.multipartMixed;
this.virtualHost = other.virtualHost;
this.expectations = other.expectations != null ? new ArrayList<>(other.expectations) : null;
}
@Override
public HttpRequest as(BodyCodec responseCodec) {
codec = (BodyCodec) responseCodec;
return (HttpRequest) this;
}
@Override
public HttpRequest method(HttpMethod value) {
method = value;
return this;
}
public HttpMethod method() {
return method;
}
@Override
public HttpRequest port(int value) {
port = value;
return this;
}
public int port() {
return port != null ? port : options.getDefaultPort();
}
@Override
public HttpRequest host(String value) {
host = value;
return this;
}
public String host() {
return host != null ? host : options.getDefaultHost();
}
@Override
public HttpRequest virtualHost(String value) {
virtualHost = value;
return this;
}
public String virtualHost() {
return virtualHost;
}
@Override
public HttpRequest uri(String value) {
params = null;
uri = value;
return this;
}
public String uri() {
return uri;
}
@Override
public HttpRequest putHeaders(MultiMap headers) {
headers().addAll(headers);
return this;
}
@Override
public HttpRequest putHeader(String name, String value) {
headers().set(name, value);
return this;
}
@Override
public HttpRequest putHeader(String name, Iterable value) {
headers().set(name, value);
return this;
}
@Override
public MultiMap headers() {
if (headers == null) {
headers = HttpHeaders.headers();
}
return headers;
}
@Override
public HttpRequest authentication(Credentials credentials) {
putHeader(
HttpHeaders.AUTHORIZATION.toString(),
credentials.toHttpAuthorization());
return this;
}
@Override
public HttpRequest ssl(Boolean value) {
ssl = value;
return this;
}
public Boolean ssl() {
return ssl;
}
@Override
public HttpRequest timeout(long value) {
timeout = value;
return this;
}
public long timeout() {
return timeout;
}
@Override
public HttpRequest addQueryParam(String paramName, String paramValue) {
queryParams().add(paramName, paramValue);
return this;
}
@Override
public HttpRequest setQueryParam(String paramName, String paramValue) {
queryParams().set(paramName, paramValue);
return this;
}
@Override
public HttpRequest followRedirects(boolean value) {
followRedirects = value;
return this;
}
public boolean followRedirects() {
return followRedirects;
}
@Override
public HttpRequest expect(ResponsePredicate expectation) {
if (expectations == null) {
expectations = new ArrayList<>();
}
expectations.add(expectation);
return this;
}
@Override
public MultiMap queryParams() {
if (params == null) {
params = MultiMap.caseInsensitiveMultiMap();
}
if (params.isEmpty()) {
int idx = uri.indexOf('?');
if (idx >= 0) {
QueryStringDecoder dec = new QueryStringDecoder(uri);
dec.parameters().forEach((name, value) -> params.add(name, value));
uri = uri.substring(0, idx);
}
}
return params;
}
@Override
public HttpRequest copy() {
return new HttpRequestImpl<>(this);
}
@Override
public HttpRequest multipartMixed(boolean allow) {
multipartMixed = allow;
return this;
}
@Override
public void sendStream(ReadStream body, Handler>> handler) {
send(null, body, handler);
}
@Override
public void send(Handler>> handler) {
send(null, null, handler);
}
@Override
public void sendBuffer(Buffer body, Handler>> handler) {
send(null, body, handler);
}
@Override
public void sendJsonObject(JsonObject body, Handler>> handler) {
send("application/json", body,handler);
}
@Override
public void sendJson(Object body, Handler>> handler) {
send("application/json", body, handler);
}
@Override
public void sendForm(MultiMap body, Handler>> handler) {
sendForm(body, "UTF-8", handler);
}
@Override
public void sendForm(MultiMap body, String charset, Handler>> handler) {
MultipartForm parts = MultipartForm.create();
for (Map.Entry attribute : body) {
parts.attribute(attribute.getKey(), attribute.getValue());
}
parts.setCharset(charset);
send("application/x-www-form-urlencoded", parts, handler);
}
@Override
public void sendMultipartForm(MultipartForm body, Handler>> handler) {
send("multipart/form-data", body, handler);
}
void send(String contentType, Object body, Handler>> handler) {
HttpContext ctx = client.createContext(handler);
ctx.prepareRequest(this, contentType, body);
}
void mergeHeaders(RequestOptions options) {
if (headers != null) {
MultiMap tmp = options.getHeaders();
if (tmp == null) {
tmp = MultiMap.caseInsensitiveMultiMap();
options.setHeaders(tmp);
}
tmp.addAll(headers);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy