com.bazaarvoice.ostrich.pool.ServiceCacheBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ostrich-core Show documentation

Core classes that form Ostrich

There is a newer version: 2.1.0

package com.bazaarvoice.ostrich.pool;

import com.bazaarvoice.ostrich.MultiThreadedServiceFactory;
import com.bazaarvoice.ostrich.ServiceFactory;
import com.codahale.metrics.MetricRegistry;
import com.google.common.util.concurrent.ThreadFactoryBuilder;

import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;

public class ServiceCacheBuilder {

    private ServiceCachingPolicy _cachingPolicy;
    private ServiceFactory _serviceFactory;
    private MetricRegistry _metricRegistry;

    public ServiceCacheBuilder withCachingPolicy(ServiceCachingPolicy cachingPolicy) {
        _cachingPolicy = cachingPolicy;
        return this;
    }

    public ServiceCacheBuilder withServiceFactory(ServiceFactory serviceFactory) {
        _serviceFactory = serviceFactory;
        return this;
    }

    public ServiceCacheBuilder withMetricRegistry(MetricRegistry metricRegistry) {
        _metricRegistry = metricRegistry;
        return this;
    }

    public ServiceCache build() {
        checkNotNull(_cachingPolicy, "cachingPolicy");
        if (_cachingPolicy.useMultiThreadedClientPolicy()) {
            checkNotNull(_serviceFactory, "serviceFactory");
            checkArgument((_serviceFactory instanceof MultiThreadedServiceFactory), "Please implement MultiThreadedServiceFactory to construct MultiThreadedClientServiceCache");
            return new MultiThreadedClientServiceCache<>((MultiThreadedServiceFactory) _serviceFactory, _metricRegistry);
        }
        else {
            checkNotNull(_serviceFactory, "serviceFactory");
            checkNotNull(_metricRegistry, "metricRegistry");
            return new SingleThreadedClientServiceCache<>(_cachingPolicy, _serviceFactory, _metricRegistry);
        }
    }

    /**
     * This ensures the {@link java.util.concurrent.ScheduledExecutorService} in not loaded onto jvm
     * until the class is loaded by explicitly calling the constructor.
     */
    public static ScheduledExecutorService buildDefaultExecutor() {
        return Executors.newScheduledThreadPool(1,
                new ThreadFactoryBuilder().setNameFormat("ServiceCache-CleanupThread-%d").setDaemon(true).build());
    }
}