java 8如何自定义收集器（collector）详解

更新时间：2017年06月28日 10:25:01 作者：chauncy

这篇文章主要给大家介绍了关于java 8如何自定义收集器（collector）的相关资料，文中介绍的非常详细，对大家具有一定的参考学习价值，需要的朋友们下面来一起看看吧。

需求：

将一个容器List<Bean> 按照一定的字段进行分组，分组过后的值为特定的BEAN 里面的属性例如：

假定有这样一个Bean

 public class SubjectOberser{
  private String subjectKey;
  private AbstractObserver abstractObserver;
  ...geter seter 方法...
 }

我们需要按照 subjectKey 进行分组，分组过后的内容应该为这样一个容器Map<String,List<AbstractObserver>>

map 中的key,为SubjectOberser 属性的subjectKey,值为List<AbstractObserver>

实现过程

首先来看看collector 的接口定义

 public interface Collector<T, A, R> { 
   Supplier<A> supplier();
   BiConsumer<A, T> accumulator();
   Function<A, R> finisher();
   BinaryOperator<A> combiner();
   Set<Characteristics> characteristics();
 }

类型 T ,是在容器里面元素的类型

类型 A ,是accumulator 返回的类型，即是累加器的返回类型

类型 R ,是最终结果的类型

supplier 方法返回的结果必须为一个空的Supplier,也就是一个空的无参函数(签名就是这样的 (）->{}),在调用的时候它会创建一个空的累加器(accumulator)实例，供数据收集的时候使用，很明显如果按照我们的需求试下你自己collector 这里应该返回一个 () -> new HashMap<>() ，一个Map 来收集结果

accumulator 方法返回归约操作的函数（签名是这样的（a,b）->void ），当遍历到流中第n个元素时，这个函数执行时会有两个参数：保存归约结果的累加器(已收集了流中的前n-1个项目),还有第n个元素本身。签名也展示该函数是void，因为该操作是在原来的容器里面进行更新的，所以返回的是void 类型。按照需求的中的实现应该是是这样的:

 public BiConsumer<Map<String, List<AbstractObserver>>, SubjectObserver> accumulator() {
  return (Map<String, List<AbstractObserver>> acc, SubjectObserver v) -> {
   if (acc.containsKey(v.getSubjectKey())){
    acc.get(v.getSubjectKey()).add(v.getAbstractObserver());
   }else{
    List<AbstractObserver> l = new ArrayList<>();
    l.add(v.getAbstractObserver());
    acc.put(v.getSubjectKey(),l);
   }
  };
 }

这里的逻辑就是if else 逻辑判断就是，这个key ,在map 中是否存在，如果不存在，那么我们需要给他new一个list 的实例，不然我的的数据没有地方存储

finisher 可从名字看出方法累积过程的最后要调用的一个函数，以便将累加器对象转换为整个集合操作的最终结果。通常来说累加器的类型也是返回的结果的类型，那么就返回identity 就可以了，如果不是的话，就行自行转换了。在当前需求的情况下我们的累加器和返回结果的类型是一致的，所以这里的实现是这样的:

 public Function<Map<String, List<AbstractObserver>>,
     Map<String, List<AbstractObserver>>> finisher(){
   return Function.identity();
 }

combiner 方法是将两个累加的结果进行一个合并的过程，当然这个过程并不是每一个collector 都会调用得到(后面会讲到)
按照我们的需求，只需要将两个累加器的，中间结果合并成为一个结果即可，所以是现实这样的:

  public BinaryOperator<Map<String, List<AbstractObserver>>> combiner() {
  return ((Map<String, List<AbstractObserver>> map1, 
    Map<String, List<AbstractObserver>> map2) -> {
     map1.putAll(map2);
     return map1;
   });
 }

characteristics 该方法返回一个 Characteristics 的集合，它有如下值可选

UNORDERED—— 归约结果不受流中项目的遍历和累积顺序的影响。

CONCURRENT—— accumulator函数可以从多个线程同时调用,且该收集器可以并行执行。如果收集器没有标为UNORDERED，那它仅在用于用于无序数据源时才可以并行归约。

IDENTITY_ FINISH—— 这表明完成器方法返回的函数是一个不改变的函数，这种情况下，累加器对象将会直接用作合并过程的最终结果。

   public Set<Characteristics> characteristics() {
    return Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH));
   }

最终collector 代码合在一起就是:

 public class MyCollector implements Collector<SubjectObserver,
  Map<String, List<AbstractObserver>>,
  Map<String, List<AbstractObserver>>> {


  @Override
  public Supplier<Map<String, List<AbstractObserver>>> supplier() {
   return () -> new HashMap<>();
  }

  @Override
  public BiConsumer<Map<String, List<AbstractObserver>>, SubjectObserver> accumulator() {
   return (Map<String, List<AbstractObserver>> acc, SubjectObserver v) -> {
    if (acc.containsKey(v.getSubjectKey())) {
     acc.get(v.getSubjectKey()).add(v.getAbstractObserver());
    } else {
     List<AbstractObserver> l = new ArrayList<>();
     l.add(v.getAbstractObserver());
     acc.put(v.getSubjectKey(), l);
    }
   };
  }

  @Override
  public BinaryOperator<Map<String, List<AbstractObserver>>> combiner() {
   return ((Map<String, List<AbstractObserver>> map1, Map<String, List<AbstractObserver>> map2) -> {
    map1.putAll(map2);
    return map1;
   });
  }

  @Override
  public Function<Map<String, List<AbstractObserver>>, Map<String, List<AbstractObserver>>> finisher() {
   return Function.identity();
  }

  @Override
  public Set<Characteristics> characteristics() {
   return Collections.unmodifiableSet(EnumSet.of(Characteristics.IDENTITY_FINISH));
  }
 }

调用的过程就是:

 public static Map<String, List<AbstractObserver>> initObjectMap() {
 ClassScaner classScaner = new ClassScaner();
 Set<Class> set = classScaner.doScan("com.souche.datacenter.observer");
 return set
   .stream()
   .filter(aClass -> SubjectAnnotationResolver.getAnnotationSubjectName(aClass) != null)
   .map(aClass -> {
    String subjectKey = SubjectAnnotationResolver.getAnnotationSubjectName(aClass);
    AbstractObserver abstractObserver = getBeanByClassName(aClass.getSimpleName());
    return new SubjectObserver(subjectKey, abstractObserver);
   }).collect(new MyCollector());
 }

直接在使用的地方直接new MyCollector 就可以了

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流，谢谢大家对脚本之家的支持。

您可能感兴趣的文章:

maven依赖的version声明控制方式
这篇文章主要介绍了maven依赖的version声明控制方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-01-01
Java流程控制语句之If选择结构
今天继续带大家复习Java流程控制语句的相关知识,本文对If选择结构作了非常详细的介绍及代码示例,对正在学习的小伙伴们很有帮助,需要的朋友可以参考下
2021-06-06
Spring Boot2.3 新特性分层JAR的使用
这篇文章主要介绍了Spring Boot2.3 新特性分层JAR的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-06-06
聊聊java中引用数据类型有哪些
这篇文章主要介绍了java中引用数据类型有哪些，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2021-10-10
JDK13的新特性之AppCDS详解
AppCDS的全称是Application Class-Data Sharing。主要是用来在不同的JVM中共享Class-Data信息，从而提升应用程序的启动速度。这篇文章主要介绍了JDK13的新特性:AppCDS详解,需要的朋友可以参考下
2020-05-05
解决mapper接口无法映射mapper.xml的问题
这篇文章主要介绍了解决mapper接口无法映射mapper.xml的问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-06-06
详解J2EE开发的网站部署到阿里云服务器的方法
这篇文章主要介绍了详解J2EE开发的网站部署到阿里云服务器的方法,需要的朋友可以参考下
2018-01-01
详解Java多线程编程中LockSupport类的线程阻塞用法
LockSupport类提供了park()和unpark()两个方法来实现线程的阻塞和唤醒,下面我们就来详解Java多线程编程中LockSupport类的线程阻塞用法:
2016-07-07
springboot项目整合注册功能模块开发实战
这篇文章主要介绍了springboot项目整合注册功能模块开发实战,在用户的注册是首先需要查询当前的用户名是否存在，如果存在则不能进行注册，相当于一个查询语句，本文通过实例代码详细讲解，需要的朋友可以参考下
2022-11-11
Java 数据结构与算法系列精讲之背包问题
背包问题是一个非常典型的考察动态规划应用的题目，对其加上不同的限制和条件，可以衍生出诸多变种，若要全面理解动态规划，就必须对背包问题了如指掌
2022-02-02