伟大的Java 18都快面世了,而我们可能还对Java 8的核心Lambda用法还不熟悉,这篇文章就Stream技术分享几个开发过程中操作Map的使用小技巧,看完这篇文章,相信你一定会对Stream有一个全新的认识。

一、优雅地初始化一个Map

往下看之前,先想想如果不借助于Apache common等三方包,你会如何初始化一个有着初始值的Map,想好了继续往下看,代码大致是如下面的代码所示吧。

final HashMap<String, String> maps = new HashMap<>();
maps.put("key1", "value1");
maps.put("key2", "value2");
复制代码

以上的方式确实能做到初始化Map,但是最大的问题是我们可能需要换很多行去put键值对,显得很重复啰嗦,为了保持代码的整洁,发挥Java 8 的优势,我们其实可以像下面的代码使用stream去进行初始化。

final HashMap<String, String> maps = Stream.of(new Object[][]{
  {"key1", "value1"}})
    .collect(Collectors.toMap(data -> (String) data[0], data -> (String) data[1]));
复制代码

一行代码完成Map的初始化工作,是不是要显得优雅许多。如果你们用过jooqTuple,那么你会发现使用Tuple会更加优雅,能完成更多不可思议的事情。

二、优雅地List to Map

在日常开发过程中,后端的小伙伴为了构造特定的数据格式给前端,无法避免需要将List数据转化成Map数据,那么请问遇到这样的问题,你们会如何去完成这件事情呢? 一般来说,肯定是通过遍历的方式去完成数据的转化。但Java 8 Stream,我们能用它来做点有趣的事情。

假如我们有一个类Product,如下面的代码所示。

@Getter
@Setter
@ToString
@Builder
class Product{
    private Long id;
    private String category;
    private String name;
}
复制代码

我们现在获取到了List<Product>格式的数据。

static List<Product> getList(){
    final List<Product> productList = new ArrayList<>(100);
    for(int i =1;i<=100;i++){
        productList.add(Product.builder()
            .id((long) i)
            .name("name"+i)
            .category("category"+i%9)
            .build());
    }
    return productList;
}
复制代码

现在,我们要获取以id为key,name为value的Map格式数据,代码如下。

Map<Long, String> map = productList.stream().collect(Collectors.toMap(Product::getId, Product::getName));
复制代码

那如果要获取以id为key,product为value的Map格式数据,代码如下。

Map<Long, Product> map = productList.stream().collect(Collectors.toMap(Product::getId, data -> data));
复制代码

那如果要获取以category为key,name为value的Map格式数据,key可能会冲突,如何解决呢,需要传入key冲突解决的策略,代码如下。

#冲突发生时,保留先存在的数据,可按照业务自定义
Map<String, String> map = productList.stream().collect(Collectors.toMap(Product::getCategory, Product::getName, (existing, replacement) -> existing));
复制代码

那如果要最终获取线程安全的Map,如何解决呢,需要传入一个Supplier<M> mapSupplier,代码如下。

#这里引入的是ConcurrentHashMap,同样的也可以用TreeMap的实现
ConcurrentHashMap<String, String> map = productList.stream().collect(Collectors.toMap(Product::getCategory, Product::getName, (existing, replacement) -> existing, ConcurrentHashMap::new));
复制代码

总结

尽信书则不如无书,以上内容,纯属一家之言,因个人能力有限,难免有疏漏和错误之处,如发现bug或者有更好的建议,欢迎批评指正,不吝感激。

分享不易,感谢大家的阅读!



链接:https://juejin.cn/post/7034660287395397669