Replaced the java.lang.reflect to the PersistenceEntity of Spring DATA

Author: AntonLGVS

Diff for: src/main/java/org/springframework/data/elasticsearch/core/convert/MappingElasticsearchConverter.java

@@ -50,16 +50,14 @@
 import org.springframework.data.mapping.MappingException;
 import org.springframework.data.mapping.PersistentPropertyAccessor;
 import org.springframework.data.mapping.PreferredConstructor;
+import org.springframework.data.mapping.SimplePropertyHandler;
 import org.springframework.data.mapping.context.MappingContext;
 import org.springframework.data.mapping.model.*;
 import org.springframework.data.util.ClassTypeInformation;
 import org.springframework.data.util.TypeInformation;
 import org.springframework.format.datetime.DateFormatterRegistrar;
 import org.springframework.lang.Nullable;
-import org.springframework.util.Assert;
-import org.springframework.util.ClassUtils;
-import org.springframework.util.CollectionUtils;
-import org.springframework.util.ObjectUtils;
+import org.springframework.util.*;
 
 /**
  * Elasticsearch specific {@link org.springframework.data.convert.EntityConverter} implementation based on domain type
@@ -354,7 +352,7 @@ private <R> R readEntity(ElasticsearchPersistentEntity<?> entity, Map<String, Ob
 
 			if (source instanceof SearchDocument) {
 				SearchDocument searchDocument = (SearchDocument) source;
-				populateScriptFields(result, searchDocument);
+				populateScriptFields(targetEntity, result, searchDocument);
 			}
 
 			return result;
@@ -555,27 +553,18 @@ private Object getPotentiallyConvertedSimpleRead(@Nullable Object value, @Nullab
 			return conversionService.convert(value, target);
 		}
 
-		private <T> void populateScriptFields(T result, SearchDocument searchDocument) {
+		private <T> void populateScriptFields(ElasticsearchPersistentEntity<?> entity, T result, SearchDocument searchDocument) {
 			Map<String, List<Object>> fields = searchDocument.getFields();
-			if (!fields.isEmpty()) {
-				for (java.lang.reflect.Field field : result.getClass().getDeclaredFields()) {
-					ScriptedField scriptedField = field.getAnnotation(ScriptedField.class);
-					if (scriptedField != null) {
-						String name = scriptedField.name().isEmpty() ? field.getName() : scriptedField.name();
-						Object value = searchDocument.getFieldValue(name);
-						if (value != null) {
-							field.setAccessible(true);
-							try {
-								field.set(result, value);
-							} catch (IllegalArgumentException e) {
-								throw new MappingException("failed to set scripted field: " + name + " with value: " + value, e);
-							} catch (IllegalAccessException e) {
-								throw new MappingException("failed to access scripted field: " + name, e);
-							}
-						}
-					}
+			entity.doWithProperties((SimplePropertyHandler) property -> {
+				if (property.isAnnotationPresent(ScriptedField.class) && fields.containsKey(property.getName())) {
+					ScriptedField scriptedField = property.findAnnotation(ScriptedField.class);
+					@SuppressWarnings({"constantCondition"})
+					String name = scriptedField.name().isEmpty() ? property.getName() : scriptedField.name();
+					Object value = searchDocument.getFieldValue(name);
+
+					entity.getPropertyAccessor(result).setProperty(property, value);
 				}
-			}
+			});
 		}
 
 		/**