accounting-analytics-persistence-postgresql/src/main/java/org/gcube/accounting/analytics/persistence/postgresql/AccountingPersistenceQueryPostgreSQL.java

/**
 * 
 */
package org.gcube.accounting.analytics.persistence.postgresql;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;
import java.time.OffsetDateTime;
import java.util.Calendar;
import java.util.List;
import java.util.Map;
import java.util.SortedMap;
import java.util.SortedSet;
import java.util.TreeMap;

import org.gcube.accounting.analytics.Filter;
import org.gcube.accounting.analytics.Info;
import org.gcube.accounting.analytics.NumberedFilter;
import org.gcube.accounting.analytics.TemporalConstraint;
import org.gcube.accounting.analytics.UsageValue;
import org.gcube.accounting.analytics.exception.DuplicatedKeyFilterException;
import org.gcube.accounting.analytics.exception.KeyException;
import org.gcube.accounting.analytics.exception.ValueException;
import org.gcube.accounting.analytics.persistence.AccountingPersistenceBackendQuery;
import org.gcube.accounting.analytics.persistence.AccountingPersistenceBackendQueryConfiguration;
import org.gcube.accounting.datamodel.aggregation.AggregatedServiceUsageRecord;
import org.gcube.accounting.datamodel.usagerecords.ServiceUsageRecord;
import org.gcube.accounting.persistence.AccountingPersistenceConfiguration;
import org.gcube.documentstore.records.AggregatedRecord;
import org.gcube.documentstore.records.RecordUtility;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Luca Frosini (ISTI - CNR)
 */
public class AccountingPersistenceQueryPostgreSQL implements AccountingPersistenceBackendQuery {
	
	private static final Logger logger = LoggerFactory.getLogger(AccountingPersistenceQueryPostgreSQL.class);
	
	public static final String DATETIME_PATTERN = "yyyy-MM-dd HH:mm:ss.SSS Z";
	
	public static final String URL_PROPERTY_KEY = AccountingPersistenceConfiguration.URL_PROPERTY_KEY;
	public static final String USERNAME_PROPERTY_KEY = AccountingPersistenceConfiguration.USERNAME_PROPERTY_KEY;
	public static final String PASSWORD_PROPERTY_KEY = AccountingPersistenceConfiguration.PASSWORD_PROPERTY_KEY;

	private String url;
	private String username;
	private String password;
	
	protected AccountingPersistenceBackendQueryConfiguration configuration;
	
	protected Map<String, String> typeNameToConnectionURL;
	
	static {
		// One Record per package is enough
		RecordUtility.addRecordPackage(ServiceUsageRecord.class.getPackage());
		RecordUtility.addRecordPackage(AggregatedServiceUsageRecord.class.getPackage());
		Map<String, Class<? extends AggregatedRecord<?,?>>> aggregatedRecords = RecordUtility.getAggregatedRecordClassesFound();
		for(String typeName : aggregatedRecords.keySet()) {
			try {
				Class<? extends AggregatedRecord<?,?>> clz = aggregatedRecords.get(typeName); 
				UsageRecordToDBMapping.getRecordToDB(clz);
			} catch (Exception e) {
				new RuntimeException(e);
			}
		}
	}
	
	protected Connection getConnection(Class<? extends AggregatedRecord<?, ?>> clz) throws Exception {
		// String typeName = UsageRecordToDBMapping.getRecordToDB(clz).getTypeName();
		Class.forName("org.postgresql.Driver");
		Connection connection = DriverManager.getConnection(url, username, password);
		logger.trace("Database opened successfully");
		connection.setAutoCommit(false);
		return connection;
	}
	
	@Override
	public void prepareConnection(AccountingPersistenceBackendQueryConfiguration configuration) throws Exception {
		logger.trace("prepareConnection");
		this.configuration = configuration;
		this.url = configuration.getProperty(URL_PROPERTY_KEY);
		this.username = configuration.getProperty(USERNAME_PROPERTY_KEY);
		this.password = configuration.getProperty(PASSWORD_PROPERTY_KEY);
	}
	
	@Override
	public SortedMap<Calendar, Info> getTimeSeries(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters)
			throws DuplicatedKeyFilterException, KeyException, ValueException, Exception {
		// TODO Auto-generated method stub
		return null;
	}
	
	protected Calendar getCalendar(OffsetDateTime offsetDateTime) {
		Calendar calendar = Calendar.getInstance();
		long epochMillis = offsetDateTime.toInstant().toEpochMilli();
		calendar.setTimeInMillis(epochMillis);
		return calendar;
	}
	
	@Override
	public SortedMap<Calendar, Info> getNoContextTimeSeries(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters)
			throws DuplicatedKeyFilterException, KeyException, ValueException, Exception {
		Connection connection = getConnection(clz);
		Statement statement = connection.createStatement();
		
		SortedMap<Calendar, Info> result = new TreeMap<>();
		
		Query query = new Query(clz);
		query.setTemporalConstraint(temporalConstraint);
		query.setFilters(filters);
		String sql = query.getNoContextTimeSeriesQuery();
		
		List<String> requestedTableField = query.getRequestedTableField();
		RecordToDBMapper recordToDBMapper = query.getRecordToDBMapper();
		
		ResultSet resultSet = statement.executeQuery(sql);
		
		while (resultSet.next()) {
			OffsetDateTime offsetDateTime = resultSet.getObject(Query.DATE_OF_TIMESERIES_AS_FIELD, OffsetDateTime.class);
			Calendar calendar = getCalendar(offsetDateTime);
			
			JSONObject jsonObject = new JSONObject();
			
			for(String tableField : requestedTableField) {
				String usageRecordField = recordToDBMapper.getUsageRecordField(tableField);
				Object object = resultSet.getObject(tableField);
				jsonObject.put(usageRecordField, object);
			}
			
			Info info = new Info(calendar, jsonObject);
			result.put(calendar, info);
		}
		
		return result;
	}

	@Override
	public SortedMap<NumberedFilter, SortedMap<Calendar, Info>> getTopValues(
			Class<? extends AggregatedRecord<?, ?>> clz, TemporalConstraint temporalConstraint, List<Filter> filters,
			String topKey, String orderingProperty)
			throws DuplicatedKeyFilterException, KeyException, ValueException, Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedSet<NumberedFilter> getNextPossibleValues(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters, String key, String orderingProperty)
			throws DuplicatedKeyFilterException, KeyException, ValueException, Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedSet<NumberedFilter> getNextPossibleValuesWithMap(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters, String key, String orderingProperty)
			throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public void close() throws Exception {
		// TODO Auto-generated method stub
		
	}

	@Override
	public SortedSet<NumberedFilter> getFilterValues(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters, String key) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedSet<NumberedFilter> getFilterValues(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters, String key, Integer limit) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public JSONObject getUsageValue(Class<? extends AggregatedRecord<?, ?>> clz, TemporalConstraint temporalConstraint,
			Filter applicant) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public List<UsageValue> getUsageValueQuotaTotal(List<UsageValue> listUsage) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedMap<Filter, SortedMap<Calendar, Info>> getContextTimeSeries(
			Class<? extends AggregatedRecord<?, ?>> clz, TemporalConstraint temporalConstraint, List<Filter> filters,
			List<String> contexts) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public String getRecord(String recordId, String type) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedSet<String> getSpaceProvidersIds() throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public SortedMap<Filter, SortedMap<Calendar, Long>> getSpaceTimeSeries(Class<? extends AggregatedRecord<?, ?>> clz,
			TemporalConstraint temporalConstraint, List<Filter> filters, List<String> providersId) throws Exception {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public boolean isConnectionActive() throws Exception {
		// TODO Auto-generated method stub
		return false;
	}
	
	
}