package com.megatim.fdxcommons.tools.database.queries;
|
|
import com.megatim.fdxcommons.tools.database.contrat.DefaultPreparedStatementProcessor;
|
import com.megatim.fdxcommons.tools.database.contrat.PreparedStatementProcessor;
|
import com.megatim.fdxcommons.tools.database.contrat.QueryMetaData;
|
import com.megatim.fdxcommons.tools.database.contrat.TypeFichierDataColumn;
|
import com.megatim.fdxcommons.tools.database.contrat.TypeFichierDataRow;
|
import com.megatim.fdxcommons.tools.database.queries.metadata.DefaultTypeFichierDataColumn;
|
import com.megatim.fdxcommons.tools.database.queries.metadata.QueryParam;
|
import java.sql.Connection;
|
import java.sql.PreparedStatement;
|
import java.sql.SQLException;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
*
|
* @author Gabuntu
|
*/
|
public class InsertQuery implements QueryMetaData<Void> {
|
|
private final String tableName;
|
protected final Connection connection;
|
private final List<String> columnNames;
|
private final List<TypeFichierDataRow> rows;
|
private final List<DefaultTypeFichierDataColumn> dataColumnsToAdd;
|
private final int BATCH_SIZE = 100_000;
|
|
public InsertQuery(String tableName, final Connection connection, List<DefaultTypeFichierDataColumn> dataColumnsToAdd, List<String> columnNames, List<TypeFichierDataRow> rows) {
|
this.tableName = tableName;
|
this.dataColumnsToAdd = dataColumnsToAdd;
|
this.connection = connection;
|
this.columnNames = columnNames;
|
this.rows = rows;
|
}
|
|
@Override
|
public Void execute() throws Exception {
|
System.out.println("<-------------------------------------------------------------------------->");
|
System.out.println(insertQueryString());
|
System.out.println("<-------------------------------------------------------------------------->");
|
|
try ( PreparedStatement statement = connection.prepareStatement(insertQueryString());) {
|
|
executeInBatch(statement, rows, dataColumnsToAdd);
|
statement.executeBatch();
|
}
|
return null;
|
}
|
|
private String insertQueryString() {
|
List<String> insertColumns = new ArrayList<>(columnNames);
|
|
for (DefaultTypeFichierDataColumn dc : dataColumnsToAdd) {
|
insertColumns.add(dc.column());
|
}
|
|
StringBuilder query = new StringBuilder("INSERT INTO " + tableName + "(");
|
|
insertColumns.forEach(column -> query.append(column).append(","));
|
|
query.replace(query.length() - 1, query.length(), ") VALUES(");
|
|
insertColumns.forEach(s -> query.append("?,"));
|
|
query.replace(query.length() - 1, query.length(), ")");
|
|
return query.toString();
|
}
|
|
private void executeInBatch(PreparedStatement preparedStatement, List<TypeFichierDataRow> rows, List<DefaultTypeFichierDataColumn> dataColumnsToAdd) throws SQLException, Exception {
|
|
int index = 0;
|
int count = 0;
|
|
for (TypeFichierDataRow row : rows) {
|
|
List<QueryParam> queryParameters = new ArrayList<>();
|
|
for (TypeFichierDataColumn dataColumn : row.columns()) {
|
queryParameters.add(new QueryParam(dataColumn.column(), dataColumn.value()));
|
}
|
|
//On ajoute les colonnes de dataColumnsToAdd
|
for (DefaultTypeFichierDataColumn dc : dataColumnsToAdd) {
|
queryParameters.add(new QueryParam(dc.column(), dc.value()));
|
}
|
|
PreparedStatementProcessor statementProcessor = new DefaultPreparedStatementProcessor(index);
|
statementProcessor.process(preparedStatement, queryParameters);
|
index = statementProcessor.lastProcessedIndex();
|
|
preparedStatement.addBatch();
|
|
if (++count % BATCH_SIZE == 0) {
|
preparedStatement.executeBatch();
|
}
|
|
}
|
|
}
|
|
}
|
