Polishing.

Reformat keys extraction code.

See spring-projects#2149

Signed-off-by: Artemiy Chereshnevvv <[email protected]>

Author: chrshnv

spring-data-jdbc/src/main/java/org/springframework/data/jdbc/repository/query/PartTreeJdbcQuery.java

@@ -44,6 +44,7 @@
 import org.springframework.data.domain.Window;
 import org.springframework.data.jdbc.core.JdbcAggregateOperations;
 import org.springframework.data.jdbc.core.convert.JdbcConverter;
+import org.springframework.data.mapping.PersistentPropertyAccessor;
 import org.springframework.data.relational.core.conversion.RelationalConverter;
 import org.springframework.data.relational.core.dialect.Dialect;
 import org.springframework.data.relational.core.mapping.RelationalMappingContext;
@@ -318,19 +319,17 @@ static class ScrollQueryExecution<T> implements JdbcQueryExecution<Window<T>> {
 				if (orders.isEmpty())
 					orders = sort.get().map(Sort.Order::getProperty).toList();
 
-				orders = orders.stream().map(it -> {
-					RelationalPersistentProperty prop = tableEntity.getPersistentProperty(it);
-
+				List<RelationalPersistentProperty> properties = new ArrayList<>();
+				for (String propertyName : orders) {
+					RelationalPersistentProperty prop = tableEntity.getPersistentProperty(propertyName);
 					if (prop == null)
-						return it;
-
-					return prop.getName();
-				}).toList();
+						continue;
 
-				keys = extractKeys(resultList, orders);
+					properties.add(prop);
+				}
 
-				Map<String, Object> finalKeys = keys;
-				positionFunction = (ignoredI) -> ScrollPosition.of(finalKeys, ((KeysetScrollPosition) position).getDirection());
+				final Map<String, Object> resultKeys = extractKeys(resultList, properties);
+				positionFunction = (ignoredI) -> ScrollPosition.of(resultKeys, ((KeysetScrollPosition) position).getDirection());
 			}
 
 			if (positionFunction == null)
@@ -347,26 +346,26 @@ else if (limit.isLimited())
 			return Window.from(resultList, positionFunction, hasNext);
 		}
 
-		private Map<String, Object> extractKeys(List<T> resultList, List<String> orders) {
+		private Map<String, Object> extractKeys(List<T> resultList, List<RelationalPersistentProperty> properties) {
 			if (resultList.isEmpty())
 				return Map.of();
 
+			Map<String, Object> result = new LinkedHashMap<>();
+
 			T last = resultList.get(resultList.size() - 1);
+			PersistentPropertyAccessor<T> accessor = tableEntity.getPropertyAccessor(last);
 
-			Field[] fields = last.getClass().getDeclaredFields();
+			for (RelationalPersistentProperty property : properties) {
+				String propertyName = property.getName();
+				Object propertyValue = accessor.getProperty(property);
 
-			// noinspection DataFlowIssue
-			return Arrays.stream(fields).filter(it -> {
-				String name = it.getName();
+				if (propertyValue == null)
+					continue;
 
-				RelationalPersistentProperty prop = tableEntity.getPersistentProperty(name);
-				if (prop != null)
-					name = prop.getName();
+				result.put(propertyName, propertyValue);
+			}
 
-				String finalName = name;
-				return orders.stream().anyMatch(order -> order.equalsIgnoreCase(finalName));
-			}).peek(ReflectionUtils::makeAccessible).collect(Collectors.toMap(Field::getName,
-					it -> ReflectionUtils.getField(it, last), (e1, e2) -> e1, LinkedHashMap::new));
+			return result;
 		}
 	}