Java8 使用流抽取List<T>集合中T的某個屬性操作

Posted on by

在java開發中，我們經常會遇到需要從一個集合中，抽取集合中元素的某一個屬性。在java8之前，我們通常采用for循環去獲取，但java8之後我們有瞭一種新的辦法，那就是stream。

話不多說，直接上代碼

import lombok.AllArgsConstructor;
import lombok.Data;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
/**
 * @author lanfangyi
 * @version 1.0
 * @since 2019/5/12 13:22
 */
@Data
@AllArgsConstructor
public class User {
  private Long id;
  private String name;
}
class TestMainService {
  public static void main(String[] args) {
    List<User> userList = new ArrayList<>();
    User user = new User(1L, "zhangsan");
    User user1 = new User(2L, "lisi");
    userList.add(user);
    userList.add(user1);
    List<Long> userIds = new ArrayList<>();
    //第一種方式
    userList.forEach(user2 -> {
      userIds.add(user2.getId());
      //這裡可以進行更多的操作
    });
    System.out.println(userIds);
    //第二種方式
    List<Long> userIds2 = userList.stream().map(User::getId).collect(Collectors.toList());
    System.out.println(userIds2);
  }
}

打印結果：

[1, 2]
[1, 2]

補充：Java根據類屬性值從一個集合中找到和該屬性相等的對象

方法

使用common-utils包提供的CollectionUtils和BeanPropertyValueEqualsPredicate

比如找id屬性值為9587的用戶

Object obj = CollectionUtils.find(UserList.get(), 
new BeanPropertyValueEqualsPredicate("id", "9587"));
if(obj == null){
  log.info("not found");
}else{
  //do your thing
}

find方法實現的大概思路

public static <T> T findElementByPropertyValue(List<T> list, String propertyName, Object value) throws Exception {
  if(list == null || list.isEmpty()) {
    return null;
  }
  PropertyDescriptor pd = new PropertyDescriptor(propertyName, list.get(0).getClass());
  for (T t : list) {
    Object obj = pd.getReadMethod().invoke(t);
    if(StringUtils.equals(String.valueOf(value), String.valueOf(obj))) {
      return t;
    }
  }
  return null;
}

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：