欢迎来到【一个正经的程序员】的博客园,笔者会经常分享一些大家在工作中经常用到的技术,欢迎大家围观!如果小伙伴们有需要讲解的技术,可以给笔者发消息或邮件留言!【louis.liu.oneself@hotmail.com】

Java8使用Stream实现List中对象属性的合并(去重并求和)

前言

在需求开发中,我们需要对一个List中的对象进行唯一值属性去重,属性求和,对象假设为Pool,有name、value两个属性,其中name表示唯一值,需要value进行求和,并最后保持一份对象。

例如:("A", 1),("A", 2),("B", 5), 求和并去重的话,就是("A", 3),("B", 5)。

 

对象与属性

@Data
@NoArgsConstructor
@AllArgsConstructor
public class Pool {
    private String name;
    private int value;
}

 

注:这里的@Data、@NoArgsConstructor、@AllArgsConstructor是使用的Lombok注解,更多关于Lombok详解,请看Lombok使用指南

 

数据与测试

public static void main(String[] args) throws Exception {
    List<Pool> list = new ArrayList<Pool>(){
      {
        add(new Pool("A", 1));
        add(new Pool("A", 2));
        add(new Pool("A", 3));
        add(new Pool("B", 4));
        add(new Pool("B", 5));
      }
    };

    List<Pool> result = merge(list);
    System.err.println(JSONObject.toJSONString(result));
  }

 

输出结果

[
  {
    "name": "A",
    "value": 6
  },
  {
    "name": "B",
    "value": 9
  }
]

 

方法一

/**
   * @Description 使用Java8的流进行处理,将name相同的对象进行合并,将value属性求和
   * @Title merge
   * @Param [list]
   * @Return java.util.List<Pool>
   * @Author Louis
   */
public static List<Pool> merge(List<Pool> list) {
    Map<String, Pool> map = new HashMap<String, Pool>();
    list.stream().forEach(pool -> {
      Pool last = map.get(pool.getName());
      if(null != last){
        pool.setValue(pool.getValue() + last.getValue());
      }
      map.put(pool.getName(), pool);
    });
    return map.values().stream().collect(Collectors.toList());
  }

 

方法二

/**
   * @Description 使用Java8的流进行处理,将name相同的对象进行合并,将value属性求和
   * @Title merge
   * @Param [list]
   * @Return java.util.List<Pool>
   * @Author Louis
   */
  public static List<Pool> merge(List<Pool> list) {
    List<Pool> result = list.stream()
        // 表示name为key,接着如果有重复的,那么从Pool对象o1与o2中筛选出一个,这里选择o1,
        // 并把name重复,需要将value与o1进行合并的o2, 赋值给o1,最后返回o1
        .collect(Collectors.toMap(Pool::getName, a -> a, (o1,o2)-> {
          o1.setValue(o1.getValue() + o2.getValue());
          return o1;
        })).values().stream().collect(Collectors.toList());
    return result;
  }

 

总结

使用Java8的流进行处理,将name相同的对象进行合并,将value属性求和,这里推荐第二种方法,既简单更符合Java8的处理。

 

往期推荐

面试官:String长度有限制吗?是多少?还好我看过!

Node起源及名称由来

Lombok使用指南

Java Logback配置文件这么写,还愁不会整理日志?

 

原创不易,记得支持哦!!!

 

微信公众号:@一个正经的程序员

 

posted @ 2021-05-18 18:17  一个正经的程序员!  阅读(17169)  评论(0编辑  收藏  举报
……