killbill-aplcache

util: add ConfigSource in constructor of CacheModule This

2/1/2013 7:13:12 PM

Details

diff --git a/util/src/main/java/com/ning/billing/util/glue/CacheModule.java b/util/src/main/java/com/ning/billing/util/glue/CacheModule.java
index a304b3c..827367c 100644
--- a/util/src/main/java/com/ning/billing/util/glue/CacheModule.java
+++ b/util/src/main/java/com/ning/billing/util/glue/CacheModule.java
@@ -24,7 +24,6 @@ import org.skife.config.SimplePropertyConfigSource;
 
 import com.ning.billing.util.cache.AccountRecordIdCacheLoader;
 import com.ning.billing.util.cache.Cachable;
-import com.ning.billing.util.cache.Cachable.CacheType;
 import com.ning.billing.util.cache.CacheController;
 import com.ning.billing.util.cache.CacheControllerDispatcher;
 import com.ning.billing.util.cache.CacheControllerProvider;
@@ -34,7 +33,6 @@ import com.ning.billing.util.cache.TenantRecordIdCacheLoader;
 import com.ning.billing.util.config.CacheConfig;
 
 import com.google.inject.AbstractModule;
-import com.google.inject.Singleton;
 import com.google.inject.TypeLiteral;
 import com.google.inject.name.Named;
 import com.google.inject.name.Names;
@@ -43,20 +41,27 @@ import net.sf.ehcache.loader.CacheLoader;
 
 public class CacheModule extends AbstractModule {
 
-
     public static final Named RECORD_ID_CACHE_NAMED = Names.named(Cachable.RECORD_ID_CACHE_NAME);
     public static final Named ACCOUNT_RECORD_ID_CACHE_NAMED = Names.named(Cachable.ACCOUNT_RECORD_ID_CACHE_NAME);
     public static final Named TENANT_RECORD_ID_CACHE_NAMED = Names.named(Cachable.TENANT_RECORD_ID_CACHE_NAME);
 
+    private final ConfigSource configSource;
+
+    public CacheModule() {
+        this(new SimplePropertyConfigSource(System.getProperties()));
+    }
+
+    public CacheModule(final ConfigSource configSource) {
+        this.configSource = configSource;
+    }
+
     protected void installConfig() {
-        final ConfigSource configSource = new SimplePropertyConfigSource(System.getProperties());
         final CacheConfig config = new ConfigurationObjectFactory(configSource).build(CacheConfig.class);
         bind(CacheConfig.class).toInstance(config);
     }
 
     @Override
     protected void configure() {
-
         installConfig();
 
         bind(CacheManager.class).toProvider(CacheManagerProvider.class).asEagerSingleton();