您的位置:首页 > 其它

利用stream对map集合进行过滤

2018-07-29 20:26 609 查看

最近公司在大张旗鼓的进行代码审核,从中也发现自己写代码的不好习惯。一次无意的点到了公司封装的对map集合过滤的方法,发现了stream。于是研究了一下。并对原有的代码再次结合Optional进行重构下


原有方法说明

主要处理过滤条件Map对象,过滤掉了null和空字符串 等操作



这里我简单画了个图,对这个过程总结



1.利用filter对null和空字符串进行过滤

2.利用map进行对于Stream中包含的元素使用给定的转换函数进行转换操作

3.collect进行聚合成map

由于公司的代码不合适自己展示,我这里自己仿照公司的写了个类似的简单方法,然后一步一步优化

自定义map工具类处理方法

//这里的代码和原先的代码相比做了修改,去掉了map进行对于Stream中包含的元素使用给定的转换函数进行转换操作

public static Map<String, Object> parseMapForFilter(Map<String, Object> map) {
if (map == null) {
return null;
} else {
map = map.entrySet().stream()
.filter((e) -> checkValue(e.getValue()))
.collect(Collectors.toMap(
(e) -> (String) e.getKey(),
(e) -> e.getValue()
));
}
return map;
}

private static boolean checkValue(Object object) {

if (object instanceof String && "".equals(object)) {
return false;
}

if (null == object) {
return false;
}

return true;

}

测试下

public static void main(String[] args) {
Map<String,Object> params = new HashMap<>(16);

params.put("a","");
params.put("b",null);
params.put("c","c");

params = MapUtil.parseMapForFilter(params);
System.out.println(params);
System.out.println(MapUtil.parseMapForFilter(null));
}

输出结果

{c=c}
null

优化parseMapForFilter 方法,加入Optional类

public static Map<String, Object> parseMapForFilterByOptional(Map<String, Object> map) {

return Optional.ofNullable(map).map(
(v) -> {
Map params = v.entrySet().stream()
.filter((e) -> checkValue(e.getValue()))
.collect(Collectors.toMap(
(e) -> (String) e.getKey(),
(e) -> e.getValue()
));

return params;
}
).orElse(null);
}

是不是感觉更清晰了呢?

测试一下

public static void main(String[] args) {
Map<String, Object> params = new HashMap<>(16);

params.put("a", "");
params.put("b", null);
params.put("c", "c");

params = MapUtil.parseMapForFilterByOptional(params);

System.out.println(params);
System.out.println(MapUtil.parseMapForFilterByOptional(null));
}

结果

{c=c}
null

优化checkValue方法

private static boolean checkValueByOptional(Object object) {
return (Boolean) Optional.ofNullable(object)
.filter((e) -> e instanceof String && e.equals("") ? false : true)
.orElse(false);
}


总结

大家是不是感觉lambda 写法更加简单明了,不再充满着if判断。但如果大家首写的时候,肯定感觉不习惯,我刚开始写的时候,也是很别扭,根本不知道怎么写。一点点去尝试。更多细节基础的问题这里不再叙述。

只是感觉这个对map的处理还是很好的。并且实际工作中用到的地方比较多。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: