從Java8開始典格,新增了一個java.util.stream
包耍缴,這個包下的類和接口用來處理集合中的元素,在這個包下面有一個Stream接口私恬,我們主要使用這個接口來對集合進行操作本鸣。
創(chuàng)建Stream
首先來看下創(chuàng)建Stream有哪幾種方式。
使用Stream自帶的靜態(tài)方法生成Stream對象荣德,常見的靜態(tài)方法有以下幾個:
- Stream.of(T)
- Stream.of(T... values)
- Stream.generate(Supplier)
- Stream.iterate(T, UnaryOperator)
- Stream.empty()
現(xiàn)在來看下每個靜態(tài)方法的作用
Stream.of(T) & Stream.of(T... values)
Stream.of是由兩個重載方法組成涮瞻,一個傳入單值,一個傳入數(shù)組
String[] arr = {"hello", "world"};
Stream streamArr = Stream.of(arr);
String str = "hello world";
Stream streamSingle = Stream.of(str);
Stream.generate & Stream.iterate
Stream.generate和Stream.iterate可以用來生成具有多個元素的Stream近顷,如果不加控制會一直生成下去宁否,一般配合limit(n)使用
先來看下Stream.iterate
Stream<Integer> stream5 = Stream.iterate(0, n-> n+1)
.limit(5);
stream5.forEach(System.out::println);
打印
0
1
2
3
4
Stream.iterate方法第一參數(shù)設定一個初始值,第二個參數(shù)表示基于這個初始值饱须,每次循環(huán)后返回一個新的值替換這個初始值台谊。limit(5)表示循環(huán)5次結束,最后Stream中包含了5個元素酪呻。
再來看下Stream.generate
Stream.generate方法只有一個Supplier參數(shù)狠角,意思是每次循環(huán)執(zhí)行Supplier接口方法返回一個新的值,放入到Stream中,由于Supplier是一個函數(shù)式接口屉凯,因此可以直接寫成Lambda表達式
AtomicInteger i = new AtomicInteger();
Stream.generate(()-> {
return i.getAndIncrement();
})
.limit(5)
.forEach(System.out::println);
上面的代碼同樣打印0~4眼溶。
除了Stream靜態(tài)方法之外堂飞,還可以使用Collection接口中的stream()方法來生成Stream對象。
Collection<String> list = Arrays.asList("hello", "world");
Stream streamList = list.stream();
同理绰筛,只要是Collection接口的子類或實現(xiàn)類都可以使用stream()方法铝噩。
操作Stream
Stream中的方法有很多,大致歸納如下表格所示:
方法 | 方法參數(shù) | 返回類型 | 描述 |
---|---|---|---|
filer | Predicate<T> | Stream<T> | 過濾數(shù)據(jù) |
distinct | 無 | Stream<T> | 去重 |
map | Function<T, R> | Stream<R> | 返回新的數(shù)據(jù) |
flatMap | Function<T, R> | Stream<T, Stream<R>> | 返回新的數(shù)據(jù)毛甲,并做扁平化處理 |
sort | Comparator<T> | Stream<T> | 對數(shù)據(jù)進行排序操作 |
limit | long | Stream<T> | 截取前幾條數(shù)據(jù) |
skip | long | Stream<T> | 跳過幾條數(shù)據(jù) |
anyMatch | Predicate<T> | boolean | 匹配任意一條數(shù)據(jù)具被,如果匹配到返回true |
noneMatch | Predicate<T> | boolean | 如果沒有匹配到數(shù)據(jù),返回true |
allMatch | Predicate<T> | boolean | 如果所有數(shù)據(jù)全部匹配到七咧,返回true |
findAny | 無 | Optional<T> | 返回任意一條數(shù)據(jù) |
findFirst | 無 | Optional<T> | 返回第一條數(shù)據(jù) |
count | 無 | long | 返回元素個數(shù) |
forEach | Consumer<T> | void | 遍歷元素叮叹,執(zhí)行Consumer |
collect | Collector<T, A, R> | R | 元素收集 |
reduce | BinaryOperator<T> | Optional<T> | 數(shù)據(jù)匯總 |
從方法的返回結果可以看出衬横,這些方法可以分為兩大類终蒂,一類是返回Stream對象,可以繼續(xù)對Stream操作拇泣,這類方法也被稱之為中間操作(Intermediate operations)
霉翔,另一類是返回非Stream,結束操作,這類方法也被稱之為中端操作(Terminal operations)
瀑凝,這兩類方法往往一起配合操作臭杰。
下面我們挑選其中的幾個方法來演示它們的作用。
filter
filter方法用來篩選出我們想要的數(shù)據(jù)寥枝,方法參數(shù)是一個Predicate接口磁奖,因為Predicate是一個函數(shù)式接口,我們可以使用Lambda表達式來寫冠跷。
Integer[] arr = { 1, 2, 3, 4, 5 };
long count = Stream.of(arr)
.filter(i -> i % 2 == 0)
.count();
System.out.println("偶數(shù)數(shù)量:" + count);
在這個例子中敢辩,我們篩選出了偶數(shù)數(shù)字,并且統(tǒng)計出偶數(shù)的數(shù)量盗冷。如果要打印每個偶數(shù)同廉,可以使用forEach方法
Stream.of(arr)
.filter(i -> i % 2 == 0)
.forEach(System.out::println);
打印:
2
4
如果要查找任意一個元素锅劝,可以使用findAny
int num = Stream.of(arr)
.filter(i -> i % 2 == 0)
.findAny()
.orElse(0);
System.out.println("findAny:" + num);
注意蟆湖,findAny()返回的是一個Optional對象,因為有可能沒有找到數(shù)據(jù)诬垂,因此需要開發(fā)者自己處理沒有找到數(shù)據(jù)的情況伦仍。同理findFirst
也是返回一個Optional對象。
distinct
distinct方法會對元素進行去重操作隧枫,類似于SQL中的SELECT distinct xx
Stream.of(1,1,2,3,3,4)
.distinct()
.forEach(System.out::println)
打印
1
2
3
4
sorted
使用sorted方法可以對元素進行排序操作
Stream.of(6,1,7,2,8,5)
.sorted()
.forEach(System.out::println);
打印
1
2
5
6
7
8
sorted()默認是從小到大排列,如果要從大到小降序协怒,可以使用.sorted(Comparator.reverseOrder())
Stream.of(6,1,7,2,8,5)
.sorted(Comparator.reverseOrder())
.forEach(System.out::println);
可以看到确买,sorted方法允許傳入一個比較器Comparator
讓開發(fā)者自己實現(xiàn)排序邏輯。下面是一個自定義Comparator例子:
@Data
@AllArgsConstructor
static class Goods {
private String goodsName;
private int price;
}
Stream.of(
new Goods("iphoneX", 4000)
, new Goods("mate30 pro", 5999)
, new Goods("redmek20", 2999)
)
.sorted((goods1, goods2) -> {
return Integer.compare(goods1.getPrice(), goods2.getPrice());
})
.forEach(System.out::println);
這個列子演示了按商品價格從低到高排序芭商。此處的sorted部分可以簡化為:.sorted(Comparator.comparing(Goods::getPrice))
map
map方法可以返回一個新的數(shù)據(jù)對象搀缠,組成一個新的Stream。
List<Goods> list = Arrays.asList(
new Goods("iphoneX", 4000)
, new Goods("mate30 pro", 5999)
, new Goods("redmek20", 2999)
);
list.stream()
.map(goods -> goods.getGoodsName())
.forEach(System.out::println);
上面的示例演示的是從原有的商品對象中拿到商品名稱簸州,然后組成一個新的List歧譬,其效果等同于
List<String> goodsNameList = new ArrayList<>(list.size());
for(Goods goods : list) {
goodsNameList.add(goods.getGoodsName());
}
map方法一般配合collect()方法一起使用
List<Goods> list = Arrays.asList(
new Goods("iphoneX", 4000)
, new Goods("mate30 pro", 5999)
, new Goods("redmek20", 2999)
);
List<String> nameList = list.stream()
.map(goods -> goods.getGoodsName())
.collect(Collectors.toList());
collect(Collectors.toList())
的意思是將Stream中的元素轉換成List
flatMap
flatMap()方法是map()方法的扁平化處理瑰步,與map不同的是,flatMap把返回Stream對象操作交給開發(fā)者自己處理读虏≡模看下面的例子:
Stream<String[]> stream = Stream.of("I am Java", "hello world")
.map(s -> s.split(" "));
這個例子的本意是想要將每個字符串進行拆分,把單詞單獨放入到Stream中揩徊,由于map返回的是一個字符串數(shù)組String[]
嵌赠,因此得到的Stream對象的泛型參數(shù)就是Stream<String[]>
,而不是Stream<String>
。
解決辦法是使用flatMap:
Stream<String> stream2 = Stream.of("I am Java", "hello world")
.flatMap(s -> Stream.of(s.split(" ")));
stream2.forEach(System.out::println);
打颖久铡:
I
am
Java
hello
world
綜合示例
下面來看一個綜合示例,演示的功能是:查詢商品名稱溜在,價格大于3000,按價格降序
public class StreamTest3 {
@Data
@AllArgsConstructor
static class Goods {
private String goodsName;
private int price;
}
public static void main(String[] args) {
List<Goods> list = Arrays.asList(
new Goods("iphoneX", 4000)
, new Goods("mate30 pro", 5999)
, new Goods("redmek20", 2999)
);
// 查詢商品名稱仆葡,價格大于3000志笼,按價格降序
List<String> nameList = list.stream()
.filter(goods -> goods.getPrice() > 3000)
.sorted(Comparator.comparing(Goods::getPrice).reversed())
.map(Goods::getGoodsName)
.collect(Collectors.toList());
System.out.println(nameList);
}
}
打尤依!:[mate30 pro, iphoneX]
代碼對應的SQL為:
SELECT goods_name FROM goods WHERE price > 3000 ORDER BY price DESC
小結
本篇講解了如何創(chuàng)建Stream以及Stream一些常用方法的使用方式,我們將會在下一篇著重講解collect()
和reduce()
的用法窖铡。
定期分享技術干貨坊谁,一起學習,一起進步口芍!