Java Stream mapToInt详解|极客教程

Java Stream mapToInt详解

在Java 8中，引入了Stream API，它为开发者提供了一种新的处理集合元素的方式。Stream API提供的操作可以简洁高效地处理集合元素，其中mapToInt是一个常用的操作之一。本文将对Java Stream中的mapToInt进行详细解释，并举例说明其使用方法及注意事项。

mapToInt方法介绍

在Java Stream中，mapToInt是一个中间操作，其作用是将Stream中的元素映射为IntStream。可以将Stream中的元素按照指定的转换函数转换为int类型，并返回一个新的IntStream。

mapToInt方法的语法如下所示：

IntStream mapToInt(ToIntFunction<? super T> mapper)

其中，mapper是一个ToIntFunction接口的实例，用于定义元素的转换规则。

使用示例

假设我们有一个存储员工薪水的集合，现在希望计算所有员工的薪水总和。可以借助mapToInt方法将员工薪水转换为int类型，然后使用sum方法计算总和。示例如下：

import java.util.Arrays;
import java.util.List;
import java.util.stream.IntStream;

public class MapToIntExample {
    public static void main(String[] args) {
        List<Employee> employees = Arrays.asList(
                new Employee("Alice", 5000),
                new Employee("Bob", 6000),
                new Employee("Cathy", 7000)
        );

        int totalSalary = employees.stream()
                .mapToInt(Employee::getSalary)
                .sum();

        System.out.println("Total salary of all employees: " + totalSalary);
    }

    static class Employee {
        private String name;
        private int salary;

        public Employee(String name, int salary) {
            this.name = name;
            this.salary = salary;
        }

        public int getSalary() {
            return salary;
        }
    }
}

在上面的示例中，首先定义了一个Employee类，包含姓名和薪水两个属性。然后创建了一个员工集合，并使用mapToInt方法将员工薪水映射为int类型，最后调用sum方法计算总和。

运行结果

当运行上述示例代码时，输出为：

Total salary of all employees: 18000

注意事项

在使用mapToInt方法时，需要注意以下几点：

返回类型为IntStream：mapToInt方法返回的是一个IntStream，因此需要使用IntStream类提供的方法进行进一步操作，如sum、average等。
避免空指针异常：在定义转换函数时，应注意处理可能为空的情况，避免空指针异常的发生。
适用性：mapToInt适用于将Stream中的元素映射为int类型，如果需要映射为其他类型，可以使用map方法或其他对应的mapToXXX方法。

通过本文的介绍，相信读者对Java Stream中的mapToInt方法有了更深入的了解。