2016-09-08 10:40:04 +02:00
|
|
|
package org.gcube.accounting.analytics.persistence.couchbase;
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import java.util.Calendar;
|
|
|
|
import java.util.Iterator;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
import java.util.SortedMap;
|
|
|
|
import java.util.SortedSet;
|
|
|
|
|
|
|
|
import org.gcube.accounting.analytics.Filter;
|
|
|
|
import org.gcube.accounting.analytics.Info;
|
|
|
|
import org.gcube.accounting.analytics.NumberedFilter;
|
|
|
|
import org.gcube.accounting.analytics.TemporalConstraint;
|
2017-01-19 11:51:31 +01:00
|
|
|
import org.gcube.accounting.analytics.TemporalConstraint.AggregationMode;
|
2016-11-15 11:41:07 +01:00
|
|
|
import org.gcube.accounting.analytics.UsageServiceValue;
|
|
|
|
import org.gcube.accounting.analytics.UsageStorageValue;
|
|
|
|
import org.gcube.accounting.analytics.UsageValue;
|
2016-09-08 10:40:04 +02:00
|
|
|
import org.gcube.accounting.analytics.persistence.AccountingPersistenceBackendQueryConfiguration;
|
2016-09-19 11:01:14 +02:00
|
|
|
import org.gcube.accounting.analytics.persistence.AccountingPersistenceQuery;
|
|
|
|
import org.gcube.accounting.datamodel.aggregation.AggregatedJobUsageRecord;
|
2016-09-08 10:40:04 +02:00
|
|
|
import org.gcube.accounting.datamodel.aggregation.AggregatedServiceUsageRecord;
|
2017-01-19 11:51:31 +01:00
|
|
|
import org.gcube.accounting.datamodel.aggregation.AggregatedStorageStatusRecord;
|
2016-11-15 11:41:07 +01:00
|
|
|
import org.gcube.accounting.datamodel.aggregation.AggregatedStorageUsageRecord;
|
2017-09-21 14:51:31 +02:00
|
|
|
import org.gcube.accounting.datamodel.usagerecords.ServiceUsageRecord;
|
2016-09-08 10:40:04 +02:00
|
|
|
import org.gcube.common.scope.api.ScopeProvider;
|
2021-03-22 12:00:41 +01:00
|
|
|
import org.gcube.documentstore.records.Record;
|
2016-09-08 10:40:04 +02:00
|
|
|
import org.json.JSONException;
|
|
|
|
import org.json.JSONObject;
|
|
|
|
import org.junit.Before;
|
|
|
|
import org.junit.Test;
|
|
|
|
import org.slf4j.Logger;
|
|
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @author Alessandro Pieve (ISTI - CNR)
|
|
|
|
*
|
|
|
|
*/
|
2017-09-21 14:51:31 +02:00
|
|
|
public class AccountingPersistenceQueryCouchBaseTest extends ScopedTest {
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
private static Logger logger = LoggerFactory.getLogger(AccountingPersistenceQueryCouchBaseTest.class);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
protected AccountingPersistenceQueryCouchBase accountingPersistenceQueryCouchBase;
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
public class ExtendedInfo extends Info {
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
protected String key;
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
/**
|
|
|
|
* @return the key
|
|
|
|
*/
|
|
|
|
public String getKey() {
|
|
|
|
return key;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
/**
|
|
|
|
* @param key the key to set
|
|
|
|
*/
|
|
|
|
public void setKey(String key) {
|
|
|
|
this.key = key;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
public ExtendedInfo(String key, Calendar calendar, JSONObject value) {
|
2016-09-08 10:40:04 +02:00
|
|
|
super(calendar, value);
|
|
|
|
this.key = key;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
public String toString() {
|
|
|
|
String info = String.format("Key : %s, %s ", key, super.toString());
|
2016-09-08 10:40:04 +02:00
|
|
|
return info;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
@Before
|
2018-01-02 17:22:13 +01:00
|
|
|
public void before() throws Exception {
|
|
|
|
AccountingPersistenceBackendQueryConfiguration configuration = new AccountingPersistenceBackendQueryConfiguration(
|
|
|
|
AccountingPersistenceQueryCouchBase.class);
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
accountingPersistenceQueryCouchBase = new AccountingPersistenceQueryCouchBase();
|
|
|
|
accountingPersistenceQueryCouchBase.prepareConnection(configuration);
|
|
|
|
}
|
2017-09-21 14:51:31 +02:00
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
public void printMap(Map<Calendar,Info> map) {
|
|
|
|
for(Info info : map.values()) {
|
2016-09-08 10:40:04 +02:00
|
|
|
logger.debug("{}", info);
|
|
|
|
}
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
@Test
|
2017-09-21 14:51:31 +02:00
|
|
|
public void getUsersInVREs() throws Exception {
|
|
|
|
List<String> contexts = new ArrayList<>();
|
|
|
|
contexts.add("/d4science.research-infrastructures.eu/gCubeApps/ICCAT_BFT-E");
|
|
|
|
contexts.add("/d4science.research-infrastructures.eu/gCubeApps/StockAssessment");
|
|
|
|
contexts.add("/d4science.research-infrastructures.eu/gCubeApps/RStudioLab");
|
|
|
|
contexts.add("/d4science.research-infrastructures.eu/gCubeApps/FAO_TunaAtlas");
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-09-21 14:51:31 +02:00
|
|
|
startTime.set(2015, Calendar.SEPTEMBER, 15);
|
|
|
|
|
|
|
|
logger.debug("StartTime {}", startTime.getTimeInMillis());
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2017-09-21 14:51:31 +02:00
|
|
|
endTime.set(2017, Calendar.JULY, 15);
|
|
|
|
|
|
|
|
logger.debug("EndTime {}", endTime.getTimeInMillis());
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.MONTHLY);
|
|
|
|
|
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
2017-09-21 14:51:31 +02:00
|
|
|
|
2016-09-19 11:01:14 +02:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2017-09-21 14:51:31 +02:00
|
|
|
//Filter filter = new Filter(ServiceUsageRecord.SERVICE_NAME, "DataMiner");
|
|
|
|
Filter filter = new Filter(ServiceUsageRecord.SERVICE_NAME, "RConnector");
|
|
|
|
filters.add(filter);
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
for(String context : contexts) {
|
2017-09-21 14:51:31 +02:00
|
|
|
ScopeProvider.instance.set(context);
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
SortedSet<NumberedFilter> top = accountingPersistenceQueryCouchBase.getNextPossibleValuesWithMap(clz,
|
|
|
|
temporalConstraint, filters, ServiceUsageRecord.CONSUMER_ID, null);
|
2017-09-21 14:51:31 +02:00
|
|
|
|
|
|
|
logger.info("Context : {} - Users [{}] : {}", context, top.size(), top);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testTop() throws Exception {
|
|
|
|
Calendar startTime = Calendar.getInstance();
|
|
|
|
startTime.set(2015, Calendar.SEPTEMBER, 15);
|
|
|
|
|
|
|
|
logger.debug("StartTime {}", startTime.getTimeInMillis());
|
|
|
|
|
|
|
|
Calendar endTime = Calendar.getInstance();
|
|
|
|
endTime.set(2017, Calendar.JULY, 15);
|
|
|
|
|
|
|
|
logger.debug("EndTime {}", endTime.getTimeInMillis());
|
|
|
|
|
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
|
|
|
Filter filter = new Filter(ServiceUsageRecord.SERVICE_NAME, "DataMiner");
|
|
|
|
//Filter filter = new Filter(ServiceUsageRecord.SERVICE_NAME, "RConnector");
|
|
|
|
filters.add(filter);
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.MONTHLY);
|
|
|
|
|
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
|
|
|
|
|
|
|
SortedMap<NumberedFilter,SortedMap<Calendar,Info>> set = accountingPersistenceQueryCouchBase.getTopValues(clz,
|
|
|
|
temporalConstraint, filters, ServiceUsageRecord.CONSUMER_ID, null);
|
2017-09-21 14:51:31 +02:00
|
|
|
|
|
|
|
logger.debug("Result {}", set);
|
2016-09-19 11:01:14 +02:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-10-13 09:44:12 +02:00
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void testTimeSeries() throws Exception {
|
|
|
|
|
2018-01-02 17:26:12 +01:00
|
|
|
// AccountingPersistenceQueryFactory.getForcedQueryScope().set("/gcube/devNext/NextNext");
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-10-13 09:44:12 +02:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
startTime.set(2017, Calendar.NOVEMBER, 29);
|
2016-10-13 09:44:12 +02:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2017, Calendar.DECEMBER, 29);
|
2016-10-13 09:44:12 +02:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
Class<AggregatedJobUsageRecord> clz = AggregatedJobUsageRecord.class;
|
|
|
|
SortedMap<Calendar,Info> set = accountingPersistenceQueryCouchBase.getTimeSeries(clz, temporalConstraint,
|
|
|
|
filters);
|
2016-10-13 09:44:12 +02:00
|
|
|
logger.debug("Result final{}", set);
|
2017-02-03 12:18:50 +01:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-02-03 12:18:50 +01:00
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void testTimeSeriesNoContext() throws Exception {
|
|
|
|
|
2017-02-03 12:18:50 +01:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-05-05 16:57:26 +02:00
|
|
|
startTime.set(2017, Calendar.APRIL, 20);
|
2017-02-03 12:18:50 +01:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2017, Calendar.APRIL, 29, 23, 59);
|
2017-02-03 12:18:50 +01:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-09-21 14:51:31 +02:00
|
|
|
/*
|
2017-02-03 12:18:50 +01:00
|
|
|
Filter filter =
|
|
|
|
new Filter(AggregatedServiceUsageRecord.CALLED_METHOD, "WebProcessingService");
|
2017-09-21 14:51:31 +02:00
|
|
|
filters.add(filter);
|
|
|
|
*/
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
|
|
|
SortedMap<Calendar,Info> set = accountingPersistenceQueryCouchBase.getNoContextTimeSeries(clz,
|
|
|
|
temporalConstraint, filters);
|
2017-02-03 12:18:50 +01:00
|
|
|
logger.debug("Result final{}", set);
|
2016-10-13 09:44:12 +02:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-10-13 09:44:12 +02:00
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void getUsageValue() throws Exception {
|
2016-10-13 09:44:12 +02:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
|
|
|
startTime.set(2015, Calendar.MAY, 1);
|
|
|
|
Calendar endTime = Calendar.getInstance();
|
|
|
|
ScopeProvider.instance.set("/gcube");
|
2018-01-02 17:22:13 +01:00
|
|
|
Filter filter = new Filter(AggregatedServiceUsageRecord.CONSUMER_ID, "alessandro.pieve");
|
|
|
|
|
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
|
|
|
|
JSONObject filterValue = accountingPersistenceQueryCouchBase.getUsageValue(AggregatedServiceUsageRecord.class,
|
|
|
|
temporalConstraint, filter);
|
|
|
|
|
|
|
|
logger.info("result:" + filterValue.toString());
|
|
|
|
|
2016-10-13 09:44:12 +02:00
|
|
|
}
|
|
|
|
|
2016-11-15 11:41:07 +01:00
|
|
|
/**
|
|
|
|
* utilizzato per effettuare una singola chiamata ad accounting analytics in modo che ritorni una lista di quote riempite
|
|
|
|
*
|
|
|
|
* OUTPUT:
|
|
|
|
* result:[
|
|
|
|
*
|
|
|
|
* [
|
|
|
|
* UsageServiceValue [
|
|
|
|
* clz=class org.gcube.accounting.datamodel.aggregation.AggregatedServiceUsageRecord,
|
|
|
|
* temporalConstraint=StartTime : 2015-05-01 10:25:49:238 UTC (1430475949238 millis), EndTime : 2016-11-11 10:25:49:238 UTC (1478859949238 millis),
|
|
|
|
* Aggregated DAILY,
|
|
|
|
* filtersValue=[
|
|
|
|
* FiltersValue [
|
|
|
|
* filters=[{ "serviceClass" : "DataAccess" }, { "serviceName" : "CkanConnector" }], d=1.0, orderingProperty=operationCount],
|
|
|
|
* FiltersValue [
|
|
|
|
* filters=[{ "serviceClass" : "VREManagement" }], d=1.0, orderingProperty=operationCount]
|
|
|
|
* ],
|
|
|
|
* identifier=lucio.lelii,
|
|
|
|
* d=2.0,
|
|
|
|
* orderingProperty=operationCount
|
|
|
|
* ],
|
|
|
|
*
|
|
|
|
* UsageStorageValue [
|
|
|
|
* clz=class org.gcube.accounting.datamodel.aggregation.AggregatedStorageUsageRecord,
|
|
|
|
* temporalConstraint=StartTime : 2015-05-01 10:25:49:238 UTC (1430475949238 millis), EndTime : 2016-11-11 10:25:49:238 UTC (1478859949238 millis),
|
|
|
|
* Aggregated DAILY,
|
|
|
|
* identifier=alessandro.pieve,
|
|
|
|
* d=714216.0,
|
|
|
|
* orderingProperty=dataVolume
|
|
|
|
* ]
|
|
|
|
* ]
|
|
|
|
*/
|
2016-10-13 18:18:38 +02:00
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void getUsageValueQuotaTotal() throws Exception {
|
2017-09-21 14:51:31 +02:00
|
|
|
|
|
|
|
String context = ScopedTest.getCurrentScope(DEFAULT_TEST_SCOPE);
|
|
|
|
|
2016-10-13 18:18:38 +02:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-05-05 16:57:26 +02:00
|
|
|
startTime.set(2017, Calendar.MAY, 1);
|
2016-10-13 18:18:38 +02:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
2016-10-13 18:18:38 +02:00
|
|
|
|
2016-11-15 11:41:07 +01:00
|
|
|
/*ask quota for user alessandro pieve
|
|
|
|
*
|
|
|
|
* SERVICE
|
|
|
|
* */
|
2018-01-02 17:22:13 +01:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2016-10-13 18:18:38 +02:00
|
|
|
filters.add(new Filter("serviceClass", "DataAccess"));
|
|
|
|
filters.add(new Filter("serviceName", "CkanConnector"));
|
2018-01-02 17:22:13 +01:00
|
|
|
UsageValue totalfilter = new UsageServiceValue(context, "lucio.lelii", AggregatedServiceUsageRecord.class,
|
|
|
|
temporalConstraint, filters);
|
2017-01-19 11:51:31 +01:00
|
|
|
|
|
|
|
/****
|
|
|
|
*Example call storage status for each consumer id (quota total used )
|
|
|
|
*/
|
|
|
|
//Call quota total for consumerID
|
2018-01-02 17:22:13 +01:00
|
|
|
UsageValue totalfilterStorageStatus = new UsageStorageValue(context, "name.surname",
|
|
|
|
AggregatedStorageStatusRecord.class);
|
|
|
|
UsageValue totalfilterStorageStatus_2 = new UsageStorageValue(context, "lucio.lelii",
|
|
|
|
AggregatedStorageStatusRecord.class);
|
|
|
|
UsageValue totalfilterStorageStatus_3 = new UsageStorageValue(context, "alessandro.pieve",
|
|
|
|
AggregatedStorageStatusRecord.class);
|
|
|
|
UsageValue totalfilterStorageStatus_4 = new UsageStorageValue(context, "giancarlo.panichi",
|
|
|
|
AggregatedStorageStatusRecord.class);
|
2016-11-15 11:41:07 +01:00
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
/****
|
|
|
|
*Example call storage status for each consumer id( quota into period)
|
|
|
|
*/
|
|
|
|
//get temporalConstraintStorage
|
|
|
|
Calendar startTimeStorage = Calendar.getInstance();
|
|
|
|
startTimeStorage.set(2015, Calendar.MAY, 1);
|
|
|
|
Calendar endTimeStorage = Calendar.getInstance();
|
2017-05-05 16:57:26 +02:00
|
|
|
endTimeStorage.set(2017, Calendar.APRIL, 13);
|
2017-02-21 11:27:53 +01:00
|
|
|
/*TemporalConstraint temporalConstraintStorage =new TemporalConstraint(startTimeStorage.getTimeInMillis(),
|
|
|
|
endTimeStorage.getTimeInMillis(), AggregationMode.DAILY);*/
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraintStorage = null;
|
|
|
|
UsageValue totalfilterStorageStatusPeriod = new UsageStorageValue(context, "alessandro.pieve",
|
|
|
|
AggregatedStorageStatusRecord.class, temporalConstraintStorage);
|
2017-01-19 11:51:31 +01:00
|
|
|
|
|
|
|
//richiedo la lista di dati usati totali
|
2018-01-02 17:22:13 +01:00
|
|
|
List<UsageValue> listTotalFilter = new ArrayList<UsageValue>();
|
2017-05-05 16:57:26 +02:00
|
|
|
listTotalFilter.add(totalfilter);
|
2017-01-19 11:51:31 +01:00
|
|
|
listTotalFilter.add(totalfilterStorageStatus);
|
|
|
|
listTotalFilter.add(totalfilterStorageStatus_2);
|
|
|
|
listTotalFilter.add(totalfilterStorageStatus_3);
|
2017-05-05 16:57:26 +02:00
|
|
|
listTotalFilter.add(totalfilterStorageStatus_4);
|
2017-01-19 11:51:31 +01:00
|
|
|
listTotalFilter.add(totalfilterStorageStatusPeriod);
|
2016-11-15 11:41:07 +01:00
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
logger.info("filterPackageQuota:" + listTotalFilter);
|
|
|
|
List<UsageValue> filterValue = accountingPersistenceQueryCouchBase.getUsageValueQuotaTotal(listTotalFilter);
|
|
|
|
logger.info("result:" + filterValue.toString());
|
2016-10-13 18:18:38 +02:00
|
|
|
}
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
@Test
|
|
|
|
public void getQuerableKeyJob() throws Exception {
|
2016-09-29 10:38:53 +02:00
|
|
|
SortedSet<String> keys;
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-29 10:38:53 +02:00
|
|
|
keys = AccountingPersistenceQuery.getQuerableKeys(AggregatedJobUsageRecord.class);
|
2018-01-02 17:22:13 +01:00
|
|
|
for(String key : keys) {
|
|
|
|
if(key != null && !key.isEmpty()) {
|
|
|
|
logger.debug("key:" + key);
|
2016-09-29 10:38:53 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
logger.debug("List FilterKeys:" + keys.toString());
|
2016-09-19 11:01:14 +02:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-19 11:01:14 +02:00
|
|
|
@Test
|
|
|
|
public void testTopService() throws Exception {
|
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-01-19 11:51:31 +01:00
|
|
|
startTime.set(2016, Calendar.AUGUST, 27);
|
2016-09-19 11:01:14 +02:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2016, Calendar.SEPTEMBER, 28, 23, 59);
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2017-03-01 14:36:22 +01:00
|
|
|
//filters.add(new Filter(AggregatedServiceUsageRecord.SERVICE_NAME, "IS-Registry"));
|
2017-02-15 11:05:26 +01:00
|
|
|
//filters.add(new Filter(AggregatedServiceUsageRecord.SERVICE_CLASS, "Common"));
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
|
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
|
|
|
|
|
|
|
SortedMap<NumberedFilter,SortedMap<Calendar,Info>> set = accountingPersistenceQueryCouchBase.getTopValues(clz,
|
|
|
|
temporalConstraint, filters, AggregatedServiceUsageRecord.OPERATION_RESULT, null);
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
logger.debug("Result final{}", set);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
}
|
2016-11-08 10:00:47 +01:00
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
@Test
|
|
|
|
public void testTopStorage() throws Exception {
|
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-03-02 12:32:32 +01:00
|
|
|
startTime.set(2017, Calendar.FEBRUARY, 1);
|
2017-01-19 11:51:31 +01:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2017, Calendar.FEBRUARY, 28, 23, 59);
|
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2017-03-02 12:32:32 +01:00
|
|
|
filters.add(new Filter(AggregatedServiceUsageRecord.CONSUMER_ID, "scarponi"));
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
|
|
|
|
Class<AggregatedStorageUsageRecord> clz = AggregatedStorageUsageRecord.class;
|
|
|
|
|
|
|
|
SortedMap<NumberedFilter,SortedMap<Calendar,Info>> set = accountingPersistenceQueryCouchBase.getTopValues(clz,
|
|
|
|
temporalConstraint, filters, AggregatedStorageUsageRecord.OPERATION_TYPE, null);
|
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
logger.debug("Result final{}", set);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
}
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
@Test
|
|
|
|
public void getQuerableKeyService() throws Exception {
|
2016-09-30 11:25:51 +02:00
|
|
|
SortedSet<String> keys;
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-30 11:25:51 +02:00
|
|
|
keys = AccountingPersistenceQuery.getQuerableKeys(AggregatedServiceUsageRecord.class);
|
2018-01-02 17:22:13 +01:00
|
|
|
for(String key : keys) {
|
|
|
|
if(key != null && !key.isEmpty()) {
|
|
|
|
logger.debug("key:" + key);
|
2016-09-30 11:25:51 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
logger.debug("List FilterKeys:" + keys.toString());
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
public static SortedMap<Calendar,Info> padMap(SortedMap<Calendar,Info> unpaddedData,
|
2016-09-08 10:40:04 +02:00
|
|
|
TemporalConstraint temporalConstraint) throws Exception {
|
|
|
|
JSONObject jsonObject = getPaddingJSONObject(unpaddedData);
|
|
|
|
SortedSet<Calendar> sequence = temporalConstraint.getCalendarSequence();
|
2018-01-02 17:22:13 +01:00
|
|
|
for(Calendar progressTime : sequence) {
|
2016-09-08 10:40:04 +02:00
|
|
|
Info info = unpaddedData.get(progressTime);
|
2018-01-02 17:22:13 +01:00
|
|
|
if(info == null) {
|
2016-09-08 10:40:04 +02:00
|
|
|
info = new Info(progressTime, jsonObject);
|
|
|
|
unpaddedData.put(progressTime, info);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return unpaddedData;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
public static JSONObject getPaddingJSONObject(Map<Calendar,Info> unpaddedResults) throws JSONException {
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
JSONObject jsonObject = new JSONObject();
|
|
|
|
//verify data consistency
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
if(unpaddedResults.size() != 0) {
|
2016-09-08 10:40:04 +02:00
|
|
|
Info auxInfo = new ArrayList<Info>(unpaddedResults.values()).get(0);
|
|
|
|
JSONObject auxJsonObject = auxInfo.getValue();
|
|
|
|
@SuppressWarnings("unchecked")
|
|
|
|
Iterator<String> keys = auxJsonObject.keys();
|
2018-01-02 17:22:13 +01:00
|
|
|
while(keys.hasNext()) {
|
2016-09-08 10:40:04 +02:00
|
|
|
String key = keys.next();
|
|
|
|
jsonObject.put(key, 0);
|
|
|
|
}
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
return jsonObject;
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
@Test
|
|
|
|
public void testFull() throws Exception {
|
|
|
|
logger.debug("test full ");
|
2018-01-02 17:22:13 +01:00
|
|
|
try {
|
2016-09-08 10:40:04 +02:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
|
|
|
//startTime.set(2016, Calendar.AUGUST, 20, 00, 00);
|
2018-01-02 17:22:13 +01:00
|
|
|
startTime.set(2016, Calendar.AUGUST, 29, 00, 00);
|
2016-09-08 10:40:04 +02:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
|
|
|
//endTime.set(2016, Calendar.AUGUST, 29, 23, 59);
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2016, Calendar.AUGUST, 31, 23, 59);
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
|
|
|
|
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
|
|
|
|
|
|
|
SortedMap<NumberedFilter,SortedMap<Calendar,Info>> set = accountingPersistenceQueryCouchBase
|
|
|
|
.getTopValues(clz, temporalConstraint, filters, AggregatedServiceUsageRecord.CALLED_METHOD, null);
|
2016-09-08 10:40:04 +02:00
|
|
|
/**pad*/
|
2018-01-02 17:22:13 +01:00
|
|
|
int limit = 0;
|
|
|
|
boolean pad = true;
|
2016-09-08 10:40:04 +02:00
|
|
|
int count = set.size() > limit ? limit : set.size();
|
|
|
|
NumberedFilter firstRemovalKey = null;
|
2018-01-02 17:22:13 +01:00
|
|
|
logger.debug("set completo" + set.toString());
|
|
|
|
for(NumberedFilter nf : set.keySet()) {
|
|
|
|
if(--count >= 0 || limit <= 0) {
|
|
|
|
if(pad) {
|
2016-09-08 10:40:04 +02:00
|
|
|
padMap(set.get(nf), temporalConstraint);
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
} else {
|
|
|
|
if(firstRemovalKey == null) {
|
2016-09-08 10:40:04 +02:00
|
|
|
firstRemovalKey = nf;
|
2018-01-02 17:22:13 +01:00
|
|
|
} else {
|
2016-09-08 10:40:04 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
if(firstRemovalKey != null) {
|
|
|
|
logger.debug("First removal key set:" + set.subMap(set.firstKey(), firstRemovalKey));
|
2016-09-08 10:40:04 +02:00
|
|
|
}
|
2018-01-02 17:22:13 +01:00
|
|
|
|
|
|
|
logger.debug("set: " + set);
|
|
|
|
|
|
|
|
} catch(Exception e) {
|
2016-09-08 10:40:04 +02:00
|
|
|
e.printStackTrace();
|
|
|
|
}
|
|
|
|
}
|
2016-11-08 10:00:47 +01:00
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testContextService() throws Exception {
|
|
|
|
Calendar startTime = Calendar.getInstance();
|
2017-05-05 16:57:26 +02:00
|
|
|
startTime.set(2017, Calendar.APRIL, 20);
|
2016-11-08 10:00:47 +01:00
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2017, Calendar.APRIL, 28, 23, 59);
|
2016-11-08 10:00:47 +01:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
2017-05-05 16:57:26 +02:00
|
|
|
//filters.add(new Filter(AggregatedServiceUsageRecord.CALLED_METHOD, "WebProcessingService"));
|
2016-11-08 10:00:47 +01:00
|
|
|
//filters.add(new Filter(AggregatedServiceUsageRecord.CONSUMER_ID, "alessandro.pieve"));
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.DAILY);
|
2017-01-31 15:17:07 +01:00
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
Class<AggregatedServiceUsageRecord> clz = AggregatedServiceUsageRecord.class;
|
|
|
|
|
|
|
|
List<String> context = new ArrayList<String>();
|
2017-01-31 15:17:07 +01:00
|
|
|
|
|
|
|
// , /gcube/devsec/TestCreation12, /gcube/devsec/TestCreation13, /gcube/devsec/TestCreation14, /gcube/devsec/TestCreation15, /gcube/devsec/TestCreation16, /gcube/devsec/TestCreation17, /gcube/devsec/TestCreation7, /gcube/devsec/TestCreation8, /gcube/devsec/TestCreation9, /gcube/devsec/TestLucio2]]
|
2016-11-08 10:00:47 +01:00
|
|
|
context.add("/gcube");
|
|
|
|
context.add("/gcube/devNext");
|
2018-01-02 17:22:13 +01:00
|
|
|
/*
|
|
|
|
context.add("/gcube/devNext/Luciotest");
|
|
|
|
context.add("/gcube/devNext/NextNext");
|
|
|
|
context.add("/gcube/devsec");
|
|
|
|
context.add("/gcube/devsec/SmartCamera");
|
|
|
|
context.add("/gcube/devsec/statVRE");
|
|
|
|
context.add("/gcube/devsec/TestAddLast");
|
|
|
|
|
|
|
|
context.add("/gcube/devsec/devVRE");
|
|
|
|
context.add("/gcube/devsec/preVRE");
|
|
|
|
context.add("/gcube/preprod/preVRE");
|
|
|
|
context.add("/gcube/preprod");
|
|
|
|
*/
|
|
|
|
SortedMap<Filter,SortedMap<Calendar,Info>> setContext = accountingPersistenceQueryCouchBase
|
|
|
|
.getContextTimeSeries(clz, temporalConstraint, filters, context);
|
2016-11-08 10:00:47 +01:00
|
|
|
logger.debug("Result final{}", setContext);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2016-11-08 10:00:47 +01:00
|
|
|
}
|
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void getListUsage() throws Exception {
|
2017-01-19 11:51:31 +01:00
|
|
|
Calendar startTime = Calendar.getInstance();
|
|
|
|
startTime.set(2015, Calendar.SEPTEMBER, 1);
|
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2016, Calendar.OCTOBER, 20, 23, 59);
|
2017-01-19 11:51:31 +01:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
|
|
|
filters.add(new Filter(AggregatedServiceUsageRecord.CONSUMER_ID, "valentina.marioli"));
|
2017-09-21 14:51:31 +02:00
|
|
|
////TemporalConstraint temporalConstraint =
|
2018-01-02 17:22:13 +01:00
|
|
|
new TemporalConstraint(startTime.getTimeInMillis(), endTime.getTimeInMillis(), AggregationMode.DAILY);
|
2017-09-21 14:51:31 +02:00
|
|
|
//String context="/gcube/devNext";
|
2018-01-02 17:22:13 +01:00
|
|
|
List<String> parameters = new ArrayList<String>();
|
2017-01-19 11:51:31 +01:00
|
|
|
parameters.add("serviceClass");
|
|
|
|
parameters.add("serviceName");
|
2017-09-21 14:51:31 +02:00
|
|
|
//Class<AggregatedServiceUsageRecord> clz =
|
2018-01-02 17:22:13 +01:00
|
|
|
// AggregatedServiceUsageRecord.class;
|
|
|
|
// SortedMap<String,Integer> result= accountingPersistenceQueryCouchBase.getListUsage(clz,temporalConstraint, filters,context,parameters);
|
|
|
|
|
2017-05-05 16:57:26 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void getRecord() throws Exception {
|
|
|
|
String recordId = "91e1c339-d811-45d7-a13f-7385af59e3c8";
|
|
|
|
String type = "service";
|
|
|
|
|
2021-03-22 12:00:41 +01:00
|
|
|
Record record = accountingPersistenceQueryCouchBase.getRecord(recordId, type);
|
|
|
|
logger.debug("document:{}", record);
|
2017-05-05 16:57:26 +02:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
@Test
|
|
|
|
public void getQuerableKeyStorageStatus() throws Exception {
|
2017-05-05 16:57:26 +02:00
|
|
|
SortedSet<String> keys;
|
|
|
|
keys = AccountingPersistenceQuery.getQuerableKeys(AggregatedStorageStatusRecord.class);
|
2018-01-02 17:22:13 +01:00
|
|
|
for(String key : keys) {
|
|
|
|
if(key != null && !key.isEmpty()) {
|
|
|
|
logger.debug("key:" + key);
|
2017-05-05 16:57:26 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
logger.debug("List FilterKeys:" + keys.toString());
|
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void testGetSpaceProvidersIds() throws Exception {
|
|
|
|
SortedSet<String> listProvidersId = accountingPersistenceQueryCouchBase.getSpaceProvidersIds();
|
2017-05-05 16:57:26 +02:00
|
|
|
logger.debug("Result final{}", listProvidersId);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
2018-01-02 17:22:13 +01:00
|
|
|
public void testGetFilterValue() throws Exception {
|
2017-05-05 16:57:26 +02:00
|
|
|
|
|
|
|
//String key="consumerId";
|
2018-01-02 17:22:13 +01:00
|
|
|
String key = "dataServiceId";
|
2017-05-05 16:57:26 +02:00
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
SortedSet<NumberedFilter> listFilterValue = accountingPersistenceQueryCouchBase
|
|
|
|
.getFilterValues(AggregatedStorageStatusRecord.class, null, null, key, null);
|
2017-05-05 16:57:26 +02:00
|
|
|
logger.debug("Result final{}", listFilterValue);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
@Test
|
|
|
|
public void testUsageStorage() throws Exception {
|
|
|
|
Calendar startTime = Calendar.getInstance();
|
|
|
|
startTime.set(2017, Calendar.APRIL, 13);
|
|
|
|
Calendar endTime = Calendar.getInstance();
|
2018-01-02 17:22:13 +01:00
|
|
|
endTime.set(2017, Calendar.APRIL, 26, 23, 59);
|
2017-05-05 16:57:26 +02:00
|
|
|
List<Filter> filters = new ArrayList<Filter>();
|
|
|
|
//filters.add(new Filter(AggregatedStorageStatusRecord.DATA_SERVICEID, "identifier"));
|
|
|
|
filters.add(new Filter(AggregatedServiceUsageRecord.CONSUMER_ID, "alessandro.pieve"));
|
2018-01-02 17:22:13 +01:00
|
|
|
TemporalConstraint temporalConstraint = new TemporalConstraint(startTime.getTimeInMillis(),
|
|
|
|
endTime.getTimeInMillis(), AggregationMode.YEARLY);
|
|
|
|
|
|
|
|
List<String> providerId = new ArrayList<String>();
|
|
|
|
Class<AggregatedStorageStatusRecord> clz = AggregatedStorageStatusRecord.class;
|
2017-05-05 16:57:26 +02:00
|
|
|
|
|
|
|
//providerId.add("Rstudio");
|
|
|
|
providerId.add("MongoDb");
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
SortedMap<Filter,SortedMap<Calendar,Long>> setContext = accountingPersistenceQueryCouchBase
|
|
|
|
.getSpaceTimeSeries(clz, temporalConstraint, filters, providerId);
|
2017-05-05 16:57:26 +02:00
|
|
|
|
|
|
|
int count = setContext.size();
|
|
|
|
Filter firstRemovalKey = null;
|
2018-01-02 17:22:13 +01:00
|
|
|
for(Filter nf : setContext.keySet()) {
|
|
|
|
if(--count >= 0) {
|
2017-05-05 16:57:26 +02:00
|
|
|
//if(pad){
|
|
|
|
padMapStorage(setContext.get(nf), temporalConstraint);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-05-05 16:57:26 +02:00
|
|
|
//}
|
2018-01-02 17:22:13 +01:00
|
|
|
} else {
|
|
|
|
if(firstRemovalKey == null) {
|
2017-05-05 16:57:26 +02:00
|
|
|
firstRemovalKey = nf;
|
2018-01-02 17:22:13 +01:00
|
|
|
} else {
|
2017-05-05 16:57:26 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
if(firstRemovalKey!=null){
|
|
|
|
return setContext.subMap(setContext.firstKey(), firstRemovalKey);
|
|
|
|
}
|
|
|
|
*/
|
|
|
|
|
|
|
|
logger.debug("Result final{}", setContext);
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-01-19 11:51:31 +01:00
|
|
|
}
|
2017-05-05 16:57:26 +02:00
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
public SortedMap<Calendar,Long> padMapStorage(SortedMap<Calendar,Long> unpaddedData,
|
2017-05-05 16:57:26 +02:00
|
|
|
TemporalConstraint temporalConstraint) throws Exception {
|
2018-01-02 17:22:13 +01:00
|
|
|
|
2017-05-05 16:57:26 +02:00
|
|
|
//JSONObject jsonObject = getPaddingJSONObject(unpaddedData);
|
|
|
|
SortedSet<Calendar> sequence = temporalConstraint.getCalendarSequence();
|
|
|
|
Long longValuePre = null;
|
2018-01-02 17:22:13 +01:00
|
|
|
for(Calendar progressTime : sequence) {
|
2017-05-05 16:57:26 +02:00
|
|
|
Long longValue = unpaddedData.get(progressTime);
|
|
|
|
|
2018-01-02 17:22:13 +01:00
|
|
|
if(longValue == null) {
|
2017-05-05 16:57:26 +02:00
|
|
|
unpaddedData.put(progressTime, longValuePre);
|
2018-01-02 17:22:13 +01:00
|
|
|
} else {
|
|
|
|
longValuePre = longValue;
|
2017-05-05 16:57:26 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
return unpaddedData;
|
|
|
|
}
|
|
|
|
|
2016-09-08 10:40:04 +02:00
|
|
|
}
|