Set comparator on Persistent SortedSet and SortedMap collection insta…

…nces
hibernate · Mar 12, 2021 · 71f633a · 71f633a
1 parent 4292d11
commit 71f633a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/...nate-core/src/main/java/org/hibernate/collection/internal/StandardSortedMapSemantics.java b/...nate-core/src/main/java/org/hibernate/collection/internal/StandardSortedMapSemantics.java
@@ -49,7 +49,9 @@ public PersistentCollection<V> instantiateWrapper(
 			Object key,
 			CollectionPersister collectionDescriptor,
 			SharedSessionContractImplementor session) {
-		return new PersistentSortedMap<>( session );
+		final PersistentSortedMap<K,V> result = new PersistentSortedMap<>( session );
+		result.setComparator( (Comparator<K>) collectionDescriptor.getSortingComparator() );
+		return result;
 	}
 
 	@Override

diff --git a/...nate-core/src/main/java/org/hibernate/collection/internal/StandardSortedSetSemantics.java b/...nate-core/src/main/java/org/hibernate/collection/internal/StandardSortedSetSemantics.java
@@ -50,7 +50,9 @@ public PersistentCollection<E> instantiateWrapper(
 			Object key,
 			CollectionPersister collectionDescriptor,
 			SharedSessionContractImplementor session) {
-		return new PersistentSortedSet<>( session );
+		final PersistentSortedSet<E> result = new PersistentSortedSet<>( session );
+		result.setComparator( (Comparator<E>) collectionDescriptor.getSortingComparator() );
+		return result;
 	}
 
 	@Override