在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]