首页 > 编程语言 > Java8 使用流抽取List<T>集合中T的某个属性操作
2021
02-20

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

在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;
}

以上为个人经验,希望能给大家一个参考,也希望大家多多支持自学编程网。如有错误或未考虑完全的地方,望不吝赐教。

编程技巧