Kotlin-高阶函数的使用(一)

用过Kotlin的同学都知道，那些扩展方法用起来简直不要太爽，那么这些扩展方法是怎么定义实现的呢，本文介绍了Kotlin高阶函数的使用,带你实现自己的超溜扩展方法，并且能让你能看懂扩展方法源码，阅读本文需要有一定的Kotlin基础，请谨慎阅读

1.定义:什么是高阶函数

高阶函数就是以另一个函数作为参数或返回值的函数，Kotlin可以以lambda或参数引用作为参数或返回值，所以，任何以lambda或函数引用作为参数或返回值的都是高阶函数

2.函数类型

要使用Kotlin的高阶函数就必须遵循它的函数类型
先来看一个简单的例子，这是一个简单的函数类型申明

val sum = { x: Int, y: Int -> x + y }

之所以能这么写得益于Kotlin的类型推导，它的显示写法是这样的：

var sum:(Int,Int)-> Int = {x , y-> x + y}

(Int,Int)是参数类型，->后面的值Int是返回类型,组合在一起就构成了一个函数类型,以上属于有参有返回值类型，下面列举其他几种类型

var action:() ->Unit = {} //无参无返回值
var a = {} //类型推导后的简化写法

var action:(Int) ->Unit = {} //有参无返回值

var action:() -> Int = {} //无参有返回值

3.使用高阶函数(将函数类型作为参数)

直接看代码

 fun getNumResult(result: (Int, Int) -> Int): Int {
        return result(1,2)                                                                                                                       
    }

//调用
var value = getNumResult{ a, b -> a + b }
==> value = 3

var value = getNumResult{ a, b -> a * b }
==> value = 2

依照函数类型所示，a和b对应两个Int类型的参数，这个名称可以任意定义，a+b为返回值，随后return 了一个lambda的返回值，看起来是不是很强大,而且依照Kotlin的特性，当参数只有一个lambda参数的时候，括号可以省略，因此就变成了实例中的那样简洁的调用

像Kotlin中的filter等操作符，都是这样定义实现的，感兴趣的可以研究下

看到这里，可能有同学会问了，Kotlin不是号称与JAVA兼容吗，那再Java中如何调用呢，这就要从Kotlin函数的生成原理说起了，Kotlin的这些函数类型其实会被申明成普通的接口，一个函数类型的参数就会变成Function1的实现，N个函数类型的参数就是FunctionN，这个接口定义了一个invoke方法，调用该方法就会执行函数，So,我们可以这么写:

Java8: 因为支持lambda，所以依旧可以这样写

int value = getNumResult(a ,b - >a + b);

Java8以下版本

int value = getNumResult(
    new Function2<Integer, Integer, Integer>() {
            @Override
            public Integer invoke(Integer a, Integer b) {
                return a + b;
            }
        });

注意，如果定义的返回值是Unit,例如forEach高阶函数，在Java中就必须显示返回一个Unit对象,因为Unit是一个单例对象，因此可以这样定义返回

return Unit.INSTANCE；

4.使用高阶函数(将函数类型作为返回值)

顾名思义，相信同学们把作为参数的高阶函数理解透了这个很好理解，示例如下:

//定义一个类用于装载参数
class Num(var num: Int)

//此方法的作用是如果type等于1，那么就从Num类中取出num进行乘法操作，如果不等于一，就进行相加操作
    fun getNum(type: Int): (Num) -> Int {
        if (type == 1) {
          //这是一种写法
            return { entity ->
                entity.num * entity.num
            }
        } else {
         //这是另一种简写，可以用it代替entity参数
            return {
                it.num + it.num
            }
        }
    }

使用:
var ride = getNum(1) //传递一个1进去代表使用乘法
var result = ride(Num(3)) //得到一个以Num作为参数的高阶函数
==> result = 9

var sum = getNum(2)
var result = sum(Num(3))
==> result = 6

如果对高阶函数感兴趣可以继续参阅我的第二篇:Kotlin-高阶函数的使用(二)