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

来自:互联网
时间:2021-01-09
阅读:

前言

需要对一个List中的对象进行唯一值属性去重,属性求和,对象假设为BillsNums,有id、nums、sums三个属性,其中id表示唯一值,需要nums与sums进行求和,并最后保持一份。
例如说:(“s1”, 1, 1),(“s1”,2,3),(“s2”,4,4), 求和并去重的话,就是(“s1”, 3, 4),(“s2”,4,4)

对象与属性

class BillsNums {
 private String id;
 private int nums;
 private int sums;
 
 public String getId() {
  return id;
 }
 public void setId(String id) {
  this.id = id;
 }
 public int getNums() {
  return nums;
 }
 public void setNums(int nums) {
  this.nums = nums;
 }
 public int getSums() {
  return sums;
 }
 public void setSums(int sums) {
  this.sums = sums;
 }
}

数据

public static void main(String[] args) {
 
 List<BillsNums> billsNumsList = new ArrayList<>();
 BillsNums billsNums = new BillsNums();
 billsNums.setId("1001");
 billsNums.setNums(2);
 billsNums.setSums(100);
 billsNumsList.add(billsNums);
 
 BillsNums billsNums2 = new BillsNums();
 billsNums2.setId("1001");
 billsNums2.setNums(3);
 billsNums2.setSums(100);
 billsNumsList.add(billsNums2);
 
 List<BillsNums> result = merge(billsNumsList);
 System.out.println("result:" + JSON.toJSONString(result, true));
}

合并

法一:list做循环求和,去重

/**
 * 将id进行合并nums, sums 相加道回合并后的集合使用Java8的流进行处理
 */
public static List<BillsNums> merge(List<BillsNums> list) {
  Map<String, BillsNums> map = new HashMap<>();
  list.stream()
   .forEach(billsNums -> {
    BillsNums last = map.get(billsNums.getId());
    if(null != last){
     billsNums.setSums(billsNums.getSums() + last.getSums());
     billsNums.setNums(billsNums.getNums() + last.getNums());
     map.put(billsNums.getId(), billsNums );
    }else{
     map.put(billsNums.getId(), billsNums);
    }
   });
  return map.values().stream().collect(Collectors.toList());
  
}

法二:法一的升级版

/**
 * 将id进行合并nums, sums 相加道回合并后的集合使用Java8的流进行处理
 */
public static List<BillsNums> merge(List<BillsNums> list) {
  List<BillsNums> result = list.stream()
  // 表示id为key, 接着如果有重复的,那么从BillsNums对象o1与o2中筛选出一个,这里选择o1,
  // 并把id重复,需要将nums和sums与o1进行合并的o2, 赋值给o1,最后返回o1
  .collect(Collectors.toMap(BillsNums::getId, a -> a, (o1,o2)-> {
   o1.setNums(o1.getNums() + o2.getNums());
   o1.setSums(o1.getSums() + o2.getSums());
   return o1;
  })).values().stream().collect(Collectors.toList());
  return result ;
}

结果

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

返回顶部
顶部