在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 userList = new ArrayList<>();
User user = new User(1L, "zhangsan");
User user1 = new User(2L, "lisi");
userList.add(user);
userList.add(user1);
List userIds = new ArrayList<>();
//第一种方式
userList.forEach(user2 -> {
userIds.add(user2.getId());
//这里可以进行更多的操作
});
System.out.println(userIds);
//第二种方式
List 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 findElementByPropertyValue(List 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;
}
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。