Merge pull request #38390 from abdullah-jaffer

* pr/38390:
  Replace Function<String, String> with UnaryOperator<String>

Closes gh-38390
This commit is contained in:
Phillip Webb
2023-11-20 20:07:52 -08:00

View File

@@ -22,8 +22,8 @@ import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.function.UnaryOperator;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
@@ -84,7 +84,7 @@ public final class CorrelationIdFormatter {
* @param resolver the resolver used to resolve named values
* @return a formatted correlation id
*/
public String format(Function<String, String> resolver) {
public String format(UnaryOperator<String> resolver) {
StringBuilder result = new StringBuilder();
formatTo(resolver, result);
return result.toString();
@@ -96,7 +96,7 @@ public final class CorrelationIdFormatter {
* @param resolver the resolver used to resolve named values
* @param appendable the appendable for the formatted correlation id
*/
public void formatTo(Function<String, String> resolver, Appendable appendable) {
public void formatTo(UnaryOperator<String> resolver, Appendable appendable) {
Predicate<Part> canResolve = (part) -> StringUtils.hasLength(resolver.apply(part.name()));
try {
if (this.parts.stream().anyMatch(canResolve)) {
@@ -169,7 +169,7 @@ public final class CorrelationIdFormatter {
private static final Pattern pattern = Pattern.compile("^(.+?)\\((\\d+)\\)$");
String resolve(Function<String, String> resolver) {
String resolve(UnaryOperator<String> resolver) {
String resolved = resolver.apply(name());
if (resolved == null) {
return blank();