Stream简单用法

Stream简单用法

1. 快速创建list
2. 取对象的某一列
3. 过滤，或者说是根据一个判断条件筛选出目标对象
4. 分组
5. 求和
6. Map、List互转
   ==========================================================================

1.快速创建list

比如我有个实体类User，User有个属性Name

public class User {
    public User(String name, String age, int height) {
        this.name = name;
        this.age = age;
        this.height = height;
    }
    private String name;
    private String age;
    private int height;
    // setter、getter方法我就不写了
}
 
// 创建三个user
User user1 = new User("111", "18", 180);
User user2 = new User("222", "18", 175);
User user3 = new User("333", "19", 170);

现在我要创建3个user放到list中

1.第一种方法
List<User> userList = new ArrayList<>();
userList.add(user1);
userList.add(user2);
userList.add(user3);
1.第二种方法
// stream流，创建的是动态数组，可以添加元素
List<User> userList = Stream.of(user1, user2, user3).collect(Collectors.toList());

如果创建一个固定长度的list，可以使用Arrays.asList(…args)直接返回一个list

// 本质是将一个数组转成list，数组的大小是固定的，所以此list不能添加元素
// 如果调用add方法增加新的元素，会报异常：java.lang.UnsupportedOperationException
List<String> s = Arrays.asList("1","2","3")

2. 取对象的某一列

拿上面userList举例，我取出list中所有user的name属性放到一个新的list中：

1.第一种方法
// 遍历
List<String> userNameList = new ArrayList<>();
for (User user : userList) {
    userNameList.add(user.getName());
}
2.第二种方法
List<String> userNameList = userList.stream().map(User::getName).collect(Collectors.toList());

3、过滤，或者说是根据一个判断条件筛选出目标对象

还拿上面的userList，比如我要筛选出userList中name不为空的user

1.第一种方法
List<User> newUserList = new ArrayList<>();
// if判断
for (User user : userList) {
    if(user.getName() != null) {
        newUserList.add(user);
    }
}
2.第二种方法
// 获取userName不为空的user的List
List<User> userList = userList.stream().filter(user-> user.getName() != null).collect(Collectors.toList());

4、分组

把userList中的user根据年龄分组：

1.第一种方法
Map<String, List<User>> map = new HashMap<>();
// if判断
for (User user : userList) {
    if (map.get(user.getAge()) == null) {
        map.put(user.getAge(), new ArrayList());
    }
    map.get(user.getAge()).add(user);
}
2.第二种方法
Map<String, List<User>> map =userList.stream().collect( Collectors.groupingBy(User::getAge, Collectors.toList()));

5、求和

求和的普通遍历方式跟上面差不多，就不举例了；

// int、double、long:
double max = userList.stream().mapToDouble(User::getHeight).sum();

6、Map、List互转

1、list转map

1.第一种方法
Map<String, User> userMap = new Map<>();
for (User user : userList) {
    userMap.put(user.getName(), user);
}
2.第二种方法
注意：
（1）转换map，key重复问题；
代码中使用(key1,key2)->key2表达式可以解决此类问题，如果出现重复的key就使用key2覆盖前面的key1,也可以定义成(key1,key2)->key1，保留key1,根据自己的业务场景来调整。
（2）空指针异常，即转为map的value是null。这个可以用filter过滤；
Map<String, User> userMap= userList.stream().collect(Collectors.toMap(User::getName, Function.identity(),(key1, key2)->key2));

2、map转list

1.第一种方法
List<User> userList = new List<>();
for (String userName : userMap.keySet()) {
    userList.add(userMap.get(userName));
}
2.第二种方法
List<User> userList = userMap.entrySet().stream().map(e ->e.getValue()).collect(Collectors.toList());

										  完美收工