aswi2020vldc-gitlab/src/main/java/vldc/aswi/service/SqlQueryManagerImpl.java @ 9e9e8d67

package vldc.aswi.service;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import vldc.aswi.database.DatabaseInterface;
import vldc.aswi.domain.Assembly;
import vldc.aswi.domain.Configuration;
import vldc.aswi.domain.Function;
import vldc.aswi.domain.parameter.ParameterInConfiguration;
import vldc.aswi.model.table.NameUserName;
import vldc.aswi.model.table.TableColumn;
import vldc.aswi.model.table.ValueFunction;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
import vldc.aswi.utils.Converter;

import javax.sql.DataSource;
import java.util.*;
import java.util.concurrent.TimeUnit;

/**
 * Manager for SqlQuery.
 */
@Service
@Slf4j
public class SqlQueryManagerImpl implements SqlQueryManager{

    /** Autowired databaseInterface component. */
    @Autowired
    private DatabaseInterface databaseInterface;

    /**
     * Get list of  contingencyTableRow.
     * @param configuration - configuration from which table will be created.
     * @return List of contingencyTableRow.
     */
    @Override
    public List<ContingencyTableRow> getContingencyTableRow(Configuration configuration) {
        long startTime = System.nanoTime();
        Map<String, TableColumn> tableColumns = this.databaseInterface.executeQueryAndReturnTableColumns(
                configuration.getAssembly().getSQLQuery()
        );
        long stopTime = System.nanoTime();
        System.out.println("Select:" + TimeUnit.MILLISECONDS.convert((stopTime - startTime), TimeUnit.NANOSECONDS));
        List<NameUserName> rowNames = new ArrayList<>();
        List<NameUserName> columnNames = new ArrayList<>();
        List<ValueFunction> valueFunctions = new ArrayList<>();

        // sort parameters in configuration to have them in order set by user
        Collections.sort(configuration.getParametersInConfiguration());

        // TODO: 28.05.2020 map would be better
        for(ParameterInConfiguration parameterInConfiguration : configuration.getParametersInConfiguration()) {
            if (parameterInConfiguration.getLocation() != null && parameterInConfiguration.getLocation().getName().equals("Řádek")) {
                rowNames.add(new NameUserName(parameterInConfiguration.getParameter().getNameOfSelect().toUpperCase(),
                        parameterInConfiguration.getColumnName()));
            }
            else if (parameterInConfiguration.getLocation() != null && parameterInConfiguration.getLocation().getName().equals("Sloupec")) {
                columnNames.add(new NameUserName(parameterInConfiguration.getParameter().getNameOfSelect().toUpperCase(),
                        parameterInConfiguration.getColumnName()));
            }
            else if(parameterInConfiguration.getLocation() != null && parameterInConfiguration.getLocation().getName().equals("Hodnota")) {
                for (Function function : parameterInConfiguration.getFunctions()) {
                    valueFunctions.add(new ValueFunction(parameterInConfiguration.getParameter().getNameOfSelect().toUpperCase(),
                            function.getName().toUpperCase()));
                }
            }
        }

        Converter converter = new Converter();
        return converter.convertTableColumnsToContingencyTableRows(tableColumns, rowNames, columnNames, valueFunctions);
    }

    /**
     * Validate given SQL query.
     * @param sqlQuery - Validated SQL query.
     * @return true if SQL query is ok, false if not.
     */
    @Override
    public boolean validateSQLQuery(String sqlQuery) {
        return this.databaseInterface.validateSQLQuery(sqlQuery);
    }

    /**
     * Get list of table column names from SQL query.
     * @param sqlQuery - SQL query.
     * @return List of table column names.
     */
    public List<String> getNameOfColumnsFromQuery(String sqlQuery) {
        return this.databaseInterface.getNameOfColumnsFromQuery(sqlQuery);
    }
}