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

19,695 阅读3分钟

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

for循环遍历

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

MultiValueMap<String, HttpCookie> cookies = request.getCookies(); // 从request中获取原始的cookie

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

for (Map.Entry<String, List<HttpCookie>> itemList : cookies.entrySet()) { // 遍历原始的MultiValueMap

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

}

stream流的方式处理

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

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

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

Collection具有的forEach方法遍历

继续用Stream处理

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

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

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

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

存在优化点

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

collection.stream.foreach
collection.stream.foreach

String.join
String.join

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

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

使用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相对简单,处理简单的逻辑,干净利落,不拖泥带水。