69 lines
2.4 KiB
Java
69 lines
2.4 KiB
Java
package org.gcube.accounting.aggregator.elaboration;
|
|
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
|
|
import org.gcube.accounting.aggregator.aggregation.AggregationInfo;
|
|
import org.gcube.accounting.aggregator.aggregation.AggregationType;
|
|
import org.gcube.accounting.aggregator.persistence.PostgreSQLConnector;
|
|
import org.gcube.accounting.aggregator.status.AggregationStatus;
|
|
import org.gcube.documentstore.records.DSMapper;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
public class RecoveryManager {
|
|
|
|
private static Logger logger = LoggerFactory.getLogger(RecoveryManager.class);
|
|
|
|
// protected final Date persistStartTime;
|
|
// protected final Date persistEndTime;
|
|
|
|
protected final Date aggregationStartDate;
|
|
protected final Date aggregationEndDate;
|
|
|
|
protected String recordType;
|
|
protected AggregationType aggregationType;
|
|
protected boolean forceRestart;
|
|
|
|
public RecoveryManager(Date aggregationStartDate, Date aggregationEndDate){
|
|
super();
|
|
// this.persistStartTime = persistStartTime;
|
|
// this.persistEndTime = persistEndTime;
|
|
this.aggregationStartDate = aggregationStartDate;
|
|
this.aggregationEndDate = aggregationEndDate;
|
|
this.forceRestart = false;
|
|
}
|
|
|
|
public void recovery() throws Exception {
|
|
PostgreSQLConnector postgreSQLConnector = PostgreSQLConnector.getPostgreSQLConnector();
|
|
List<AggregationStatus> aggregationStatusList = postgreSQLConnector.getUnterminated(recordType, aggregationType, aggregationStartDate, aggregationEndDate, forceRestart);
|
|
if(aggregationStatusList.size()==0){
|
|
logger.info("Nothing to recover :)");
|
|
}
|
|
|
|
for(AggregationStatus as : aggregationStatusList){
|
|
AggregationInfo aggregationInfo = as.getAggregationInfo();
|
|
AggregationStatus aggregationStatus = AggregationStatus.getAggregationStatus(aggregationInfo.getRecordType(), aggregationInfo.getAggregationType(),
|
|
aggregationInfo.getAggregationStartDate());
|
|
logger.info("Going to Recover unterminated elaboration {}", DSMapper.getObjectMapper().writeValueAsString(aggregationStatus));
|
|
|
|
Elaborator elaborator = new Elaborator(aggregationStatus);
|
|
elaborator.elaborate(true, true, forceRestart);
|
|
}
|
|
|
|
}
|
|
|
|
public void setForceRestart(boolean forceRestart) {
|
|
this.forceRestart = forceRestart;
|
|
}
|
|
|
|
public void setRecordType(String recordType) {
|
|
this.recordType = recordType;
|
|
}
|
|
|
|
public void setAggregationType(AggregationType aggregationType) {
|
|
this.aggregationType = aggregationType;
|
|
}
|
|
|
|
}
|