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

2022-04-11 09:12:01 浏览数 (2)

前言

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

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

对象与属性

代码语言:javascript复制
@Data
@NoArgsConstructor
@AllArgsConstructor
public class Pool {
    private String name;
    private int value;
}

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

数据与测试

代码语言:javascript复制
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));
  }

输出结果

代码语言:javascript复制
[
  {
    "name": "A",
    "value": 6
  },
  {
    "name": "B",
    "value": 9
  }
]

方法一

代码语言:javascript复制
/**
   * @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());
  }

方法二

代码语言:javascript复制
/**
   * @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的处理。

0 人点赞