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配置文件這么寫，還愁不會整理日志？

 

原創不易，記得支持哦！！！

 

微信公眾號：@一個正經的程序員