Stream简单用法

今天药忘吃喽~ 2023-10-07 23:02 19阅读 0赞

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());

完美收工