天天看點

Java 8新特性:Stream API

Java 8新特性:全新的Stream API

Java 8新特性:Stream API

Java 8引入了全新的Stream API。這裡的

Stream

和I/O流不同,它更像具有

Iterable

的集合類,但行為和集合類又有所不同。

Stream API引入的目的在于彌補Java函數式程式設計的缺陷。對于很多支援函數式程式設計的語言,

map()

reduce()

基本上都内置到語言的标準庫中了,不過,Java 8的Stream API總體來講仍然是非常完善和強大,足以用很少的代碼完成許多複雜的功能。

建立一個

Stream

有很多方法,最簡單的方法是把一個

Collection

變成

Stream

。我們來看最基本的幾個操作:

public static void main(String[] args) {
    List<Integer> numbers = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
    Stream<Integer> stream = numbers.stream();
    stream.filter((x) -> {
        return x % 2 == 0;
    }).map((x) -> {
        return x * x;
    }).forEach(System.out::println);
}
           

集合類新增的

stream()

方法用于把一個集合變成

Stream

,然後,通過

filter()

map()

等實作

Stream

的變換。

Stream

還有一個

forEach()

來完成每個元素的疊代。

為什麼不在集合類實作這些操作,而是定義了全新的Stream API?Oracle官方給出了幾個重要原因:

一是集合類持有的所有元素都是存儲在記憶體中的,非常巨大的集合類會占用大量的記憶體,而Stream的元素卻是在通路的時候才被計算出來,這種“延遲計算”的特性有點類似Clojure的lazy-seq,占用記憶體很少。

二是集合類的疊代邏輯是調用者負責,通常是

for

循環,而Stream的疊代是隐含在對Stream的各種操作中,例如

map()

要了解“延遲計算”,不妨建立一個無窮大小的Stream。

如果要表示自然數集合,顯然用集合類是不可能實作的,因為自然數有無窮多個。但是Stream可以做到。

自然數集合的規則非常簡單,每個元素都是前一個元素的值+1,是以,自然數發生器用代碼實作如下:

class NaturalSupplier implements Supplier<Long> {

    long value = 0;

    public Long get() {
        this.value = this.value + 1;
        return this.value;
    }
}
           

反複調用

get()

,将得到一個無窮數列,利用這個

Supplier

,可以建立一個無窮的

Stream

public static void main(String[] args) {
    Stream<Long> natural = Stream.generate(new NaturalSupplier());
    natural.map((x) -> {
        return x * x;
    }).limit(10).forEach(System.out::println);
}
           

對這個

Stream

做任何

map()

filter()

等操作都是完全可以的,這說明Stream API對

Stream

進行轉換并生成一個新的

Stream

并非實時計算,而是做了延遲計算。

當然,對這個無窮的

Stream

不能直接調用

forEach()

,這樣會無限列印下去。但是我們可以利用

limit()

變換,把這個無窮

Stream

變換為有限的

Stream

利用Stream API,可以設計更加簡單的資料接口。例如,生成斐波那契數列,完全可以用一個無窮流表示(受限Java的long型大小,可以改為BigInteger):

class FibonacciSupplier implements Supplier<Long> {

    long a = 0;
    long b = 1;

    @Override
    public Long get() {
        long x = a + b;
        a = b;
        b = x;
        return a;
    }
}

public class FibonacciStream {

    public static void main(String[] args) {
        Stream<Long> fibonacci = Stream.generate(new FibonacciSupplier());
        fibonacci.limit(10).forEach(System.out::println);
    }
}
           

如果想取得數列的前10項,用

limit(10)

,如果想取得數列的第20~30項,用:

List<Long> list = fibonacci.skip(20).limit(10).collect(Collectors.toList());
           

最後通過

collect()

方法把

Stream

變為

List

。該

List

存儲的所有元素就已經是計算出的确定的元素了。

用Stream表示Fibonacci數列,其接口比任何其他接口定義都要來得簡單靈活并且高效。

計算π可以利用π的展開式:

π/4 = 1 - 1/3 + 1/5 - 1/7 + 1/9 - ...
           

把π表示為一個無窮Stream如下:

class PiSupplier implements Supplier<Double> {

    double sum = 0.0;
    double current = 1.0;
    boolean sign = true;

    @Override
    public Double get() {
        sum += (sign ? 4 : -4) / this.current;
        this.current = this.current + 2.0;
        this.sign = ! this.sign;
        return sum;
    }
}

Stream<Double> piStream = Stream.generate(new PiSupplier());
piStream.skip(100).limit(10)
        .forEach(System.out::println);
           

這個級數從100項開始可以把π的值精确到3.13~3.15之間:

3.1514934010709914
3.1317889675734545
3.1513011626954057
3.131977491197821
3.1511162471786824
3.1321589012071183
3.150938243930123
3.132333592767332
3.1507667724908344
3.1325019323081857
           

利用歐拉變換對級數進行加速,可以利用下面的公式:

Java 8新特性:Stream API

用代碼實作就是把一個流變成另一個流:

class EulerTransform implements Function<Double, Double> {

    double n1 = 0.0;
    double n2 = 0.0;
    double n3 = 0.0;

    @Override
    public Double apply(Double t) {
        n1 = n2;
        n2 = n3;
        n3 = t;
        if (n1 == 0.0) {
            return 0.0;
        }
        return calc();
    }

    double calc() {
        double d = n3 - n2;
        return n3 - d * d / (n1 - 2 * n2 + n3);
    }
}

Stream<Double> piStream2 = Stream.generate(new PiSupplier());
piStream2.map(new EulerTransform())
         .limit(10)
         .forEach(System.out::println);
           

可以在10項之内把π的值計算到3.141~3.142之間:

0.0
0.0
3.166666666666667
3.1333333333333337
3.1452380952380956
3.13968253968254
3.1427128427128435
3.1408813408813416
3.142071817071818
3.1412548236077655
           

還可以多次應用這個加速器:

Stream<Double> piStream3 = Stream.generate(new PiSupplier());
piStream3.map(new EulerTransform())
         .map(new EulerTransform())
         .map(new EulerTransform())
         .map(new EulerTransform())
         .map(new EulerTransform())
         .limit(20)
         .forEach(System.out::println);
           

20項之内可以計算出極其精确的值:

...
3.14159265359053
3.1415926535894667
3.141592653589949
3.141592653589719
           

可見用Stream API可以寫出多麼簡潔的代碼,用其他的模型也可以寫出來,但是代碼會非常複雜。