java怎么在一个方法里返回一个function对象

原创

mob649e8159b30b 2024-04-13 03:33:37 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8159b30b的原创作品，请联系作者获取转载授权，否则将追究法律责任

在Java中如何在一个方法里返回一个Function对象

在Java编程中，Function对象是一个函数式接口，可以用于表示接受一个参数并产生一个结果的函数。有时候我们需要在一个方法中返回一个Function对象，以便在其他地方使用。本文将介绍如何在Java中实现这一功能，并通过一个实际问题来展示其应用。

实际问题

假设我们有一个需求：根据不同的用户输入，返回不同的操作。我们可以通过返回一个Function对象来实现这一功能。具体而言，我们需要实现一个方法getOperationByUserInput，该方法接受一个用户输入的字符串，返回一个对应的操作。

解决方案

我们可以定义一个Function接口的实现类，用于表示不同的操作。然后在getOperationByUserInput方法中根据用户输入返回相应的操作。以下是具体的实现：

import java.util.function.Function;

public class OperationFunc {

    public static Function<String, String> addOperation = input -> "Add " + input;
    public static Function<String, String> subtractOperation = input -> "Subtract " + input;
    public static Function<String, String> multiplyOperation = input -> "Multiply " + input;
    public static Function<String, String> divideOperation = input -> "Divide " + input;

    public static Function<String, String> getOperationByUserInput(String userInput) {
        switch (userInput) {
            case "add":
                return addOperation;
            case "subtract":
                return subtractOperation;
            case "multiply":
                return multiplyOperation;
            case "divide":
                return divideOperation;
            default:
                return input -> "Invalid operation";
        }
    }

    public static void main(String[] args) {
        String userInput = "add";
        Function<String, String> operation = getOperationByUserInput(userInput);
        System.out.println(operation.apply("5"));
    }
}