This commit is contained in:
Lucio Lelii 2015-10-30 15:56:38 +00:00
parent 0871f66b87
commit fc9b9bb023
1 changed files with 55 additions and 32 deletions

View File

@ -8,6 +8,7 @@ import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.ServiceLoader; import java.util.ServiceLoader;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReentrantLock;
import org.gcube.accounting.aggregation.scheduler.AggregationScheduler; import org.gcube.accounting.aggregation.scheduler.AggregationScheduler;
import org.gcube.common.scope.api.ScopeProvider; import org.gcube.common.scope.api.ScopeProvider;
@ -20,23 +21,27 @@ import org.slf4j.LoggerFactory;
* *
*/ */
public abstract class AccountingPersistenceBackendFactory { public abstract class AccountingPersistenceBackendFactory {
private static final Logger logger = LoggerFactory.getLogger(AccountingPersistenceBackendFactory.class); private static final Logger logger = LoggerFactory.getLogger(AccountingPersistenceBackendFactory.class);
public final static String HOME_SYSTEM_PROPERTY = "user.home"; public final static String HOME_SYSTEM_PROPERTY = "user.home";
private static final String ACCOUTING_FALLBACK_FILENAME = "accountingFallback.log"; private static final String ACCOUTING_FALLBACK_FILENAME = "accountingFallback.log";
private static String fallbackLocation; private static String fallbackLocation;
private static ReentrantLock lock = new ReentrantLock();
private static ReentrantLock persistenceLock = new ReentrantLock();
private static Map<String, AccountingPersistenceBackend> persistencePersistenceBackends; private static Map<String, AccountingPersistenceBackend> persistencePersistenceBackends;
static { static {
persistencePersistenceBackends = new HashMap<String, AccountingPersistenceBackend>(); persistencePersistenceBackends = new HashMap<String, AccountingPersistenceBackend>();
} }
private static File file(File file) throws IllegalArgumentException { private static File file(File file) throws IllegalArgumentException {
if(!file.isDirectory()){ if(!file.isDirectory()){
file = file.getParentFile(); file = file.getParentFile();
} }
@ -44,42 +49,57 @@ public abstract class AccountingPersistenceBackendFactory {
//create folder structure if not exist //create folder structure if not exist
if (!file.exists()) if (!file.exists())
file.mkdirs(); file.mkdirs();
return file; return file;
} }
protected synchronized static void setFallbackLocation(String path){ protected static void setFallbackLocation(String path){
if(fallbackLocation == null){ lock.lock();
if(path==null){ try{
path = System.getProperty(HOME_SYSTEM_PROPERTY); if(fallbackLocation == null){
if(path==null){
path = System.getProperty(HOME_SYSTEM_PROPERTY);
}
file(new File(path));
fallbackLocation = path;
} }
file(new File(path)); }finally{
fallbackLocation = path; lock.unlock();
} }
} }
protected static synchronized AccountingPersistenceBackend getPersistenceBackend() { protected static AccountingPersistenceBackend getPersistenceBackend() {
String scope = ScopeProvider.instance.get(); String scope = ScopeProvider.instance.get();
if(scope==null){ if(scope==null){
logger.error("No Scope available. FallbackPersistence will be used"); logger.error("No Scope available. FallbackPersistence will be used");
File fallbackFile = new File(fallbackLocation, ACCOUTING_FALLBACK_FILENAME); File fallbackFile = new File(fallbackLocation, ACCOUTING_FALLBACK_FILENAME);
return new FallbackPersistence(fallbackFile); return new FallbackPersistence(fallbackFile);
} }
persistenceLock.lock();
AccountingPersistenceBackend persistence = persistencePersistenceBackends.get(scope); AccountingPersistenceBackend persistence = persistencePersistenceBackends.get(scope);
if(persistence==null){ if(persistence==null){
FallbackPersistence fallbackPersistence;
ScopeBean bean = new ScopeBean(scope); try{
/* ScopeBean bean = new ScopeBean(scope);
if(bean.is(Type.VRE)){ /*
bean = bean.enclosingScope(); if(bean.is(Type.VRE)){
bean = bean.enclosingScope();
}
*/
String name = bean.name();
File fallbackFile = new File(fallbackLocation, String.format("%s.%s", name, ACCOUTING_FALLBACK_FILENAME));
fallbackPersistence = new FallbackPersistence(fallbackFile);
// set only to avoid deadlock
persistencePersistenceBackends.put(scope, fallbackPersistence);
}finally{
persistenceLock.unlock();
} }
*/
String name = bean.name();
File fallbackFile = new File(fallbackLocation, String.format("%s.%s", name, ACCOUTING_FALLBACK_FILENAME));
FallbackPersistence fallbackPersistence = new FallbackPersistence(fallbackFile);
try { try {
ServiceLoader<AccountingPersistenceBackend> serviceLoader = ServiceLoader.load(AccountingPersistenceBackend.class); ServiceLoader<AccountingPersistenceBackend> serviceLoader = ServiceLoader.load(AccountingPersistenceBackend.class);
for (AccountingPersistenceBackend foundPersistence : serviceLoader) { for (AccountingPersistenceBackend foundPersistence : serviceLoader) {
@ -115,8 +135,11 @@ public abstract class AccountingPersistenceBackendFactory {
persistence.setAggregationScheduler(AggregationScheduler.newInstance()); persistence.setAggregationScheduler(AggregationScheduler.newInstance());
persistence.setFallback(fallbackPersistence); persistence.setFallback(fallbackPersistence);
persistencePersistenceBackends.put(scope, persistence); persistencePersistenceBackends.put(scope, persistence);
} else{
persistenceLock.unlock();
} }
return persistence; return persistence;
} }
@ -137,6 +160,6 @@ public abstract class AccountingPersistenceBackendFactory {
} }
} }
} }
} }