使用 Java 8 Lambdas 和 Map 重构 If 语句_技术教程

本文介绍如何利用 Java 8 的 Lambda 表达式和 Map 数据结构，优雅地重构包含大量 if 语句的代码，特别是针对对象字段的非空校验场景。通过将校验逻辑与字段关联，可以简化代码结构，提高代码的可读性和可维护性，并方便后续扩展。

使用 Map 和 Lambda 重构 If 语句

在处理对象字段校验时，传统的 if 语句方式可能会导致代码冗长且难以维护。Java 8 引入的 Lambda 表达式和 Map 数据结构提供了一种更简洁、更灵活的解决方案。

核心思想: 将字段名与对应的校验函数（Lambda 表达式）存储在 Map 中，然后遍历 Map，对每个字段执行校验。

示例:

假设我们有一个 User 类，需要校验其 name、lastName、dni 和 vehicle 字段是否为空。

public class User {
    private String name;
    private String lastName;
    private String dni;
    private String vehicle;

    // Getters and setters
    public String getName() {
        return name;
    }

    public String getLastName() {
        return lastName;
    }

    public String getDni() {
        return dni;
    }

    public String getVehicle() {
        return vehicle;
    }

    public void setName(String name) {
        this.name = name;
    }

    public void setLastName(String lastName) {
        this.lastName = lastName;
    }

    public void setDni(String dni) {
        this.dni = dni;
    }

    public void setVehicle(String vehicle) {
        this.vehicle = vehicle;
    }
}

重构后的代码:

import java.util.Map;
import java.util.function.Function;

public class UserValidator {

    private static final Map> VALIDATIONS = Map.of(
            "name", User::getName,
            "lastName", User::getLastName,
            "dni", User::getDni,
            "vehicle", User::getVehicle
    );

    public void validateUserFields(User user) {
        VALIDATIONS.entrySet().stream()
                .filter(entry -> entry.getValue().apply(user) == null)
                .map(Map.Entry::getKey)
                .map(field -> String.format("The user's %s cannot be null", field))
                .map(RuntimeException::new)
                .findFirst()
                .ifPresent(e -> {
                    throw e;
                });
    }

    public static void main(String[] args) {
        User user = new User();
        user.setName("John");
        user.setLastName("Doe");
        // user.setDni("123456789"); // DNI is null
        user.setVehicle("Car");

        UserValidator validator = new UserValidator();
        try {
            validator.validateUserFields(user);
            System.out.println("User is valid.");
        } catch (RuntimeException e) {
            System.out.println("Validation failed: " + e.getMessage());
        }
    }
}

代码解释:

VALIDATIONS Map: 定义一个 Map>，其中 Key 是字段名（String），Value 是一个 Function，它接受一个 User 对象作为输入，并返回对应字段的值。 User::getName 等是方法引用，指向 User 类的 getter 方法。
validateUserFields 方法:
- VALIDATIONS.entrySet().stream(): 将 Map 转换为流，以便进行链式操作。
- .filter(entry -> entry.getValue().apply(user) == null): 过滤掉字段值为 null 的条目。 entry.getValue().apply(user) 执行与字段名关联的 Lambda 表达式，获取字段值。
- .map(Map.Entry::getKey): 将流中的 Map.Entry 对象转换为字段名（String）。
- .map(field -> String.format("The user's %s cannot be null", field)): 将字段名转换为异常消息。
- .map(RuntimeException::new): 将异常消息转换为 RuntimeException 对象。
- .findFirst(): 查找第一个异常。
- .ifPresent(e -> { throw e; }): 如果找到异常，则抛出该异常。

更简洁的版本:

import java.util.Map;
import java.util.function.Function;

public class UserValidator {

    private static final Map> VALIDATIONS = Map.of(
            "name", User::getName,
            "lastName", User::getLastName,
            "dni", User::getDni,
            "vehicle", User::getVehicle
    );

    public void validateUserFields(User user) {
        VALIDATIONS.entrySet().stream()
                .filter(entry -> entry.getValue().apply(user) == null)
                .findFirst()
                .ifPresent(e -> {
                    throw new RuntimeException("The user's " + e.getKey() + " cannot be null");
                });
    }
}

这个版本避免了多次 .map() 操作，直接在 .ifPresent() 中构建异常信息。

优点: