All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.proofpoint.discovery.store.ReplicatedStoreModule 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.proofpoint.discovery.store;
import com.google.common.base.Supplier;
import com.google.inject.Binder;
import com.google.inject.Injector;
import com.google.inject.Key;
import com.google.inject.Module;
import com.google.inject.Provider;
import com.google.inject.Scopes;
import com.google.inject.TypeLiteral;
import com.proofpoint.discovery.client.ServiceSelector;
import com.proofpoint.http.client.HttpClient;
import com.proofpoint.http.client.balancing.HttpServiceBalancerStats;
import com.proofpoint.node.NodeInfo;
import com.proofpoint.reporting.ReportCollectionFactory;
import com.proofpoint.reporting.ReportExporter;
import org.joda.time.DateTime;
import org.weakref.jmx.ObjectNameBuilder;
import javax.annotation.PreDestroy;
import javax.annotation.concurrent.GuardedBy;
import javax.annotation.concurrent.ThreadSafe;
import javax.inject.Inject;
import java.lang.annotation.Annotation;
import static com.google.inject.multibindings.MapBinder.newMapBinder;
import static com.google.inject.name.Names.named;
import static com.proofpoint.configuration.ConfigurationModule.bindConfig;
import static com.proofpoint.http.client.HttpClientBinder.httpClientBinder;
import static org.weakref.jmx.ObjectNames.generatedNameOf;
import static org.weakref.jmx.guice.ExportBinder.newExporter;
/**
* Expects a LocalStore to be bound elsewhere.
* Provides a DistributedStore with the specified annotation.
*/
public class ReplicatedStoreModule
implements Module
{
private final String name;
private final Class annotation;
private final Class localStoreClass;
public ReplicatedStoreModule(String name, Class annotation, Class localStoreClass)
{
this.name = name;
this.annotation = annotation;
this.localStoreClass = localStoreClass;
}
@Override
public void configure(Binder binder)
{
binder.requireExplicitBindings();
binder.disableCircularProxies();
// global
binder.bind(StoreResource.class).in(Scopes.SINGLETON);
binder.bind(new TypeLiteral>() {}).to(RealTimeSupplier.class).in(Scopes.SINGLETON);
binder.bind(SmileMapper.class).in(Scopes.SINGLETON);
binder.bind(ConflictResolver.class).in(Scopes.SINGLETON);
// per store
Key httpClientKey = Key.get(HttpClient.class, annotation);
Key localStoreKey = Key.get(LocalStore.class, annotation);
Key storeConfigKey = Key.get(StoreConfig.class, annotation);
Key remoteStoreKey = Key.get(RemoteStore.class, annotation);
bindConfig(binder).annotatedWith(annotation).prefixedWith(name).to(StoreConfig.class);
httpClientBinder(binder).bindHttpClient(name, annotation);
binder.bind(DistributedStore.class).annotatedWith(annotation).toProvider(new DistributedStoreProvider(name, localStoreKey, storeConfigKey, remoteStoreKey)).in(Scopes.SINGLETON);
binder.bind(Replicator.class).annotatedWith(annotation).toProvider(new ReplicatorProvider(name, localStoreKey, httpClientKey, storeConfigKey)).in(Scopes.SINGLETON);
binder.bind(HttpRemoteStore.class).annotatedWith(annotation).toProvider(new RemoteHttpStoreProvider(name, httpClientKey, storeConfigKey)).in(Scopes.SINGLETON);
binder.bind(LocalStore.class).annotatedWith(annotation).to(localStoreClass).in(Scopes.SINGLETON);
binder.bind(RemoteStore.class).annotatedWith(annotation).to(Key.get(HttpRemoteStore.class, annotation));
newExporter(binder).export(DistributedStore.class).annotatedWith(annotation).as(generatedNameOf(DistributedStore.class, named(name)));
newExporter(binder).export(HttpRemoteStore.class).annotatedWith(annotation).as(generatedNameOf(HttpRemoteStore.class, named(name)));
newExporter(binder).export(Replicator.class).annotatedWith(annotation).as(generatedNameOf(Replicator.class, named(name)));
newMapBinder(binder, String.class, LocalStore.class)
.addBinding(name)
.to(localStoreKey);
newMapBinder(binder, String.class, StoreConfig.class)
.addBinding(name)
.to(storeConfigKey);
}
@ThreadSafe
private static class ReplicatorProvider
implements Provider
{
private final String name;
private final Key localStoreKey;
private final Key httpClientKey;
private final Key storeConfigKey;
@GuardedBy("this")
private Injector injector;
@GuardedBy("this")
private NodeInfo nodeInfo;
@GuardedBy("this")
private ServiceSelector serviceSelector;
@GuardedBy("this")
private Replicator replicator;
private ReplicatorProvider(String name, Key localStoreKey, Key httpClientKey, Key storeConfigKey)
{
this.name = name;
this.localStoreKey = localStoreKey;
this.httpClientKey = httpClientKey;
this.storeConfigKey = storeConfigKey;
}
@Override
public synchronized Replicator get()
{
if (replicator == null) {
LocalStore localStore = injector.getInstance(localStoreKey);
HttpClient httpClient = injector.getInstance(httpClientKey);
StoreConfig storeConfig = injector.getInstance(storeConfigKey);
ReportCollectionFactory reportCollectionFactory = injector.getInstance(ReportCollectionFactory.class);
String objectName = new ObjectNameBuilder(HttpServiceBalancerStats.class.getPackage().getName())
.withProperty("type", "replicator-" + name)
.build();
HttpServiceBalancerStats httpServiceBalancerStats = reportCollectionFactory.createReportCollection(HttpServiceBalancerStats.class, objectName);
replicator = new Replicator(name, nodeInfo, serviceSelector, httpClient, httpServiceBalancerStats, localStore, storeConfig);
replicator.start();
}
return replicator;
}
@PreDestroy
public synchronized void shutdown()
{
if (replicator != null) {
replicator.shutdown();
}
}
@Inject
public synchronized void setInjector(Injector injector)
{
this.injector = injector;
}
@Inject
public synchronized void setNodeInfo(NodeInfo nodeInfo)
{
this.nodeInfo = nodeInfo;
}
@Inject
public synchronized void setServiceSelector(ServiceSelector serviceSelector)
{
this.serviceSelector = serviceSelector;
}
}
@ThreadSafe
private static class RemoteHttpStoreProvider
implements Provider
{
@GuardedBy("this")
private HttpRemoteStore remoteStore;
@GuardedBy("this")
private Injector injector;
@GuardedBy("this")
private NodeInfo nodeInfo;
@GuardedBy("this")
private ServiceSelector serviceSelector;
@GuardedBy("this")
private ReportExporter reportExporter;
private final String name;
private final Key httpClientKey;
private final Key storeConfigKey;
@Inject
private RemoteHttpStoreProvider(String name, Key httpClientKey, Key storeConfigKey)
{
this.name = name;
this.httpClientKey = httpClientKey;
this.storeConfigKey = storeConfigKey;
}
public synchronized HttpRemoteStore get()
{
if (remoteStore == null) {
HttpClient httpClient = injector.getInstance(httpClientKey);
StoreConfig storeConfig = injector.getInstance(storeConfigKey);
remoteStore = new HttpRemoteStore(name, nodeInfo, serviceSelector, storeConfig, httpClient, reportExporter);
remoteStore.start();
}
return remoteStore;
}
@PreDestroy
public synchronized void shutdown()
{
if (remoteStore != null) {
remoteStore.shutdown();
}
}
@Inject
public synchronized void setInjector(Injector injector)
{
this.injector = injector;
}
@Inject
public synchronized void setNodeInfo(NodeInfo nodeInfo)
{
this.nodeInfo = nodeInfo;
}
@Inject
public synchronized void setServiceSelector(ServiceSelector serviceSelector)
{
this.serviceSelector = serviceSelector;
}
@Inject
private synchronized void setReportExporter(ReportExporter reportExporter)
{
this.reportExporter = reportExporter;
}
}
private static class DistributedStoreProvider
implements Provider
{
private final String name;
private final Key localStoreKey;
private final Key storeConfigKey;
private final Key remoteStoreKey;
private Injector injector;
private Supplier timeSupplier;
private DistributedStore store;
public DistributedStoreProvider(String name,
Key localStoreKey,
Key storeConfigKey,
Key remoteStoreKey)
{
this.name = name;
this.localStoreKey = localStoreKey;
this.storeConfigKey = storeConfigKey;
this.remoteStoreKey = remoteStoreKey;
}
@Override
public synchronized DistributedStore get()
{
if (store == null) {
LocalStore localStore = injector.getInstance(localStoreKey);
StoreConfig storeConfig = injector.getInstance(storeConfigKey);
RemoteStore remoteStore = injector.getInstance(remoteStoreKey);
store = new DistributedStore(name, localStore, remoteStore, storeConfig, timeSupplier);
store.start();
}
return store;
}
@PreDestroy
public synchronized void shutdown()
{
if (store != null) {
store.shutdown();
}
}
@Inject
public synchronized void setInjector(Injector injector)
{
this.injector = injector;
}
@Inject
public synchronized void setTimeSupplier(Supplier timeSupplier)
{
this.timeSupplier = timeSupplier;
}
}
}
