stream流的使用-获取list集合中对象的单个字段list集合,进行累加操作

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 mapper)

// 返回一个流,由将给定函数应用于此流的元素的结果组成。
<R> Stream<R> map(Function<? super T, ? extends R> mapper);

Stream的方法,作用是返回一个流,里面的数据使我们定义的,比如:

stream.map(Person::getAge);

其中 Person::getAge定义的就是“年龄”,map方法会为我们返回一个仅包含“年龄”的stream
我们可以接着调用collect方法,获取到一个list集合。

List<Integer> ageList = stream.map(Person::getAge).collect(Collectors.toList());

reduce

// 使用关联累加函数对此流的元素执行缩减,并返回描述缩减值(如果有的话)的Optional。
Optional<T> reduce(BinaryOperator<T> accumulator);

其中的参数表示对两个相同类型的操作数的操作,产生与操作数相同类型的结果。这是BiFunction的专门化,用于操作数和结果都是同一类型的情况。其中的参数:

@FunctionalInterface
public interface BinaryOperator<T> extends BiFunction<T,T,T>

简单来说,reduce方法就是将stream流中的多个元素缩减为一个我们想要的结果。缩减的规则,就是我们传入的表达式 ,比如

PersonList.stream().map(Person::getAge).reduce((a, b) -> (a + b)).get();

(a,b) -> (a + b) 表示将流中的元素进行累加操作,并返回包含累加结果的Option,而get()方法,则是获取Option中的累加结果。

通过一系列的操作,便获得了集合中投资人年龄之和。Stream流还有很多其他的用法有待学习,用的熟练了,确实可以提升不少效率,并且对于一些疑难的问题它都可以很好的解决。

你可能感兴趣的:(lambda,stream,方法引用,Java,java,lambda,Stream流,stream,java8)