fix compilation errors after merging master
This commit is contained in:
Andrea Boriero
parent
638a3e234b
commit
076a0aa25a
|
|
@ -1,120 +0,0 @@
|
||||||
/*
|
|
||||||
* Hibernate, Relational Persistence for Idiomatic Java
|
|
||||||
*
|
|
||||||
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
|
|
||||||
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
|
|
||||||
*/
|
|
||||||
package org.hibernate.hql.internal.ast.exec;
|
|
||||||
|
|
||||||
import antlr.RecognitionException;
|
|
||||||
import org.hibernate.AssertionFailure;
|
|
||||||
import org.hibernate.HibernateException;
|
|
||||||
import org.hibernate.dialect.Dialect;
|
|
||||||
import org.hibernate.dialect.MySQLDialect;
|
|
||||||
import org.hibernate.hql.internal.ast.HqlSqlWalker;
|
|
||||||
import org.hibernate.hql.internal.ast.QuerySyntaxException;
|
|
||||||
import org.hibernate.hql.internal.ast.SqlGenerator;
|
|
||||||
import org.hibernate.hql.internal.ast.tree.AssignmentSpecification;
|
|
||||||
import org.hibernate.hql.internal.ast.tree.UpdateStatement;
|
|
||||||
import org.hibernate.param.ParameterSpecification;
|
|
||||||
import org.hibernate.persister.entity.Queryable;
|
|
||||||
import org.hibernate.sql.Update;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.stream.IntStream;
|
|
||||||
|
|
||||||
import static org.hibernate.hql.spi.id.AbstractTableBasedBulkIdHandler.generateIdSelect;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Executes HQL bulk updates against a single table, using a subselect
|
|
||||||
* against multiple tables to collect ids, which is needed when the
|
|
||||||
* where condition of the query touches columns from multiple tables.
|
|
||||||
*
|
|
||||||
* @author Gavin King
|
|
||||||
*/
|
|
||||||
public class IdSubselectUpdateExecutor extends BasicExecutor {
|
|
||||||
|
|
||||||
private final Queryable persister;
|
|
||||||
private final String sql;
|
|
||||||
private final List<ParameterSpecification> parameterSpecifications;
|
|
||||||
|
|
||||||
public Queryable getPersister() {
|
|
||||||
return persister;
|
|
||||||
}
|
|
||||||
@Override
|
|
||||||
public String getSql() {
|
|
||||||
return sql;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<ParameterSpecification> getParameterSpecifications() {
|
|
||||||
return parameterSpecifications;
|
|
||||||
}
|
|
||||||
|
|
||||||
public IdSubselectUpdateExecutor(HqlSqlWalker walker) {
|
|
||||||
persister = walker.getFinalFromClause().getFromElement().getQueryable();
|
|
||||||
|
|
||||||
Dialect dialect = walker.getDialect();
|
|
||||||
UpdateStatement updateStatement = (UpdateStatement) walker.getAST();
|
|
||||||
List<AssignmentSpecification> assignments = walker.getAssignmentSpecifications();
|
|
||||||
|
|
||||||
String whereClause;
|
|
||||||
if ( updateStatement.getWhereClause().getNumberOfChildren() == 0 ) {
|
|
||||||
whereClause = "";
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
try {
|
|
||||||
SqlGenerator gen = new SqlGenerator( walker.getSessionFactoryHelper().getFactory() );
|
|
||||||
gen.whereClause( updateStatement.getWhereClause() );
|
|
||||||
gen.getParseErrorHandler().throwQueryException();
|
|
||||||
whereClause = gen.getSQL().substring( 7 ); // strip the " where "
|
|
||||||
}
|
|
||||||
catch ( RecognitionException e ) {
|
|
||||||
throw new HibernateException( "Unable to generate id select for DML operation", e );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
String tableAlias = updateStatement.getFromClause().getFromElement().getTableAlias();
|
|
||||||
String idSelect = generateIdSelect( tableAlias, whereClause, dialect, persister );
|
|
||||||
|
|
||||||
String[] tableNames = persister.getConstraintOrderedTableNameClosure();
|
|
||||||
String[][] columnNames = persister.getContraintOrderedTableKeyColumnClosure();
|
|
||||||
|
|
||||||
int[] affectedTables =
|
|
||||||
IntStream.range( 0, tableNames.length ).filter(
|
|
||||||
table -> assignments.stream().anyMatch(
|
|
||||||
assign -> assign.affectsTable( tableNames[table] )
|
|
||||||
)
|
|
||||||
).toArray();
|
|
||||||
if ( affectedTables.length > 1 ) {
|
|
||||||
throw new AssertionFailure("more than one affected table");
|
|
||||||
}
|
|
||||||
int affectedTable = affectedTables[0];
|
|
||||||
|
|
||||||
String tableName = tableNames[affectedTable];
|
|
||||||
String idColumnNames = String.join( ", ", columnNames[affectedTable] );
|
|
||||||
Update update = new Update( dialect ).setTableName( tableName );
|
|
||||||
if ( dialect instanceof MySQLDialect) {
|
|
||||||
//MySQL needs an extra subselect to hack the query optimizer
|
|
||||||
String selectedIdColumns = String.join( ", ", persister.getIdentifierColumnNames() );
|
|
||||||
update.setWhere( "(" + idColumnNames + ") in (select " + selectedIdColumns + " from (" + idSelect + ") as ht_ids)" );
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
update.setWhere( "(" + idColumnNames + ") in (" + idSelect + ")" );
|
|
||||||
}
|
|
||||||
for ( AssignmentSpecification assignment: assignments ) {
|
|
||||||
update.appendAssignmentFragment( assignment.getSqlAssignmentFragment() );
|
|
||||||
}
|
|
||||||
sql = update.toStatementString();
|
|
||||||
|
|
||||||
// now collect the parameters from the whole query
|
|
||||||
// parameters included in the list
|
|
||||||
try {
|
|
||||||
SqlGenerator gen = new SqlGenerator( walker.getSessionFactoryHelper().getFactory() );
|
|
||||||
gen.statement( walker.getAST() );
|
|
||||||
parameterSpecifications = gen.getCollectedParameters();
|
|
||||||
}
|
|
||||||
catch ( RecognitionException e ) {
|
|
||||||
throw QuerySyntaxException.convert( e );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,59 +0,0 @@
|
||||||
/*
|
|
||||||
* Hibernate, Relational Persistence for Idiomatic Java
|
|
||||||
*
|
|
||||||
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
|
|
||||||
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
|
|
||||||
*/
|
|
||||||
package org.hibernate.hql.internal.ast.exec;
|
|
||||||
|
|
||||||
import antlr.RecognitionException;
|
|
||||||
import org.hibernate.hql.internal.antlr.HqlSqlTokenTypes;
|
|
||||||
import org.hibernate.hql.internal.ast.HqlSqlWalker;
|
|
||||||
import org.hibernate.hql.internal.ast.QuerySyntaxException;
|
|
||||||
import org.hibernate.hql.internal.ast.SqlGenerator;
|
|
||||||
import org.hibernate.hql.internal.ast.tree.InsertStatement;
|
|
||||||
import org.hibernate.hql.internal.ast.tree.Statement;
|
|
||||||
import org.hibernate.param.ParameterSpecification;
|
|
||||||
import org.hibernate.persister.entity.Queryable;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Executes HQL insert statements.
|
|
||||||
*
|
|
||||||
* @author Gavin King
|
|
||||||
*/
|
|
||||||
public class InsertExecutor extends BasicExecutor {
|
|
||||||
private final Queryable persister;
|
|
||||||
private final String sql;
|
|
||||||
private final List<ParameterSpecification> parameterSpecifications;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Queryable getPersister() {
|
|
||||||
return persister;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getSql() {
|
|
||||||
return sql;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<ParameterSpecification> getParameterSpecifications() {
|
|
||||||
return parameterSpecifications;
|
|
||||||
}
|
|
||||||
|
|
||||||
public InsertExecutor(HqlSqlWalker walker) {
|
|
||||||
persister = ( (InsertStatement) walker.getAST() ).getIntoClause().getQueryable();
|
|
||||||
try {
|
|
||||||
SqlGenerator gen = new SqlGenerator( walker.getSessionFactoryHelper().getFactory() );
|
|
||||||
gen.statement( walker.getAST() );
|
|
||||||
sql = gen.getSQL();
|
|
||||||
gen.getParseErrorHandler().throwQueryException();
|
|
||||||
parameterSpecifications = gen.getCollectedParameters();
|
|
||||||
}
|
|
||||||
catch ( RecognitionException e ) {
|
|
||||||
throw QuerySyntaxException.convert( e );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,69 +0,0 @@
|
||||||
/*
|
|
||||||
* Hibernate, Relational Persistence for Idiomatic Java
|
|
||||||
*
|
|
||||||
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
|
|
||||||
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
|
|
||||||
*/
|
|
||||||
package org.hibernate.hql.internal.ast.exec;
|
|
||||||
|
|
||||||
import antlr.RecognitionException;
|
|
||||||
import org.hibernate.AssertionFailure;
|
|
||||||
import org.hibernate.hql.internal.ast.HqlSqlWalker;
|
|
||||||
import org.hibernate.hql.internal.ast.QuerySyntaxException;
|
|
||||||
import org.hibernate.hql.internal.ast.SqlGenerator;
|
|
||||||
import org.hibernate.param.ParameterSpecification;
|
|
||||||
import org.hibernate.persister.entity.Queryable;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Executes HQL bulk updates against a single table, where the
|
|
||||||
* query only touches columns from the table it's updating, and
|
|
||||||
* so we don't need to use a subselect.
|
|
||||||
*
|
|
||||||
* @author Gavin King
|
|
||||||
*/
|
|
||||||
public class SimpleUpdateExecutor extends BasicExecutor {
|
|
||||||
|
|
||||||
private final Queryable persister;
|
|
||||||
private final String sql;
|
|
||||||
private final List<ParameterSpecification> parameterSpecifications;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public Queryable getPersister() {
|
|
||||||
return persister;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String getSql() {
|
|
||||||
return sql;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public List<ParameterSpecification> getParameterSpecifications() {
|
|
||||||
return parameterSpecifications;
|
|
||||||
}
|
|
||||||
|
|
||||||
public SimpleUpdateExecutor(HqlSqlWalker walker) {
|
|
||||||
persister = walker.getFinalFromClause().getFromElement().getQueryable();
|
|
||||||
|
|
||||||
if ( persister.isMultiTable() && walker.getQuerySpaces().size() > 1 ) {
|
|
||||||
throw new AssertionFailure("not a simple update");
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
SqlGenerator gen = new SqlGenerator( walker.getSessionFactoryHelper().getFactory() );
|
|
||||||
gen.statement( walker.getAST() );
|
|
||||||
gen.getParseErrorHandler().throwQueryException();
|
|
||||||
// workaround for a problem where HqlSqlWalker actually generates
|
|
||||||
// broken SQL with undefined aliases in the where clause, because
|
|
||||||
// that is what MultiTableUpdateExecutor is expecting to get
|
|
||||||
String alias = walker.getFinalFromClause().getFromElement().getTableAlias();
|
|
||||||
sql = gen.getSQL().replace( alias + ".", "" );
|
|
||||||
parameterSpecifications = gen.getCollectedParameters();
|
|
||||||
}
|
|
||||||
catch ( RecognitionException e ) {
|
|
||||||
throw QuerySyntaxException.convert( e );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue