5. 函数声明与闭包（Closure）

sickworm

发布于 2020-02-11 10:28:42

1.1K0

发布于 2020-02-11 10:28:42

文章被收录于专栏：sickworm

今天介绍闭包。闭包也不是新东西了。其实 Kotlin 就基本没有新东西，不，是新语言都基本没有新东西。新语言都是把近些年好用的特性组装起来，再加点自己的见解，因地制宜

一下。

0. 闭包概念介绍

闭包我第一次接触是在 JavaScript 上，函数当作“一等公民”的编程语言都有这个概念。函数是“一等公民”的意思是，函数和变量一样，它是某种类型的实例，可以被赋值，可以被引用。当然函数还可以被调用。变量类型是某个声明的类，函数类型就是规定了入参个数，类型和返回值类型（不规定名字。函数名就和变量名一样，随便起）。如我要声明 Kotlin 一个函数类型，它的入参是两个整数，出参是一个整数，那应该这样写：val add: (Int, Int) -> Int。箭头左边括号内表示入参，括号不可省略。箭头右边表示返回值。

wiki 上闭包的定义是：引用了自由变量的函数，这个被引用的自由变量将和这个函数一同存在，即使已经离开了创造它的环境也不例外。从定义来说，对闭包的理解，是基于普通函数之上的。一般的函数，能处理的只有入参和全局变量，然后返回一个结果。闭包比普通函数更多一点能力，它还捕获了当前环境的局部变量。当然了，捕获局部变量的前提是可以在局部环境里声明一个函数，这只有把函数当作“一等公民”才可以做到。

1. 闭包与匿名类比较

在函数不是“一等公民”的 Java 老大哥这里，匿名类其实就是代替闭包而存在的。只不过 Java 严格要求所有函数都需要在类里面，所以巧妙的把“声明一个函数”这样的行为变成了“声明一个接口”或“重写一个方法”。匿名类也可以捕获当前环境的 final 局部变量。但和闭包不一样的是，匿名类无法修改捕获的局部变量（final 不可修改）。

而匿名类能引用 final 的局部变量，是因为在编译阶段，会把该局部变量作为匿名类的构造参数传入。因为匿名类修改的变量不是真正的局部变量，而是自己的构造参数，外部局部变量并没有被修改。所以 Java 编译器不允许匿名类引用非 final 变量。

Java8 lambda 是进一步接近闭包的特性，lambda 的 JVM 实现是类似函数指针的东西。但注意： Java7 的 lambda 语法糖兼容不是真正的 lambda，它只是简化了匿名类的书写。同样的 lambda 也只能引用 final 变量。

2. 闭包使用

我们来看一个闭包的例子：

fun returnFun(): () -> Int {
    var count = 0
    return { count++ }
}

fun main() {
    val function = returnFun()
    val function2 = returnFun()
    println(function()) // 0
    println(function()) // 1
    println(function()) // 2

    println(function2()) // 0
    println(function2()) // 1
    println(function2()) // 2
}

分析上面的代码，returnFun返回了一个函数，这个函数没有入参，返回值是Int。我们可以用变量接收它，还可以调用它。function和function2分别是我创建的两个函数实例。

可以看到，我每调用一次function()，count都会加一，说明count 被function持有了而且可以被修改。而function2和function的count是独立的，不是共享的。

而我们通过 jadx 反编译可以看到：

public final class ClosureKt {
    @NotNull
    public static final Function0<Integer> returnFun() {
        IntRef intRef = new IntRef();
        intRef.element = 0;
        return (Function0) new 1<>(intRef);
    }

    public static final void main() {
        Function0 function = returnFun();
        Function0 function2 = returnFun();
        System.out.println(((Number) function.invoke()).intValue());
        System.out.println(((Number) function.invoke()).intValue());
        System.out.println(((Number) function2.invoke()).intValue());
        System.out.println(((Number) function2.invoke()).intValue());
    }
}

被闭包引用的 int 局部变量，会被封装成 IntRef 这个类。这个 IntRef 里面保存着 int 变量，原函数和闭包都可以通过 intRef 来读写 int 变量。Kotlin 正是通过这种办法使得局部变量可修改。除了 IntRef，还有 LongRef，FloatRef 等，如果是非基础类型，就统一用 ObjectRef 即可。Ref 家族源码：https://github.com/JetBrains/kotlin/blob/master/libraries/stdlib/jvm/runtime/kotlin/jvm/internal/Ref.java