Call the value adapter during NamedContributorsMapAdapter construction

Closes gh-29640

spring-boot-project/spring-boot-actuator/src/main/java/org/springframework/boot/actuate/health/NamedContributorsMapAdapter.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2021 the original author or authors.
+ * Copyright 2012-2022 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -31,23 +31,29 @@
  * @param <V> the value type
  * @param <C> the contributor type
  * @author Phillip Webb
+ * @author Guirong Hu
  * @see CompositeHealthContributorMapAdapter
  * @see CompositeReactiveHealthContributorMapAdapter
  */
 abstract class NamedContributorsMapAdapter<V, C> implements NamedContributors<C> {
 
-	private final Map<String, V> map;
-
-	private final Function<V, ? extends C> valueAdapter;
+	private final Map<String, C> namedContributorsMap;
 
 	NamedContributorsMapAdapter(Map<String, V> map, Function<V, ? extends C> valueAdapter) {
 		Assert.notNull(map, "Map must not be null");
 		Assert.notNull(valueAdapter, "ValueAdapter must not be null");
-		map.keySet().forEach(this::validateKey);
-		map.values().stream().map(valueAdapter)
-				.forEach((value) -> Assert.notNull(value, "Map must not contain null values"));
-		this.map = Collections.unmodifiableMap(new LinkedHashMap<>(map));
-		this.valueAdapter = valueAdapter;
+		this.namedContributorsMap = getContributorsMap(map, valueAdapter);
+	}
+
+	private Map<String, C> getContributorsMap(Map<String, V> map, Function<V, ? extends C> valueAdapter) {
+		Map<String, C> contributorsMap = new LinkedHashMap<>(map.size());
+		map.forEach((name, value) -> {
+			this.validateKey(name);
+			C contributor = adapt(value, valueAdapter);
+			Assert.notNull(contributor, "Map must not contain null values");
+			contributorsMap.put(name, contributor);
+		});
+		return Collections.unmodifiableMap(contributorsMap);
 	}
 
 	private void validateKey(String value) {
@@ -58,7 +64,7 @@ private void validateKey(String value) {
 
 	@Override
 	public Iterator<NamedContributor<C>> iterator() {
-		Iterator<Entry<String, V>> iterator = this.map.entrySet().iterator();
+		Iterator<Entry<String, C>> iterator = this.namedContributorsMap.entrySet().iterator();
 		return new Iterator<NamedContributor<C>>() {
 
 			@Override
@@ -68,20 +74,20 @@ public boolean hasNext() {
 
 			@Override
 			public NamedContributor<C> next() {
-				Entry<String, V> entry = iterator.next();
-				return NamedContributor.of(entry.getKey(), adapt(entry.getValue()));
+				Entry<String, C> entry = iterator.next();
+				return NamedContributor.of(entry.getKey(), entry.getValue());
 			}
 
 		};
 	}
 
 	@Override
 	public C getContributor(String name) {
-		return adapt(this.map.get(name));
+		return this.namedContributorsMap.get(name);
 	}
 
-	private C adapt(V value) {
-		return (value != null) ? this.valueAdapter.apply(value) : null;
+	private C adapt(V value, Function<V, ? extends C> valueAdapter) {
+		return (value != null) ? valueAdapter.apply(value) : null;
 	}
 
 }

spring-boot-project/spring-boot-actuator/src/test/java/org/springframework/boot/actuate/health/NamedContributorsMapAdapterTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2021 the original author or authors.
+ * Copyright 2012-2022 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -20,6 +20,7 @@
 import java.util.Iterator;
 import java.util.LinkedHashMap;
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Function;
 
 import org.junit.jupiter.api.Test;
@@ -92,6 +93,22 @@ void getContributorReturnsAdaptedEntry() {
 		assertThat(adapter.getContributor("two")).isEqualTo("owt");
 	}
 
+	@Test
+	void eachValueAdapterShouldBeCalledOnlyOnce() {
+		Map<String, String> map = new LinkedHashMap<>();
+		map.put("one", "one");
+		map.put("two", "two");
+		int callCount = map.size();
+
+		AtomicInteger counter = new AtomicInteger(0);
+		TestNamedContributorsMapAdapter<String> adapter = new TestNamedContributorsMapAdapter<>(map,
+				(name) -> count(name, counter));
+		assertThat(adapter.getContributor("one")).isEqualTo("eno");
+		assertThat(counter.get()).isEqualTo(callCount);
+		assertThat(adapter.getContributor("two")).isEqualTo("owt");
+		assertThat(counter.get()).isEqualTo(callCount);
+	}
+
 	@Test
 	void getContributorWhenNotInMapReturnsNull() {
 		TestNamedContributorsMapAdapter<String> adapter = createAdapter();
@@ -106,6 +123,11 @@ private TestNamedContributorsMapAdapter<String> createAdapter() {
 		return adapter;
 	}
 
+	private String count(CharSequence charSequence, AtomicInteger counter) {
+		counter.set(counter.get() + 1);
+		return reverse(charSequence);
+	}
+
 	private String reverse(CharSequence charSequence) {
 		return new StringBuilder(charSequence).reverse().toString();
 	}