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 list = new ArrayList(){
      {
        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 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
   * @Author Louis
   */
public static List merge(List list) {
    Map map = new HashMap();
    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
   * @Author Louis
   */
  public static List merge(List list) {
    List 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配置文件这么写,还愁不会整理日志?

 

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

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