时间：2022-11-22 10:57:36 | 栏目：JAVA代码 | 点击：次
在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
}
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;
}