package org.glassfish.jersey.client;
import java.security.KeyStore;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.core.Configuration;
import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.SSLContext;
import org.glassfish.jersey.SslConfigurator;
import org.glassfish.jersey.client.internal.LocalizationMessages;
import org.glassfish.jersey.internal.util.collection.UnsafeValue;
import org.glassfish.jersey.internal.util.collection.Values;
public class JerseyClientBuilder extends ClientBuilder {
private final ClientConfig config;
private HostnameVerifier hostnameVerifier;
private SslConfigurator sslConfigurator;
private SSLContext sslContext;
public static JerseyClient createClient() {
return new JerseyClientBuilder().build();
}
public static JerseyClient createClient(Configuration configuration) {
return new JerseyClientBuilder().withConfig(configuration).build();
}
public JerseyClientBuilder() {
this.config = new ClientConfig();
}
@Override
public JerseyClientBuilder sslContext(SSLContext sslContext) {
if (sslContext == null) {
throw new NullPointerException(LocalizationMessages.NULL_SSL_CONTEXT());
}
this.sslContext = sslContext;
sslConfigurator = null;
return this;
}
@Override
public JerseyClientBuilder keyStore(KeyStore keyStore, char[] password) {
if (keyStore == null) {
throw new NullPointerException(LocalizationMessages.NULL_KEYSTORE());
}
if (password == null) {
throw new NullPointerException(LocalizationMessages.NULL_KEYSTORE_PASWORD());
}
if (sslConfigurator == null) {
sslConfigurator = SslConfigurator.newInstance();
}
sslConfigurator.keyStore(keyStore);
sslConfigurator.keyPassword(password);
sslContext = null;
return this;
}
@Override
public JerseyClientBuilder trustStore(KeyStore trustStore) {
if (trustStore == null) {
throw new NullPointerException(LocalizationMessages.NULL_TRUSTSTORE());
}
if (sslConfigurator == null) {
sslConfigurator = SslConfigurator.newInstance();
}
sslConfigurator.trustStore(trustStore);
sslContext = null;
return this;
}
@Override
public JerseyClientBuilder hostnameVerifier(HostnameVerifier hostnameVerifier) {
this.hostnameVerifier = hostnameVerifier;
return this;
}
@Override
public ClientBuilder executorService(ExecutorService executorService) {
config.executorService(executorService);
return this;
}
@Override
public ClientBuilder scheduledExecutorService(ScheduledExecutorService scheduledExecutorService) {
config.scheduledExecutorService(scheduledExecutorService);
return this;
}
@Override
public ClientBuilder connectTimeout(long timeout, TimeUnit unit) {
if (timeout < 0) {
throw new IllegalArgumentException("Negative timeout.");
}
this.property(ClientProperties.CONNECT_TIMEOUT, Math.toIntExact(unit.toMillis(timeout)));
return this;
}
@Override
public ClientBuilder readTimeout(long timeout, TimeUnit unit) {
if (timeout < 0) {
throw new IllegalArgumentException("Negative timeout.");
}
this.property(ClientProperties.READ_TIMEOUT, Math.toIntExact(unit.toMillis(timeout)));
return this;
}
@Override
public JerseyClient build() {
if (sslContext != null) {
return new JerseyClient(config, sslContext, hostnameVerifier, null);
} else if (sslConfigurator != null) {
final SslConfigurator sslConfiguratorCopy = sslConfigurator.copy();
return new JerseyClient(
config,
Values.lazy(new UnsafeValue<SSLContext, IllegalStateException>() {
@Override
public SSLContext get() {
return sslConfiguratorCopy.createSSLContext();
}
}),
hostnameVerifier);
} else {
return new JerseyClient(config, (UnsafeValue<SSLContext, IllegalStateException>) null, hostnameVerifier);
}
}
@Override
public ClientConfig getConfiguration() {
return config;
}
@Override
public JerseyClientBuilder property(String name, Object value) {
this.config.property(name, value);
return this;
}
@Override
public JerseyClientBuilder register(Class<?> componentClass) {
this.config.register(componentClass);
return this;
}
@Override
public JerseyClientBuilder register(Class<?> componentClass, int priority) {
this.config.register(componentClass, priority);
return this;
}
@Override
public JerseyClientBuilder register(Class<?> componentClass, Class<?>... contracts) {
this.config.register(componentClass, contracts);
return this;
}
@Override
public JerseyClientBuilder register(Class<?> componentClass, Map<Class<?>, Integer> contracts) {
this.config.register(componentClass, contracts);
return this;
}
@Override
public JerseyClientBuilder register(Object component) {
this.config.register(component);
return this;
}
@Override
public JerseyClientBuilder register(Object component, int priority) {
this.config.register(component, priority);
return this;
}
@Override
public JerseyClientBuilder register(Object component, Class<?>... contracts) {
this.config.register(component, contracts);
return this;
}
@Override
public JerseyClientBuilder register(Object component, Map<Class<?>, Integer> contracts) {
this.config.register(component, contracts);
return this;
}
@Override
public JerseyClientBuilder withConfig(Configuration config) {
this.config.loadFrom(config);
return this;
}
}