HSEARCH-4142 Use projection id in real backend

hibernate · Mar 1, 2021 · fa5bc5c · fa5bc5c
1 parent fc2e471
commit fa5bc5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/...ch/integrationtest/mapper/orm/realbackend/routing/RoutingBridgeConditionalIndexingIT.java b/...ch/integrationtest/mapper/orm/realbackend/routing/RoutingBridgeConditionalIndexingIT.java
@@ -88,11 +88,10 @@ private List<Integer> searchAllIds() {
 		List<Integer> results = new ArrayList<>();
 		withinJPATransaction( entityManagerFactory, entityManager -> Search.session( entityManager )
 				.search( Book.class )
-				.select( f -> f.entityReference() )
+				.select( f -> f.id( Integer.class ) )
 				.where( f -> f.matchAll() )
 				.fetchAllHits()
 				.stream()
-				.map( ref -> (Integer) ref.id() )
 				.forEach( results::add ) );
 		return results;
 	}

diff --git a/...nate/search/integrationtest/mapper/orm/realbackend/routing/RoutingBridgeRoutingKeyIT.java b/...nate/search/integrationtest/mapper/orm/realbackend/routing/RoutingBridgeRoutingKeyIT.java
@@ -99,12 +99,11 @@ private List<Integer> searchIdsByRoutingKey(Genre genre) {
 		List<Integer> results = new ArrayList<>();
 		withinJPATransaction( entityManagerFactory, entityManager -> Search.session( entityManager )
 				.search( Book.class )
-				.select( f -> f.entityReference() )
+				.select( f -> f.id( Integer.class ) )
 				.where( f -> f.matchAll() )
 				.routing( routingKey )
 				.fetchAllHits()
 				.stream()
-				.map( ref -> (Integer) ref.id() )
 				.forEach( results::add ) );
 		return results;
 	}