diff --git a/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/TableSet.java b/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/TableSet.java new file mode 100644 index 0000000000..284eacce05 --- /dev/null +++ b/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/TableSet.java @@ -0,0 +1,85 @@ +/* + * 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.persister.entity.mutation; + +import java.util.Arrays; +import java.util.BitSet; + +import org.hibernate.sql.model.MutationTarget; +import org.hibernate.sql.model.TableMapping; + +/** + * Represents a Set of TableMapping(s); table mappings are + * identified by an ordered unique id: the order in which + * they are updated within the scope of a particular persister. + * This makes it possible to store a set of them as a bitset, + * which is typically more efficient than using a {@link java.util.Set}. + * These table ids relate to the use in method {@link MutationTarget#forEachMutableTable} + * and {@link MutationTarget#forEachMutableTableReverse}. + *
N.B. Make sure to not store TableMappings from different + * persisters, as their unique identifiers will overlap: + * we'll only verify a mismatch if assertions are enabled.
+ */ +public final class TableSet { + + private BitSet bits; + private Object[] checks; //Meant for assertions only + + public void add(final TableMapping tableMapping) { + if ( bits == null ) { + bits = new BitSet(); + } + assert addForChecks( tableMapping ); + bits.set( tableMapping.getRelativePosition() ); + } + + public boolean isEmpty() { + return bits == null; + } + + public boolean contains(final TableMapping tableMapping) { + assert matchRead( tableMapping ); + return bits != null && bits.get( tableMapping.getRelativePosition() ); + } + + //Meant for assertions only + private boolean matchRead(final TableMapping tableMapping) { + if ( bits != null ) { + final int index = tableMapping.getRelativePosition(); + if ( bits.get( index ) ) { + return checks[index] == tableMapping; + } + } + return true; //to make the assertion happy + } + + //Meant for assertions only + private boolean addForChecks(final TableMapping tableMapping) { + final int position = tableMapping.getRelativePosition(); + ensureCapacity( position ); + if ( checks[position] != null ) { + //pre-existing in the set: verify it's the same one. + if ( checks[position] != tableMapping ) { + return false;//fail the assertion + } + } + checks[position] = tableMapping; + return true; //to make the assertion happy + } + + //Meant for assertions only + private void ensureCapacity(final int position) { + final int increments = 3; //Needs to be at least 1. + if ( checks == null ) { + checks = new Object[position + increments]; + } + else if ( checks.length <= position ) { + checks = Arrays.copyOf( checks, position + increments ); + } + } + +} diff --git a/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java b/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java index 541f2a1422..3f0c11c309 100644 --- a/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java +++ b/hibernate-core/src/main/java/org/hibernate/persister/entity/mutation/UpdateCoordinatorStandard.java @@ -1274,10 +1274,10 @@ public class UpdateCoordinatorStandard extends AbstractMutationCoordinator imple private final int[] dirtyAttributeIndexes; private final InclusionChecker dirtinessChecker; - private final Set