将 Java 8 流映射函数作为参数传递

Question

我有一个以逗号分隔的字符串，我想将其转换为数组。但是在某些情况下，我需要整数解析，有时是双精度解析。有没有一种方法可以传递 mapToDouble 或 mapToInt 而不是再次编写整个内容。

return Arrays.stream(test.split(",")).mapToDouble(x -> {
        if (StringUtils.isEmpty(x)) {
            return condition ? -1 : 0;
        }
        return Double.parseDouble(x);
}).toArray();

return Arrays.stream(test.split(",")).mapToInt(x -> {
        if (StringUtils.isEmpty(x)) {
            return condition ? -1 : 0;
        }
        return Integer.parseInt(x);
}).toArray();

有没有办法把它变成一个函数，在那里我可以有一个通用函数并存储适当的数组？

Answer 1

您可以接受一个简单的功能做String和Function<String, T>将使用此功能每串元素转换。好消息是，这个函数可以返回任何你想要的类型：Integer，Double，BigDecimal，String或者你想要的任何其他类型。在下面的示例中，我使用了一个方法引用，如：

• Integer::valueOf将元素转换为Integer值
• Double::valueOf将元素转换为Double值
• String::valueOf将元素转换为String值

考虑以下示例：

import java.util.Arrays;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;

public class ParsingStringTest {

    public static void main(String[] args) {
        String str = "1, , 3, 4, 5, , 7, sasd, aaa, 0";

        List<Double> doubles = parse(str, Double::valueOf);
        List<Integer> integers = parse(str, Integer::valueOf);
        List<String> strings = parse(str, String::valueOf);

        System.out.println(doubles);
        System.out.println(integers);
        System.out.println(strings);

        Double[] array = doubles.toArray(new Double[doubles.size()]);

        System.out.println(Arrays.toString(array));
    }

    public static <T> List<T> parse(String str, Function<String, T> parseFunction) {
        return Arrays.stream(str.split(","))
                .filter(s -> !s.isEmpty())
                .map(s -> {
                    try {
                        return parseFunction.apply(s.trim());
                    } catch (Exception e) {}
                    return null;
                })
                .filter(Objects::nonNull)
                .collect(Collectors.toList());
    }
}

以下示例的控制台输出是：

[1.0, 3.0, 4.0, 5.0, 7.0, 0.0]
[1, 3, 4, 5, 7, 0]
[1, , 3, 4, 5, , 7, sasd, aaa, 0]
[1.0, 3.0, 4.0, 5.0, 7.0, 0.0]

我希望它有帮助。