Java8遍历Map的三种方式——for/stream/forEach

最近写在基于Spring WebFlux项目遇到一个需求,希望将请求中的cookie/headers/params等信息获取,而获取后的数据结构都是MultiValueMap的数据结构,实质上可以看做是Map>这种数据结构。而我需要将其转换。

for循环遍历

Show me the code first!以下是代码,解释一下逻辑,原来的cookies数据结构为Map>,其中HttpCookie为cookie键值对,由于业务需要,我们需要将其转换成Map才更方便处理,于是乎就有了以下代码。(我这里直接用了foreach循环,也可以用fori循环,例如for(int i = 0; i< xx; i++))

MultiValueMap cookies = request.getCookies(); // 从request中获取原始的cookie

Map cookieMap = new HashMap<>(); // 新建一个map,将cookie转入该map中

for (Map.Entry> itemList : cookies.entrySet()) { // 遍历原始的MultiValueMap

	for (HttpCookie item :itemList.getValue()) { // 遍历每个item中的List,其中的HttpCookie是我们需要的内容
		cookieMap.put(item.getName(), item.getValue()); // 存入内容
	}

}

stream流的方式处理

在Java8中,我们可以使用流,将Collections或者数组转化成Stream,并用链式的调用更加逻辑更加清晰。

MultiValueMap cookies = request.getCookies();
Map cookieMap = new HashMap<>();

cookies.entrySet() // 获取entrySet
	.stream() // 将其转化成流
	.map(Map.Entry>::getValue) // MultiValueMap -> List
	.flatMap(List::stream) // List -> HttpCookie
	.forEach(cookie -> cookieMap.put(cookie.getName(), cookie.getValue())); // 遍历,存入内容

Collection具有的forEach方法遍历

继续用Stream处理

我们可以看到通过流的方法处理cookie的方法,接下来,我们接着用相同的方法来处理请求参数,请求参数原本的数据格式依然为MultiValueMap,可以看做是Map>,其中请求参数名(key)对应的值(value)可能为多行,我们需要将其处理成一行。

MultiValueMap params = request.getQueryParams();
Map paramMap = new HashMap<>();

params.entrySet()
	.stream() // 将Set转换为Stream
	.forEach(entry ->
		paramMap.put(
			entry.getKey(), // 将参数名写入Key
			entry.getValue().stream().collect(Collectors.joining())) // 参数值多行合并成一行写入value
	);

大家可以看到,在处理参数值(value)的时候,值为List数据结构,以上代码通过entry.getValue().stream().collect(Collectors.joining()))将其List先转化为Stream,再用流的collection方法,将其合并。这个Collectors还具有将toSet/toList/groupingBy等功能,大家可以自行研究,这里就是使用的是joining合并方法。

存在优化点

写完后,我发现IntelliJ IDEA给我提示,显示我的代码‘不优雅’,还可以改进。IDE会对有改进空间的代码标黄,下图即为提示内容。

collection.stream.foreach

Java8遍历Map的三种方式——for/stream/forEach_第1张图片

图一,事实上在Java8中Collection可以直接使用foreach的方法,无需转成stream再使用foreach方法。

图二,Java8增强了String的方法,可以直接使用String.join合并List,第一个参数为连接字符串的字符,我这里用的是空格" ",第二个参数是待连接的字符串集合。

使用Collection的forEach方法遍历Map

修改后的代码如下:

params.forEach((key, value) -> paramMap.put(key, String.join(" ", value)));

这个时候我就想,为什么在处理例2(处理cookie的例子)的时候没有让我直接使用Collection.forEach?因为处理这个的例子相对复杂,使用了流的map/flatMap等方法。

总结

此时我们可以看到流的遍历Map和Collection.forEach遍历Map的区别(事实上Collection数据结构都可以使用以上方法):

  • 流的方法更加灵活,根据业务需要可以使用map/flatMap/filter/reduce等更复杂的操作。
  • collection.forEach相对简单,处理简单的逻辑,干净利落,不拖泥带水。

转载于:https://my.oschina.net/yangzijing/blog/3098737

你可能感兴趣的:(Java8遍历Map的三种方式——for/stream/forEach)