com.facebook.presto.jdbc.internal.airlift.http.client.HttpClientBinder Maven / Gradle / Ivy
/*
* Copyright 2010 Proofpoint, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.facebook.presto.jdbc.internal.airlift.http.client;
import com.facebook.presto.jdbc.internal.guava.annotations.Beta;
import com.google.inject.Binder;
import com.google.inject.binder.LinkedBindingBuilder;
import com.google.inject.multibindings.Multibinder;
import java.lang.annotation.Annotation;
import java.util.Collection;
import static com.facebook.presto.jdbc.internal.guava.base.Preconditions.checkNotNull;
import static com.google.inject.multibindings.Multibinder.newSetBinder;
@Beta
public class HttpClientBinder
{
private final Binder binder;
private HttpClientBinder(Binder binder)
{
this.binder = checkNotNull(binder, "binder is null");
}
public static HttpClientBinder httpClientBinder(Binder binder)
{
return new HttpClientBinder(binder);
}
public HttpClientBindingBuilder bindHttpClient(String name, Class annotation)
{
checkNotNull(name, "name is null");
checkNotNull(annotation, "annotation is null");
return createBindingBuilder(new HttpClientModule(name, annotation));
}
private HttpClientBindingBuilder createBindingBuilder(HttpClientModule module)
{
binder.install(module);
return new HttpClientBindingBuilder(module,
newSetBinder(binder, HttpRequestFilter.class, module.getFilterQualifier()));
}
public static class HttpClientBindingBuilder
{
private final HttpClientModule module;
private final Multibinder multibinder;
public HttpClientBindingBuilder(HttpClientModule module, Multibinder multibinder)
{
this.module = module;
this.multibinder = multibinder;
}
public HttpClientBindingBuilder withAlias(Class alias)
{
module.addAlias(alias);
return this;
}
public HttpClientBindingBuilder withAliases(Collection> aliases)
{
for (Class annotation : aliases) {
module.addAlias(annotation);
}
return this;
}
public LinkedBindingBuilder addFilterBinding()
{
return multibinder.addBinding();
}
public HttpClientBindingBuilder withFilter(Class filterClass)
{
multibinder.addBinding().to(filterClass);
return this;
}
public HttpClientBindingBuilder withTracing()
{
return withFilter(TraceTokenRequestFilter.class);
}
public HttpClientBindingBuilder withPrivateIoThreadPool()
{
module.withPrivateIoThreadPool();
return this;
}
}
}