git-svn-id: https://svn.d4science.research-infrastructures.eu/gcube/trunk/accounting/accounting-lib@120135 82a268e6-3cf1-43bd-a215-b396298e98cf
This commit is contained in:
parent
0871f66b87
commit
fc9b9bb023
|
@ -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 {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue