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

一个正经的程序员

发布于 2022-04-11 09:12:01

7.3K0

发布于 2022-04-11 09:12:01

文章被收录于专栏：一个正经的程序员

前言

在需求开发中，我们需要对一个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;
  }