最近写在基于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会对有改进空间的代码标黄,下图即为提示内容。
图一,事实上在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相对简单,处理简单的逻辑,干净利落,不拖泥带水。