package com.megatim.fdxcommons.tools.database.queries;
|
|
import com.megatim.fdxcommons.tools.database.queries.metadata.QueryCriterion;
|
import com.megatim.fdxcommons.tools.database.contrat.DefaultPreparedStatementProcessor;
|
import com.megatim.fdxcommons.tools.database.tables.FdxTableRow;
|
import com.megatim.fdxcommons.tools.database.contrat.QueryMetaData;
|
import com.megatim.fdxcommons.tools.database.queries.metadata.SelectQueryResult;
|
import com.megatim.fdxcommons.tools.database.queries.metadata.WhereQueryParameters;
|
import com.megatim.fdxcommons.tools.database.tables.appcolumns.FdxApiColumnDefinitions;
|
import com.megatim.fdxcommons.tools.database.tables.FdxTableColumnData;
|
import com.megatim.fdxcommons.tools.database.tables.appcolumns.FdxConsultationColumnDefinitions;
|
import com.megatim.fdxcommons.tools.database.tables.appcolumns.IndexColumnDefinition;
|
import com.megatim.fdxcommons.tools.database.tables.appcolumns.TokenColumnDefinition;
|
import java.sql.Connection;
|
import java.sql.PreparedStatement;
|
import java.sql.ResultSet;
|
import java.sql.ResultSetMetaData;
|
import java.sql.SQLException;
|
import java.util.ArrayList;
|
import java.util.List;
|
import javax.naming.NamingException;
|
|
/**
|
*
|
* @author Gabuntu
|
*/
|
public class SelectQuery implements QueryMetaData<SelectQueryResult> {
|
|
private final String tableName;
|
private final Connection connection;
|
private final QueryCriterion queryCriterion;
|
private final List<String> selectColumns;
|
private final int limit;
|
private final boolean withCustomColumns;
|
private final int offset;
|
|
public SelectQuery(String tableName, Connection connection, QueryCriterion queryCriterion, List<String> selectColumns, boolean withCustomColumns) {
|
this(tableName, connection, queryCriterion, selectColumns, -1, -1, withCustomColumns);
|
}
|
|
public SelectQuery(String tableName, Connection connection, QueryCriterion queryCriterion, List<String> selectColumns, int limit, int offset, boolean withCustomColumns) {
|
this.tableName = tableName;
|
this.connection = connection;
|
this.queryCriterion = queryCriterion;
|
this.selectColumns = selectColumns;
|
this.limit = limit;
|
this.offset = offset;
|
this.withCustomColumns = withCustomColumns;
|
}
|
|
@Override
|
public SelectQueryResult execute() throws Exception {
|
return result();
|
}
|
|
private SelectQueryResult result() throws NamingException, SQLException {
|
String query = selectQuery();
|
System.out.println("------------------------------------------");
|
System.out.println(query);
|
System.out.println("------------------------------------------------");
|
|
try ( PreparedStatement statement = connection.prepareStatement(query)) {
|
|
new DefaultPreparedStatementProcessor().process(statement, new WhereQueryParameters(queryCriterion).parameters());
|
|
long rowCount = 1;
|
|
ResultSet resultSet = statement.executeQuery();
|
ResultSetMetaData metaData = resultSet.getMetaData();
|
int columnCount = metaData.getColumnCount();
|
|
List<FdxTableRow> rows = new ArrayList<>();
|
|
int lastIndexRead = 0;
|
|
FdxApiColumnDefinitions apiColumnDefinitions = new FdxApiColumnDefinitions();
|
FdxConsultationColumnDefinitions fdxConsColumnDefinitions = new FdxConsultationColumnDefinitions();
|
final String tokenColumn = new TokenColumnDefinition().name();
|
|
while (resultSet.next()) {
|
|
List<FdxTableColumnData> data = new ArrayList<>();
|
|
for (int i = 1; i <= columnCount; i++) {
|
|
String name = metaData.getColumnName(i).trim();
|
Object value = resultSet.getObject(i);
|
// int sqlDataType = metaData.getColumnType(i);
|
|
if (!name.equals(tokenColumn)) {
|
if (withCustomColumns) {
|
data.add(new FdxTableColumnData(value, name, i));
|
} else {
|
if (!apiColumnDefinitions.isAppColumnDefinition(name) && !fdxConsColumnDefinitions.isAppColumnDefinition(name)) {
|
data.add(new FdxTableColumnData(value, name, i));
|
}
|
}
|
if (apiColumnDefinitions.isIndexColumn(name)) {
|
int currentIndex = Integer.parseInt(value.toString());
|
if (currentIndex > lastIndexRead) {
|
lastIndexRead = currentIndex;
|
}
|
}
|
}
|
}
|
|
rows.add(new FdxTableRow(rowCount++, data));
|
}
|
return new SelectQueryResult(lastIndexRead, rows);
|
}
|
}
|
|
private String selectQuery() {
|
|
StringBuilder query = new StringBuilder("SELECT ");
|
|
if (selectColumns.isEmpty()) {
|
|
query.append(" * ");
|
|
} else {
|
|
selectColumns.forEach(c -> query.append(c.toLowerCase()).append(","));
|
query.replace(query.length() - 1, query.length(), ""); //retirer la sernière virgule
|
|
}
|
|
query
|
.append(" FROM ")
|
.append(tableName);
|
|
query.append(new WhereQueryString(queryCriterion).query());
|
query.append(" ORDER BY ").append(new IndexColumnDefinition().name());
|
|
if (limit > 0 && offset >= 0) {
|
|
query
|
.append(" LIMIT ")
|
.append(limit)
|
.append(" OFFSET ")
|
.append(offset);
|
}
|
return query.toString();
|
}
|
}
|
