Stream流的使用
場景及代碼案例源于實際項目
1. 需求描述
現存一list集合,其中儲存了投資人的資訊。
Person (name, age, money rateOfReturn)
其中的money表示投資人現有資産,rateOfReturn表示投資年回報率
需求:
1.計算投資人年齡之和;
2.根據現有資産及投資回報率,計算所有投資人一年後的資産總和;
2. 代碼
實體類
package com.drc.java8.stream.practice.domain;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.io.Serializable;
import java.math.BigDecimal;
/**
* @author dairuichuang
*/
@Data
@AllArgsConstructor
@NoArgsConstructor
public class Person {
private String name;
private Integer age;
private BigDecimal money;
private BigDecimal rateOfReturn;
}
主程式
package com.drc.java8.stream.practice;
import com.drc.java8.stream.practice.domain.Person2;
import org.junit.Test;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;
public class ListPersonPractice {
// 投資人集合
List<Person> PersonList = new ArrayList<>();
// 初始化投資人
{
PersonList.add(new Person2("張三", 11, BigDecimal.valueOf(10000), BigDecimal.valueOf(0.08)));
PersonList.add(new Person2("李四", 21, BigDecimal.valueOf(20000), BigDecimal.valueOf(0.18)));
PersonList.add(new Person2("王五", 31, BigDecimal.valueOf(30000), BigDecimal.valueOf(0.28)));
PersonList.add(new Person2("趙四", 41, BigDecimal.valueOf(40000), BigDecimal.valueOf(0.38)));
}
/**
* 計算
*/
@Test
public void testCalcAgeAndMoney() {
// 計算年齡之和
Integer ages = PersonList.stream().map(Person::getAge).reduce((a, b) -> (a + b)).get();
System.out.println(ages);
// money是資産,rateOfReturn 是資産年收益率,計算一年後幾人的資産總額
BigDecimal assetsAmt = PersonList.stream().map(person ->
person.getMoney().multiply(person.getRateOfReturn().add(BigDecimal.ONE))).reduce(BigDecimal::add).get();
System.out.println(assetsAmt);
}
}
3. 涉及方法
stream()
/*
* 傳回以此集合為源的順序流。
* 當spliterator()方法不能傳回IMMUTABLE、CONCURRENT或後期綁定的分裂器時,應該重寫此方法。(詳見spliterator())
*/
default Stream<E> stream() {
return StreamSupport.stream(spliterator(), false);
}
stream是Collection的一個預設方法,它的作用是傳回集合的stream流
用法:
List<String> list = new ArrayList<>();
Stream<String> stream = list.stream();
map(Function<? super T, ? extends R> mapper)
// 傳回一個流,由将給定函數應用于此流的元素的結果組成。
<R> Stream<R> map(Function<? super T, ? extends R> mapper);
Stream的方法,作用是傳回一個流,裡面的資料使我們定義的,比如:
其中
Person::getAge
定義的就是“年齡”,map方法會為我們傳回一個僅包含“年齡”的stream
我們可以接着調用collect方法,擷取到一個list集合。
reduce
// 使用關聯累加函數對此流的元素執行縮減,并傳回描述縮減值(如果有的話)的Optional。
Optional<T> reduce(BinaryOperator<T> accumulator);
其中的參數表示對兩個相同類型的操作數的操作,産生與操作數相同類型的結果。這是BiFunction的專門化,用于操作數和結果都是同一類型的情況。其中的參數:
@FunctionalInterface
public interface BinaryOperator<T> extends BiFunction<T,T,T>
簡單來說,reduce方法就是将stream流中的多個元素縮減為一個我們想要的結果。縮減的規則,就是我們傳入的表達式 ,比如
(a,b) -> (a + b)
表示将流中的元素進行累加操作,并傳回包含累加結果的Option,而
get()
方法,則是擷取Option中的累加結果。
通過一系列的操作,便獲得了集合中投資人年齡之和。Stream流還有很多其他的用法有待學習,用的熟練了,确實可以提升不少效率,并且對于一些疑難的問題它都可以很好的解決。