/*
* 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 io.airlift.dbpool;
import com.google.common.collect.ImmutableList;
import com.google.inject.Inject;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.Provider;
import com.google.inject.Scopes;
import io.airlift.discovery.client.ServiceSelector;
import org.weakref.jmx.guice.MBeanModule;
import javax.sql.DataSource;
import java.lang.annotation.Annotation;
import java.util.Collections;
import java.util.List;
import static io.airlift.configuration.ConfigurationModule.bindConfig;
import static io.airlift.discovery.client.DiscoveryBinder.discoveryBinder;
import static io.airlift.discovery.client.ServiceTypes.serviceType;
public class MySqlDataSourceModule extends MBeanModule
{
private final Class<? extends Annotation> annotation;
private final List<Class<? extends Annotation>> aliases;
private final String type;
@SafeVarargs
public MySqlDataSourceModule(String type, Class<? extends Annotation> annotation, Class<? extends Annotation>... aliases)
{
if (annotation == null) {
throw new NullPointerException("annotation is null");
}
if (type == null) {
throw new NullPointerException("type is null");
}
this.annotation = annotation;
this.type = type;
if (aliases != null) {
this.aliases = ImmutableList.copyOf(aliases);
}
else {
this.aliases = Collections.emptyList();
}
}
@Override
public void configureMBeans()
{
// bind the configuration
bindConfig(binder()).annotatedWith(annotation).prefixedWith(type).to(MySqlDataSourceConfig.class);
// bind the service selector
discoveryBinder(binder()).bindSelector(type);
// Bind the datasource
bind(DataSource.class).annotatedWith(annotation).toProvider(new MySqlDataSourceProvider(type, annotation)).in(Scopes.SINGLETON);
export(DataSource.class).annotatedWith(annotation).withGeneratedName();
// Bind aliases
Key<DataSource> key = Key.get(DataSource.class, annotation);
for (Class<? extends Annotation> alias : aliases) {
bind(DataSource.class).annotatedWith(alias).to(key);
}
}
private static class MySqlDataSourceProvider implements Provider<MySqlDataSource>
{
private final String type;
private final Class<? extends Annotation> annotation;
private Injector injector;
private MySqlDataSourceProvider(String type, Class<? extends Annotation> annotation)
{
this.type = type;
this.annotation = annotation;
}
@Inject
public void setInjector(Injector injector)
{
this.injector = injector;
}
@Override
public MySqlDataSource get()
{
MySqlDataSourceConfig config = injector.getInstance(Key.get(MySqlDataSourceConfig.class, annotation));
ServiceSelector serviceSelector = injector.getInstance(Key.get(ServiceSelector.class, serviceType(type)));
return new MySqlDataSource(serviceSelector, config);
}
}
}