Don't migrate properties that cause a circular reference

Update `PropertiesMigrationReporter` so that properties are only
migrated automatically when they don't cause a circular reference.

Fixes gh-35919
pull/36048/head
Phillip Webb 1 year ago
parent 2f39ebfe89
commit efa072204a

@ -17,9 +17,11 @@
package org.springframework.boot.context.properties.migrator; package org.springframework.boot.context.properties.migrator;
import java.util.Collections; import java.util.Collections;
import java.util.HashSet;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set;
import java.util.function.Predicate; import java.util.function.Predicate;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -86,14 +88,26 @@ class PropertiesMigrationReporter {
if (renamed.isEmpty()) { if (renamed.isEmpty()) {
return null; return null;
} }
String target = "migrate-" + name; NameTrackingPropertySource nameTrackingPropertySource = new NameTrackingPropertySource();
Map<String, OriginTrackedValue> content = new LinkedHashMap<>(); this.environment.getPropertySources().addFirst(nameTrackingPropertySource);
for (PropertyMigration candidate : renamed) { try {
OriginTrackedValue value = OriginTrackedValue.of(candidate.getProperty().getValue(), String target = "migrate-" + name;
candidate.getProperty().getOrigin()); Map<String, OriginTrackedValue> content = new LinkedHashMap<>();
content.put(candidate.getNewPropertyName(), value); for (PropertyMigration candidate : renamed) {
String newPropertyName = candidate.getNewPropertyName();
Object value = candidate.getProperty().getValue();
if (nameTrackingPropertySource.isPlaceholderThatAccessesName(value, newPropertyName)) {
continue;
}
OriginTrackedValue originTrackedValue = OriginTrackedValue.of(value,
candidate.getProperty().getOrigin());
content.put(newPropertyName, originTrackedValue);
}
return new OriginTrackedMapPropertySource(target, content);
}
finally {
this.environment.getPropertySources().remove(nameTrackingPropertySource.getName());
} }
return new OriginTrackedMapPropertySource(target, content);
} }
private boolean isMapType(ConfigurationMetadataProperty property) { private boolean isMapType(ConfigurationMetadataProperty property) {
@ -172,4 +186,33 @@ class PropertiesMigrationReporter {
return source.getUnderlyingSource().toString(); return source.getUnderlyingSource().toString();
} }
/**
* {@link PropertySource} used to track accessed properties to protect against
* circular references.
*/
private class NameTrackingPropertySource extends PropertySource<Object> {
private final Set<String> accessedNames = new HashSet<>();
NameTrackingPropertySource() {
super(NameTrackingPropertySource.class.getName());
}
boolean isPlaceholderThatAccessesName(Object value, String name) {
if (value instanceof String) {
this.accessedNames.clear();
PropertiesMigrationReporter.this.environment.resolvePlaceholders((String) value);
return this.accessedNames.contains(name);
}
return false;
}
@Override
public Object getProperty(String name) {
this.accessedNames.add(name);
return null;
}
}
} }

@ -205,6 +205,14 @@ class PropertiesMigrationReporterTests {
.contains("Replacement: custom.the-map-replacement.key"); .contains("Replacement: custom.the-map-replacement.key");
} }
@Test // gh-35919
void directCircularReference() {
this.environment.getPropertySources()
.addFirst(new MapPropertySource("backcompat", Collections.singletonMap("wrong.two", "${test.two}")));
createAnalyzer(loadRepository("metadata/sample-metadata.json")).getReport();
assertThat(this.environment.getProperty("test.two")).isNull();
}
private List<String> mapToNames(PropertySources sources) { private List<String> mapToNames(PropertySources sources) {
List<String> names = new ArrayList<>(); List<String> names = new ArrayList<>();
for (PropertySource<?> source : sources) { for (PropertySource<?> source : sources) {

Loading…
Cancel
Save