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

时间:2017-07-29 02:40:16

标签: java-8 java-stream

我有一个以逗号分隔的字符串,我希望将其转换为数组。但是在某些情况下,我需要整数解析,有时需要加倍。 有没有办法可以传递mapToDouble或mapToInt,而不是再写一遍。

lapply

有没有办法将它变成一个函数,在那里我可以有一个泛型函数并存储适当的数组?

2 个答案:

答案 0 :(得分:4)

您可以使用一个接受movzxString的简单函数来完成此操作,该函数将使用此函数转换每个字符串元素。好消息是此函数可能会返回您想要的任何类型:Function<String, T>IntegerDoubleBigDecimal或您想要的任何其他类型。在下面的示例中,我使用方法引用,如:

  • 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 :(得分:3)

有一种方法,但你不会再使用原语了,因为Java中的泛型不支持原始类型。相反,您可以使用IntegerDouble包装类型:

public static <T> T[] convert(
        String test, 
        Function<String, T> parser,
        boolean condition, 
        T ifConditionTrue,
        T ifConditionFalse,
        IntFunction<T[]> arrayGenerator) {

    return Arrays.stream(test.split(","))
        .map(x -> {
            if (StringUtils.isEmpty(x)) {
                return condition ? ifConditionTrue : ifConditionFalse;
            }
            return parser.apply(x);
        })
        .toArray(arrayGenerator);
}

此方法可按如下方式使用:

Integer[] ints = convert("1, ,3", Integer::parseInt, true, -1, 0, Integer[]::new);