Java 8 Streams - 收集可能为null的值

时间:2018-02-15 04:01:03

标签: java java-8 hashmap java-stream keyvaluepair

我有以下代码:

    private static <T> Map<String, ?> getDifference(final T a, final T b, final Map<String, Function<T, Object>> fields) {
    return fields.entrySet().stream()
            .map(e -> {
                final String name = e.getKey();
                final Function<T, Object> getter = e.getValue();
                final Object pairKey = getter.apply(a);
                final Object pairValue = getter.apply(b);
                if (Objects.equals(pairKey, pairValue)) {
                    return null;
                } else {
                    return Pair.of(name, pairValue);
                }
            })
            .filter(Objects::nonNull)
            .collect(Collectors.toMap(Pair::getKey, Pair::getValue));
    }

现在,pairValue可以为null。为了避免描述here中的NPE,在收集&#34;时,我希望确保只发送那些非空的值。如果为null,我想发送&#34;&#34;。

所以,我尝试用这个替换最后一行:

.collect(Collectors.toMap(Pair::getKey,Optional.ofNullable(Pair::getValue).orElse(""));

及其他修改:

.collect(Collectors.toMap(pair -> pair.getKey(), Optional.ofNullable(pair -> pair.getValue()).orElse(""));

不编译。我不确定这里需要什么。有什么帮助吗?

2 个答案:

答案 0 :(得分:5)

您只需收集HashMap即可获得null值,而无需Optional

private static <T> Map<String, Object> getDifference(
        final T a, final T b, final Map<String, Function<T, Object>> fields) {
    return fields.entrySet().stream()
        .map(e -> {
            final Function<T, Object> getter = e.getValue();
            final Object value = getter.apply(b);
            return Objects.equals(getter.apply(a),value)? null: Pair.of(e.getKey(), value);
        })
        .filter(Objects::nonNull)
        .collect(HashMap::new, (m,p) -> m.put(p.getKey(),p.getValue()), Map::putAll);
}

顺便说一句,不鼓励在返回类型中使用通配符,它​​们可以使得来电者的生活变得不必要而且没有任何好处。

为了比较,这里没有Stream的相同操作:

private static <T> Map<String, Object> getDifference(
        final T a, final T b, final Map<String, Function<T, Object>> fields) {
    HashMap<String, Object> result = new HashMap<>();
    fields.forEach((key, getter) -> {
        final Object value = getter.apply(b);
        if(!Objects.equals(getter.apply(a), value)) result.put(key, value);
    });
    return result;
}

当然,这也适用于可选:

private static <T> Map<String, Optional<Object>> getDifference(
        final T a, final T b, final Map<String, Function<T, Object>> fields) {
    HashMap<String, Optional<Object>> result = new HashMap<>();
    fields.forEach((key, getter) -> {
        final Object value = getter.apply(b);
        if(!Objects.equals(getter.apply(a), value))
            result.put(key, Optional.ofNullable(value));
    });
    return result;
}

但如果你想要做的只是用空字符串替换null,则不需要Optional

private static <T> Map<String, Object> getDifference(
        final T a, final T b, final Map<String, Function<T, Object>> fields) {
    HashMap<String, Object> result = new HashMap<>();
    fields.forEach((key,getter) -> {
            final Object value = getter.apply(b);
            if(!Objects.equals(getter.apply(a), value))
                result.put(key, value==null? "": value);
        });
    return result;
}

并且,如果您只是在map函数而不是收集器中执行此替换,那么此替换也可以与原始代码一起使用:

private static <T> Map<String, ?> getDifference(final T a, final T b, final Map<String, Function<T, Object>> fields) {
    return fields.entrySet().stream()
        .map(e -> {
            final String name = e.getKey();
            final Function<T, Object> getter = e.getValue();
            final Object pairKey = getter.apply(a);
            final Object pairValue = getter.apply(b);
            if (Objects.equals(pairKey, pairValue)) {
                return null;
            } else {
                return Pair.of(name, pairValue==null? "": pairValue);
            }
        })
        .filter(Objects::nonNull)
        .collect(Collectors.toMap(Pair::getKey, Pair::getValue));
}

private static <T> Map<String, Object> getDifference(
        final T a, final T b, final Map<String, Function<T, Object>> fields) {
    return fields.entrySet().stream()
        .map(e -> {
            final Function<T, Object> getter = e.getValue();
            final Object pairValue = getter.apply(b);
            return Objects.equals(getter.apply(a), pairValue)? null:
                Pair.of(e.getKey(), pairValue==null? "": pairValue);
        })
        .filter(Objects::nonNull)
        .collect(Collectors.toMap(Pair::getKey, Pair::getValue));
}

答案 1 :(得分:3)

语法不正确。 toMap()的第二个参数必须是lambda,所以

.collect(Collectors.toMap(
             pair -> pair.getKey(),
             pair -> Optional.ofNullable(pair.getValue()).orElse("")
));

您可以按如下方式修改map()部分

return Pair.of(name, Optional.ofNullable(pairValue).orElse(""));

并使用原始collect()