Go语言 和 Java语言对比理解系列三：函数作为参数和返回值

Go函数

基于Go语言的特性，其天生支持将函数作为参数或者返回值。

函数作为参数

func main() {
	fun1("xujian", fun2)
}

// 函数fun1的第二个参数是一个函数类型的参数
func fun1(str string, f func(s string)) {
	fmt.Println("执行fun1")
	// 调用参数传过来的函数
	f(str);
}

// 被当成参数传递的函数
func fun2(s string) {
	fmt.Println("执行fun2", s)
}

输出结果如下：

可以看出在主函数里调用fun1函数，然后在fun1里又去执行了参数传过来的fun2函数。

函数作为返回值

func main() {
	// fun1返回的函数赋给fun2
	fun2 := fun1()
	// 执行fun1返回的函数即fun2
	s := fun2("xujian")
	// 打印fun2返回值
	fmt.Println("====>",s)
}

// 函数fun1的返回一个匿名函数
func fun1() func(s string) string {
	fmt.Println("执行fun1")
	return func(s string) string {
		fmt.Println("执行返回的函数", s)
		return "嘿嘿嘿";
	}
}

执行结果如下：

Java函数

Java一开始是不支持将函数作为函数的参数或返回值的，直到Java8引入了函数式接口public interface Function<T, R>。

函数作为参数

public static void main(String[] args) {
        fun1(fun2("嘿嘿"));
    }

    /**
     * 返回一个函数式接口
     * @return
     */
    public static Function<User, String> fun2(String s) {
        return user -> {
            System.out.println("执行返回的函数 " + s);
            return user.getName();
        };
    }

    /**
     * 将函数式接口作为参数
     * @param f
     */
    public static void fun1(Function<User, String> f) {
        System.out.println("执行fun1");
        User user = new User();
        user.setName("xujian");
        // 执行参数传来的函数
        String res = f.apply(user);
        System.out.println(res);
    }

执行结果如下：

可以看到达到了Go语言的效果，即将一个函数作为参数传给另一个函数，然后在其内部执行传递来的函数。

函数作为返回值

public static void main(String[] args) {
        User user = new User();
        user.setName("xujian");
        // 执行fun1返回一个函数
        Function<User, String> function = fun1("嘿嘿");
        // 执行返回来的函数
        String res = function.apply(user);
        System.out.println(res);
    }

    /**
     * 返回一个函数式接口
     * @return
     */
    public static Function<User, String> fun1(String s) {
    	System.out.println("执行fun1");
        return user -> {
            System.out.println("执行返回的函数 " + s);
            return user.getName();
        };
    }

执行结果如下：

可以看到达到了类似于Go语言的效果，即一个函数作为另一个函数的返回值。

总结

对于将函数作为参数和返回值，Go和Java都有自己的语言特性实现，但是在理解上是有异曲同工的，这样对比下来更能加深印象～