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