HHH-9222 - Bulk delete and element collections

This commit is contained in:
Steve Ebersole 2014-05-27 15:43:04 -05:00
parent 420296fd26
commit d36b3a3e97
4 changed files with 246 additions and 44 deletions

View File

@ -79,13 +79,28 @@ public class DeleteExecutor extends BasicExecutor {
idSubselectWhere = ""; idSubselectWhere = "";
} }
// If many-to-many, delete the FK row in the collection table. // find plural attributes defined for the entity being deleted...
for ( Type type : persister.getPropertyTypes() ) { for ( Type type : persister.getPropertyTypes() ) {
if ( type.isCollectionType() ) { if ( ! type.isCollectionType() ) {
continue;
}
// if the plural attribute maps to a "collection table" we need
// to remove the rows from that table corresponding to any
// owners we are about to delete. "collection table" is
// (unfortunately) indicated in a number of ways, but here we
// are mainly concerned with:
// 1) many-to-many mappings
// 2) basic collection mappings
final CollectionType cType = (CollectionType) type; final CollectionType cType = (CollectionType) type;
final AbstractCollectionPersister cPersister = (AbstractCollectionPersister) factory final AbstractCollectionPersister cPersister = (AbstractCollectionPersister) factory
.getCollectionPersister( cType.getRole() ); .getCollectionPersister( cType.getRole() );
if ( cPersister.isManyToMany() ) { final boolean hasCollectionTable = cPersister.isManyToMany()
|| !cPersister.getElementType().isAssociationType();
if ( !hasCollectionTable ) {
continue;
}
if ( persister.getIdentifierColumnNames().length > 1 if ( persister.getIdentifierColumnNames().length > 1
&& !dialect.supportsTuplesInSubqueries() ) { && !dialect.supportsTuplesInSubqueries() ) {
LOG.warn( LOG.warn(
@ -93,8 +108,9 @@ public class DeleteExecutor extends BasicExecutor {
" when the entity has multiple primary keys. Either properly setup cascading on" + " when the entity has multiple primary keys. Either properly setup cascading on" +
" the constraints or manually clear the associations prior to deleting the entities." " the constraints or manually clear the associations prior to deleting the entities."
); );
continue;
} }
else {
final String idSubselect = "(select " final String idSubselect = "(select "
+ StringHelper.join( ", ", persister.getIdentifierColumnNames() ) + " from " + StringHelper.join( ", ", persister.getIdentifierColumnNames() ) + " from "
+ persister.getTableName() + idSubselectWhere + ")"; + persister.getTableName() + idSubselectWhere + ")";
@ -102,14 +118,11 @@ public class DeleteExecutor extends BasicExecutor {
+ ") in " + idSubselect; + ") in " + idSubselect;
final Delete delete = new Delete().setTableName( cPersister.getTableName() ).setWhere( where ); final Delete delete = new Delete().setTableName( cPersister.getTableName() ).setWhere( where );
if ( factory.getSettings().isCommentsEnabled() ) { if ( factory.getSettings().isCommentsEnabled() ) {
delete.setComment( "delete FKs in join table" ); delete.setComment( "bulk delete - collection table clean up (" + cPersister.getRole() + ")" );
} }
deletes.add( delete.toStatementString() ); deletes.add( delete.toStatementString() );
} }
} }
}
}
}
catch (RecognitionException e) { catch (RecognitionException e) {
throw new HibernateException( "Unable to delete the FKs in the join table!", e ); throw new HibernateException( "Unable to delete the FKs in the join table!", e );
} }

View File

@ -83,17 +83,39 @@ public class TableBasedDeleteHandlerImpl
final String idSubselect = generateIdSubselect( targetedPersister ); final String idSubselect = generateIdSubselect( targetedPersister );
deletes = new ArrayList<String>(); deletes = new ArrayList<String>();
// If many-to-many, delete the FK row in the collection table. // find plural attributes defined for the entity being deleted...
// This partially overlaps with DeleteExecutor, but it instead uses the temp table in the idSubselect. //
// NOTE : this partially overlaps with DeleteExecutor, but it instead
// uses the temp table in the idSubselect.
for ( Type type : targetedPersister.getPropertyTypes() ) { for ( Type type : targetedPersister.getPropertyTypes() ) {
if ( type.isCollectionType() ) { if ( ! type.isCollectionType() ) {
CollectionType cType = (CollectionType) type; continue;
AbstractCollectionPersister cPersister = (AbstractCollectionPersister)factory.getCollectionPersister( cType.getRole() );
if ( cPersister.isManyToMany() ) {
deletes.add( generateDelete( cPersister.getTableName(),
cPersister.getKeyColumnNames(), idSubselect, "bulk delete - m2m join table cleanup"));
} }
final CollectionType cType = (CollectionType) type;
final AbstractCollectionPersister cPersister = (AbstractCollectionPersister)factory.getCollectionPersister( cType.getRole() );
// if the plural attribute maps to a "collection table" we need
// to remove the rows from that table corresponding to any
// owners we are about to delete. "collection table" is
// (unfortunately) indicated in a number of ways, but here we
// are mainly concerned with:
// 1) many-to-many mappings
// 2) basic collection mappings
final boolean hasCollectionTable = cPersister.isManyToMany()
|| !cPersister.getElementType().isAssociationType();
if ( !hasCollectionTable ) {
continue;
} }
deletes.add(
generateDelete(
cPersister.getTableName(),
cPersister.getKeyColumnNames(),
idSubselect,
"bulk delete - collection table clean up (" + cPersister.getRole() + ")"
)
);
} }
String[] tableNames = targetedPersister.getConstraintOrderedTableNameClosure(); String[] tableNames = targetedPersister.getConstraintOrderedTableNameClosure();
@ -103,7 +125,7 @@ public class TableBasedDeleteHandlerImpl
// the difficulty is the ordering of the tables here vs the cascade attributes on the persisters -> // the difficulty is the ordering of the tables here vs the cascade attributes on the persisters ->
// the table info gotten here should really be self-contained (i.e., a class representation // the table info gotten here should really be self-contained (i.e., a class representation
// defining all the needed attributes), then we could then get an array of those // defining all the needed attributes), then we could then get an array of those
deletes.add( generateDelete( tableNames[i], columnNames[i], idSubselect, "bulk delete")); deletes.add( generateDelete( tableNames[i], columnNames[i], idSubselect, "bulk delete" ) );
} }
} }

View File

@ -23,16 +23,15 @@
*/ */
package org.hibernate.test.hql; package org.hibernate.test.hql;
import static org.junit.Assert.assertEquals; import java.sql.Connection;
import static org.junit.Assert.assertNotNull; import java.sql.ResultSet;
import static org.junit.Assert.assertTrue; import java.sql.SQLException;
import static org.junit.Assert.fail; import java.sql.Statement;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.TreeSet;
import junit.framework.AssertionFailedError;
import org.hibernate.QueryException; import org.hibernate.QueryException;
import org.hibernate.Session; import org.hibernate.Session;
@ -43,7 +42,9 @@ import org.hibernate.dialect.MySQLDialect;
import org.hibernate.exception.ConstraintViolationException; import org.hibernate.exception.ConstraintViolationException;
import org.hibernate.id.BulkInsertionCapableIdentifierGenerator; import org.hibernate.id.BulkInsertionCapableIdentifierGenerator;
import org.hibernate.id.IdentifierGenerator; import org.hibernate.id.IdentifierGenerator;
import org.hibernate.jdbc.Work;
import org.hibernate.persister.entity.EntityPersister; import org.hibernate.persister.entity.EntityPersister;
import org.hibernate.testing.DialectChecks; import org.hibernate.testing.DialectChecks;
import org.hibernate.testing.RequiresDialectFeature; import org.hibernate.testing.RequiresDialectFeature;
import org.hibernate.testing.SkipForDialect; import org.hibernate.testing.SkipForDialect;
@ -51,6 +52,12 @@ import org.hibernate.testing.SkipLog;
import org.hibernate.testing.TestForIssue; import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase; import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import org.junit.Test; import org.junit.Test;
import junit.framework.AssertionFailedError;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
/** /**
* Tests execution of bulk UPDATE/DELETE statements through the new AST parser. * Tests execution of bulk UPDATE/DELETE statements through the new AST parser.
@ -1413,6 +1420,144 @@ public class BulkManipulationTest extends BaseCoreFunctionalTestCase {
} }
} }
@Test
@TestForIssue( jiraKey = "HHH-9222" )
public void testBulkDeleteOfEntityWithElementCollection() {
// set up test data
{
Session s = openSession();
s.beginTransaction();
Farm farm = new Farm();
farm.setName( "Old McDonald Farm 'o the Earth" );
farm.setAccreditations( new HashSet<Farm.Accreditation>() );
farm.getAccreditations().add( Farm.Accreditation.ORGANIC );
farm.getAccreditations().add( Farm.Accreditation.SUSTAINABLE );
s.save( farm );
s.getTransaction().commit();
s.close();
}
// assertion that accreditations collection table got populated
{
Session s = openSession();
s.beginTransaction();
s.doWork(
new Work() {
@Override
public void execute(Connection connection) throws SQLException {
final Statement statement = connection.createStatement();
final ResultSet resultSet = statement.executeQuery( "select count(*) from farm_accreditations" );
assertTrue( resultSet.next() );
final int count = resultSet.getInt( 1 );
assertEquals( 2, count );
}
}
);
s.getTransaction().commit();
s.close();
}
// do delete
{
Session s = openSession();
s.beginTransaction();
s.createQuery( "delete Farm" ).executeUpdate();
s.getTransaction().commit();
s.close();
}
// assertion that accreditations collection table got cleaned up
// if they didn't, the delete should have caused a constraint error, but just to be sure...
{
Session s = openSession();
s.beginTransaction();
s.doWork(
new Work() {
@Override
public void execute(Connection connection) throws SQLException {
final Statement statement = connection.createStatement();
final ResultSet resultSet = statement.executeQuery( "select count(*) from farm_accreditations" );
assertTrue( resultSet.next() );
final int count = resultSet.getInt( 1 );
assertEquals( 0, count );
}
}
);
s.getTransaction().commit();
s.close();
}
}
@Test
@TestForIssue( jiraKey = "HHH-9222" )
public void testBulkDeleteOfMultiTableEntityWithElementCollection() {
// set up test data
{
Session s = openSession();
s.beginTransaction();
Human human = new Human();
human.setNickNames( new TreeSet() );
human.getNickNames().add( "Johnny" );
s.save( human );
s.getTransaction().commit();
s.close();
}
// assertion that nickname collection table got populated
{
Session s = openSession();
s.beginTransaction();
s.doWork(
new Work() {
@Override
public void execute(Connection connection) throws SQLException {
final Statement statement = connection.createStatement();
final ResultSet resultSet = statement.executeQuery( "select count(*) from human_nick_names" );
assertTrue( resultSet.next() );
final int count = resultSet.getInt( 1 );
assertEquals( 1, count );
}
}
);
s.getTransaction().commit();
s.close();
}
// do delete
{
Session s = openSession();
s.beginTransaction();
s.createQuery( "delete Human" ).executeUpdate();
s.getTransaction().commit();
s.close();
}
// assertion that nickname collection table got cleaned up
// if they didn't, the delete should have caused a constraint error, but just to be sure...
{
Session s = openSession();
s.beginTransaction();
s.doWork(
new Work() {
@Override
public void execute(Connection connection) throws SQLException {
final Statement statement = connection.createStatement();
final ResultSet resultSet = statement.executeQuery( "select count(*) from human_nick_names" );
assertTrue( resultSet.next() );
final int count = resultSet.getInt( 1 );
assertEquals( 0, count );
}
}
);
s.getTransaction().commit();
s.close();
}
}
private class TestData { private class TestData {
private Animal polliwog; private Animal polliwog;

View File

@ -21,9 +21,12 @@
package org.hibernate.test.hql; package org.hibernate.test.hql;
import java.util.List; import java.util.List;
import java.util.Set;
import javax.persistence.CascadeType; import javax.persistence.CascadeType;
import javax.persistence.CollectionTable;
import javax.persistence.ElementCollection;
import javax.persistence.Entity; import javax.persistence.Entity;
import javax.persistence.Enumerated;
import javax.persistence.GeneratedValue; import javax.persistence.GeneratedValue;
import javax.persistence.Id; import javax.persistence.Id;
import javax.persistence.ManyToMany; import javax.persistence.ManyToMany;
@ -42,6 +45,11 @@ public class Farm {
@ManyToMany(cascade = CascadeType.ALL) @ManyToMany(cascade = CascadeType.ALL)
private List<Crop> crops; private List<Crop> crops;
@ElementCollection
@Enumerated
@CollectionTable( name = "farm_accreditations" )
private Set<Accreditation> accreditations;
public long getId() { public long getId() {
return id; return id;
} }
@ -65,4 +73,18 @@ public class Farm {
public void setCrops(List<Crop> crops) { public void setCrops(List<Crop> crops) {
this.crops = crops; this.crops = crops;
} }
public Set<Accreditation> getAccreditations() {
return accreditations;
}
public void setAccreditations(Set<Accreditation> accreditations) {
this.accreditations = accreditations;
}
public static enum Accreditation {
ORGANIC,
SUSTAINABLE,
FARM_TO_TABLE
}
} }