想象一下如果以前的Java写法针对List与Map容器类相互转换,特定属性的排重,集合元素排序,过滤筛选,删除元素,奇偶性判断,返回限制的个数……之类的操作,第一步就是需要写for循环,把每个元素遍历出来,再做自己逻辑操作。
这里抛开以前在for循环里面写代码的固化思维,减少一点for,简化代码,如何写得更加简洁?
首先创建今天的主角类:Zoo,类属性很简单:id和name
public class Zoo {
private int id;
private String name;
public Zoo(int s, String a) {
this.id = s;
this.name = a;
}
@Override
public String toString() {
return "Zoo{" +
"id=" + id +
", name='" + name + '\'' +
'}';
}
public int getId() {
return id;
}
}
在一个main 方法或者随便哪里创一个Zoo类型的List,填充几条数据进去,下面就针对这个list和这几条数据进行操作
ArrayList<Zoo> list = new ArrayList<>();
list.add(new Zoo(3,"C"));
list.add(new Zoo(1,"A"));
list.add(new Zoo(2,"B"));
list.add(new Zoo(2,"D"));
(1)获取list中的最大值,最小值,平均值
忘掉你那暴力的for循环里写sum+=sum,下面几行代码即可:
// 获取最大值
int max = list.stream().mapToInt(Zoo::getId).max().getAsInt();
// 获取最小值
int min = list.stream().mapToInt(Zoo::getId).min().getAsInt();
// 获取平均值
Double avg = list.stream().mapToInt(Zoo::getId).average().getAsDouble();
查看结果:
(2)根据id大小进行排序
// 根据id大小进行排序
Collections.sort(list, (o1,o2) -> o1.getId() - o2.getId());
// 或者
Collections.sort(list, Comparator.comparingInt(Zoo::getId));
// 循环输出
list.forEach(o -> System.out.println(o.toString()));
(3)筛选出ID为奇数的记录
List<Zoo> result = list.stream().filter(o -> o.getId() % 2 != 0).collect(Collectors.toList());
result.forEach(o -> System.out.println(o.toString()));
可以看见id为2的记录直接就没了
(4)提取所有的Zoo的ID保存到一个List中
List<Integer> idList = list.stream().map(Zoo::getId).collect(Collectors.toList());
System.out.println(idList.toString());
(5)针对ID去重(两个id=2,只保留一个)
针对某个属性去重,首先要在Zoo类中重写hashCode方法,指定是针对哪个属性做哈希,还要重写equal方法:
这里因为我只需要针对id做操作,所以只需要勾选id
改造后的Zoo:
public class Zoo {
private int id;
private String name;
public Zoo(int s, String a) {
this.id = s;
this.name = a;
}
@Override
public String toString() {
return "Zoo{" +
"id=" + id +
", name='" + name + '\'' +
'}';
}
public int getId() {
return id;
}
/** 如果要根据id查重,就必须重写hashCode和equals方法 **/
@Override
public int hashCode(){
return Objects.hash(id);
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Zoo zoo = (Zoo) o;
return id == zoo.id;
}
}
stream流式写法:
// 针对ID去重
List<Zoo> distinct = list.stream().distinct().collect(Collectors.toList());
// 输出
distinct.forEach(o -> System.out.println(o.toString()));
(6)限制集合中元素的返回个数(顺序)
只会返回集合中第N个元素,这个N,就是你可以限制的返回个数。
// 限制返回个数
List<Zoo> limits = list.stream().limit(1).collect(Collectors.toList());
limits.forEach(o-> System.out.println(o.toString()));
(7)跳过N个元素并生成新list
// N = 1
List<Zoo> skip = list.stream().skip(1).collect(Collectors.toList());
skip.forEach(o-> System.out.println(o.toString()));
可以看到id=1的记录在生成新List的时候被跳过去了
(8)删除指定元素
这个很常见,我就是想删除id=3的元素:
// 删除元素
list.removeIf(o -> o.getId() == 3);
System.out.println(list.toString());
可以看到操作完之后的list中id=3的zoo就被删掉了
(9)List转Map
这个也是很常见的,假设要求Map的key是List里面的Zoo的id(注意这里id必须唯一,有重复的话转换会报错主键冲突)
// List转Map(key必须唯一)
Map<Integer, Zoo> zooMap = distinct.stream().collect(Collectors.toMap(Zoo::getId, o -> o));
System.out.println(zooMap.toString());
转换成功
(10)它们可以嵌套起来使用,例如,针对List,我想查出查重后的id是偶数的跳过第N条后限制返回第一条,并且把返回结果转换成以id为Key的Map。
Map<Integer, Zoo> collect = list.stream()
.distinct()
.filter(o -> o.getId() % 2 == 0)
.skip(0)
.limit(1)
.collect(Collectors.toMap(Zoo::getId, o -> o));
System.out.println(collect.toString());
我们分析一下:
第一步查重,肯定只剩下1 A,2 B ,3 C ; 另外一个重复的2 D被排除掉了。
第二步过滤,1,3为奇数,被过滤掉剩下2
第三步跳跃,skip(0)跳过0条,还是2
第四步限制,limit(1)限制返回1条,还是2
第五步list转map,并且用id作为key, 那返回的map就是 {2, {2,‘B’}}
看下结果对不对: